Add CPropertyPageImpl that allows us to make property pages object oriented
[reactos.git] / sdk / tools / mkshelllink / mkshelllink.c
1 /* COPYRIGHT: See COPYING in the top level directory
2 * PROJECT: ReactOS Shell Link maker
3 * FILE: tools/mkshelllink/mkshelllink.c
4 * PURPOSE: Shell Link maker
5 * PROGRAMMER: Rafal Harabien
6 */
7
8 #include <stdio.h>
9 #include <stdlib.h>
10 #include <string.h>
11 #include <ctype.h>
12
13 #ifndef _MSC_VER
14 #include <stdint.h>
15 #else
16 typedef unsigned __int8 uint8_t;
17 typedef unsigned __int16 uint16_t;
18 typedef unsigned __int32 uint32_t;
19 #endif
20
21 #define SW_SHOWNORMAL 1
22 #define SW_SHOWMINNOACTIVE 7
23
24 typedef struct _GUID
25 {
26 uint32_t Data1;
27 uint16_t Data2;
28 uint16_t Data3;
29 uint8_t Data4[8];
30 } GUID;
31
32 typedef struct _FILETIME {
33 uint32_t dwLowDateTime;
34 uint32_t dwHighDateTime;
35 } FILETIME, *PFILETIME;
36
37 #define DEFINE_GUID2(name,l,w1,w2,b1,b2,b3,b4,b5,b6,b7,b8) const GUID name = { l,w1,w2,{ b1,b2,b3,b4,b5,b6,b7,b8 } }
38 DEFINE_GUID2(CLSID_ShellLink,0x00021401L,0,0,0xC0,0,0,0,0,0,0,0x46);
39 DEFINE_GUID2(CLSID_MyComputer,0x20D04FE0,0x3AEA,0x1069,0xA2,0xD8,0x08,0x00,0x2B,0x30,0x30,0x9D);
40
41 #define LINK_ID_LIST 0x01
42 #define LINK_FILE 0x02
43 #define LINK_DESCRIPTION 0x04
44 #define LINK_RELATIVE_PATH 0x08
45 #define LINK_WORKING_DIR 0x10
46 #define LINK_CMD_LINE_ARGS 0x20
47 #define LINK_ICON 0x40
48 #define LINK_UNICODE 0x80
49
50 #define LOCATOR_LOCAL 0x1
51 #define LOCATOR_NETWORK 0x2
52
53 #pragma pack(push, 1)
54
55 /* Specification: http://ithreats.files.wordpress.com/2009/05/lnk_the_windows_shortcut_file_format.pdf */
56
57 typedef struct _LNK_HEADER
58 {
59 uint32_t Signature;
60 GUID Guid;
61 uint32_t Flags;
62 uint32_t Attributes;
63 FILETIME CreationTime;
64 FILETIME ModificationTime;
65 FILETIME LastAccessTime;
66 uint32_t FileSize;
67 uint32_t IconNr;
68 uint32_t Show;
69 uint32_t Hotkey;
70 uint32_t Unknown;
71 uint32_t Unknown2;
72 } LNK_HEADER;
73
74 typedef struct _LNK_LOCATOR_INFO
75 {
76 uint32_t Size;
77 uint32_t DataOffset;
78 uint32_t Flags;
79 uint32_t LocalVolumeInfoOffset;
80 uint32_t LocalBasePathnameOffset;
81 uint32_t NetworkVolumeInfoOffset;
82 uint32_t RemainingPathnameOffset;
83 char Data[0];
84 } LNK_LOCATOR_INFO;
85
86 typedef struct _LNK_LOCAL_VOLUME_INFO
87 {
88 uint32_t Size;
89 uint32_t VolumeType; /* See GetDriveType */
90 uint32_t SerialNumber;
91 uint32_t VolumeNameOffset;
92 char VolumeLabel[0];
93 } LNK_LOCAL_VOLUME_INFO;
94
95 #define PT_GUID 0x1F
96 #define PT_DRIVE1 0x2F
97 #define PT_FOLDER 0x31
98 #define PT_VALUE 0x32
99
100 typedef struct _ID_LIST_FILE
101 {
102 uint16_t Size;
103 uint8_t Type;
104 uint8_t dummy;
105 uint32_t dwFileSize;
106 uint16_t uFileDate;
107 uint16_t uFileTime;
108 uint16_t uFileAttribs;
109 char szName[0];
110 } ID_LIST_FILE;
111
112 typedef struct _ID_LIST_GUID
113 {
114 uint16_t Size;
115 uint8_t Type;
116 uint8_t dummy;
117 GUID guid;
118 } ID_LIST_GUID;
119
120 typedef struct _ID_LIST_DRIVE
121 {
122 uint16_t Size;
123 uint8_t Type;
124 char szDriveName[20];
125 uint16_t unknown;
126 } ID_LIST_DRIVE;
127
128 #pragma pack(pop)
129
130 int main(int argc, const char *argv[])
131 {
132 unsigned i;
133 const char *pszOutputPath = "shortcut.lnk";
134 const char *pszTarget = NULL;
135 const char *pszDescription = "Description";
136 const char *pszWorkingDir = NULL;
137 const char *pszCmdLineArgs = NULL;
138 const char *pszIcon = NULL;
139 int IconNr = 0;
140 GUID Guid = CLSID_MyComputer;
141 int bHelp = 0, bMinimized = 0;
142 FILE *pFile;
143 LNK_HEADER Header;
144 uint16_t uhTmp;
145 uint32_t dwTmp;
146
147 for (i = 1; i < argc; ++i)
148 {
149 if (argv[i][0] != '-' && argv[i][0] != '/')
150 pszTarget = argv[i];
151 else if (!strcmp(argv[i] + 1, "h"))
152 bHelp = 1;
153 else if (!strcmp(argv[i] + 1, "o") && i + 1 < argc)
154 pszOutputPath = argv[++i];
155 else if (!strcmp(argv[i] + 1, "d") && i + 1 < argc)
156 pszDescription = argv[++i];
157 else if (!strcmp(argv[i] + 1, "w") && i + 1 < argc)
158 pszWorkingDir = argv[++i];
159 else if (!strcmp(argv[i] + 1, "c") && i + 1 < argc)
160 pszCmdLineArgs = argv[++i];
161 else if (!strcmp(argv[i] + 1, "i") && i + 1 < argc)
162 {
163 pszIcon = argv[++i];
164 if (i + 1 < argc && isdigit(argv[i + 1][0]))
165 IconNr = atoi(argv[++i]);
166 }
167 else if (!strcmp(argv[i] + 1, "m"))
168 bMinimized = 1;
169 else if (!strcmp(argv[i] + 1, "g") && i + 1 < argc)
170 {
171 unsigned Data4Tmp[8], j;
172
173 sscanf(argv[++i], "{%8x-%4hx-%4hx-%2x%2x-%2x%2x%2x%2x%2x%2x}",
174 &Guid.Data1, &Guid.Data2, &Guid.Data3,
175 &Data4Tmp[0], &Data4Tmp[1], &Data4Tmp[2], &Data4Tmp[3],
176 &Data4Tmp[4], &Data4Tmp[5], &Data4Tmp[6], &Data4Tmp[7]);
177 for (j = 0; j < 8; ++j)
178 Guid.Data4[j] = (uint8_t)Data4Tmp[j];
179 }
180 else
181 printf("Invalid option: %s\n", argv[i]);
182 }
183
184 if (!pszTarget || bHelp)
185 {
186 printf("Usage: %s [-o path][-d descr][-w path][-c cmd_line_args][-i icon_path [nr]][-h][-g guid] target\n"
187 "-o path\tSets output path\n"
188 "-d descr\tSets shortcut description\n"
189 "-w path\tSets working directory for executable\n"
190 "-c cmd_line_args\tSets command line arguments passed to program\n"
191 "-i icon_path [nr]\tSets icon file and optionally icon index\n"
192 "-m\tStart minimized\n"
193 "-g guid\tSets GUID to which target path is relative. Default value is MyComputer GUID.\n"
194 "target\tAbsolute or relative to guid specified with -g option path\n", argv[0]);
195 return 0;
196 }
197
198 pFile = fopen(pszOutputPath, "wb");
199 if (!pFile)
200 {
201 printf("Failed to open %s\n", pszOutputPath);
202 return -1;
203 }
204
205 // Header
206 memset(&Header, 0, sizeof(Header));
207 Header.Signature = (uint32_t)'L';
208 Header.Guid = CLSID_ShellLink;
209 Header.Flags = LINK_ID_LIST;
210 if (pszDescription)
211 Header.Flags |= LINK_DESCRIPTION;
212 if (pszWorkingDir)
213 Header.Flags |= LINK_WORKING_DIR;
214 if (pszCmdLineArgs)
215 Header.Flags |= LINK_CMD_LINE_ARGS;
216 if (pszIcon)
217 Header.Flags |= LINK_ICON;
218 Header.IconNr = IconNr;
219 Header.Show = bMinimized ? SW_SHOWMINNOACTIVE : SW_SHOWNORMAL;
220 fwrite(&Header, sizeof(Header), 1, pFile);
221
222 if (Header.Flags & LINK_ID_LIST)
223 {
224 ID_LIST_FILE IdListFile;
225 ID_LIST_GUID IdListGuid;
226 ID_LIST_DRIVE IdListDrive;
227 unsigned cbListSize = sizeof(IdListGuid) + sizeof(uint16_t), cchName;
228 const char *pszName = pszTarget;
229
230 // ID list
231 // It seems explorer does not accept links without id list. List is relative to desktop.
232
233 pszName = pszTarget;
234
235 if (pszName[0] && pszName[1] == ':')
236 {
237 cbListSize += sizeof(IdListDrive);
238 pszName += 2;
239 while (*pszName == '\\' || *pszName == '/')
240 ++pszName;
241 }
242
243 while (*pszName)
244 {
245 cchName = 0;
246 while (pszName[cchName] && pszName[cchName] != '\\' && pszName[cchName] != '/')
247 ++cchName;
248
249 if (cchName != 1 || pszName[0] != '.')
250 cbListSize += sizeof(IdListFile) + 2 * (cchName + 1);
251
252 pszName += cchName;
253 while (*pszName == '\\' || *pszName == '/')
254 ++pszName;
255 }
256
257 uhTmp = cbListSize;
258 fwrite(&uhTmp, sizeof(uhTmp), 1, pFile); // size
259
260 IdListGuid.Size = sizeof(IdListGuid);
261 IdListGuid.Type = PT_GUID;
262 IdListGuid.dummy = 0x50;
263 IdListGuid.guid = Guid;
264 fwrite(&IdListGuid, sizeof(IdListGuid), 1, pFile);
265
266 pszName = pszTarget;
267
268 if (isalpha(pszName[0]) && pszName[1] == ':')
269 {
270 memset(&IdListDrive, 0, sizeof(IdListDrive));
271 IdListDrive.Size = sizeof(IdListDrive);
272 IdListDrive.Type = PT_DRIVE1;
273 sprintf(IdListDrive.szDriveName, "%c:\\", pszName[0]);
274 fwrite(&IdListDrive, sizeof(IdListDrive), 1, pFile);
275 pszName += 2;
276 while(*pszName == '\\' || *pszName == '/')
277 ++pszName;
278 }
279
280 while (*pszName)
281 {
282 cchName = 0;
283 while (pszName[cchName] && pszName[cchName] != '\\' && pszName[cchName] != '/')
284 ++cchName;
285
286 if (cchName != 1 || pszName[0] != '.')
287 {
288 memset(&IdListFile, 0, sizeof(IdListFile));
289 IdListFile.Size = sizeof(IdListFile) + 2 * (cchName + 1);
290 if (!pszName[cchName])
291 IdListFile.Type = PT_VALUE; // File
292 else
293 IdListFile.Type = PT_FOLDER;
294 fwrite(&IdListFile, sizeof(IdListFile), 1, pFile);
295 fwrite(pszName, cchName, 1, pFile);
296 fputc(0, pFile);
297 fwrite(pszName, cchName, 1, pFile);
298 fputc(0, pFile);
299 }
300
301 pszName += cchName;
302 while (*pszName == '\\' || *pszName == '/')
303 ++pszName;
304 }
305
306 uhTmp = 0; // list end
307 fwrite(&uhTmp, sizeof(uhTmp), 1, pFile);
308 }
309
310 if (Header.Flags & LINK_DESCRIPTION)
311 {
312 // Dscription
313 uhTmp = strlen(pszDescription);
314 fwrite(&uhTmp, sizeof(uhTmp), 1, pFile);
315 fputs(pszDescription, pFile);
316 }
317
318 if (Header.Flags & LINK_RELATIVE_PATH)
319 {
320 // Relative Path
321 uhTmp = strlen(pszTarget);
322 fwrite(&uhTmp, sizeof(uhTmp), 1, pFile);
323 fputs(pszTarget, pFile);
324 }
325
326 if (Header.Flags & LINK_WORKING_DIR)
327 {
328 // Working Dir
329 uhTmp = strlen(pszWorkingDir);
330 fwrite(&uhTmp, sizeof(uhTmp), 1, pFile);
331 fputs(pszWorkingDir, pFile);
332 }
333
334 if (Header.Flags & LINK_CMD_LINE_ARGS)
335 {
336 // Command line arguments
337 uhTmp = strlen(pszCmdLineArgs);
338 fwrite(&uhTmp, sizeof(uhTmp), 1, pFile);
339 fputs(pszCmdLineArgs, pFile);
340 }
341
342 if (Header.Flags & LINK_ICON)
343 {
344 // Command line arguments
345 uhTmp = strlen(pszIcon);
346 fwrite(&uhTmp, sizeof(uhTmp), 1, pFile);
347 fputs(pszIcon, pFile);
348 }
349
350 // Extra stuff
351 dwTmp = 0;
352 fwrite(&dwTmp, sizeof(dwTmp), 1, pFile);
353
354 fclose(pFile);
355
356 return 0;
357 }