1/*
2 * ----------------------------------------------------------------------------
3 * nmakehlp.c --
4 *
5 *	This is used to fix limitations within nmake and the environment.
6 *
7 * Copyright (c) 2002 by David Gravereaux.
8 * Copyright (c) 2006 by Pat Thoyts
9 *
10 * See the file "license.terms" for information on usage and redistribution of
11 * this file, and for a DISCLAIMER OF ALL WARRANTIES.
12 * ----------------------------------------------------------------------------
13 */
14
15#define _CRT_SECURE_NO_DEPRECATE
16#include <windows.h>
17#ifdef _MSC_VER
18#pragma comment (lib, "user32.lib")
19#pragma comment (lib, "kernel32.lib")
20#endif
21#include <stdio.h>
22#include <math.h>
23
24/*
25 * This library is required for x64 builds with _some_ versions of MSVC
26 */
27#if defined(_M_IA64) || defined(_M_AMD64)
28#if _MSC_VER >= 1400 && _MSC_VER < 1500
29#pragma comment(lib, "bufferoverflowU")
30#endif
31#endif
32
33/* ISO hack for dumb VC++ */
34#ifdef _MSC_VER
35#define   snprintf	_snprintf
36#endif
37
38
39/* protos */
40
41static int CheckForCompilerFeature(const char *option);
42static int CheckForLinkerFeature(char **options, int count);
43static int IsIn(const char *string, const char *substring);
44static int SubstituteFile(const char *substs, const char *filename);
45static int QualifyPath(const char *path);
46static int LocateDependency(const char *keyfile);
47static const char *GetVersionFromFile(const char *filename, const char *match, int numdots);
48static DWORD WINAPI ReadFromPipe(LPVOID args);
49
50/* globals */
51
52#define CHUNK	25
53#define STATICBUFFERSIZE    1000
54typedef struct {
55    HANDLE pipe;
56    char buffer[STATICBUFFERSIZE];
57} pipeinfo;
58
59pipeinfo Out = {INVALID_HANDLE_VALUE, ""};
60pipeinfo Err = {INVALID_HANDLE_VALUE, ""};
61
62/*
63 * exitcodes: 0 == no, 1 == yes, 2 == error
64 */
65
66int
67main(
68    int argc,
69    char *argv[])
70{
71    char msg[300];
72    DWORD dwWritten;
73    int chars;
74    const char *s;
75
76    /*
77     * Make sure children (cl.exe and link.exe) are kept quiet.
78     */
79
80    SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOOPENFILEERRORBOX);
81
82    /*
83     * Make sure the compiler and linker aren't effected by the outside world.
84     */
85
86    SetEnvironmentVariable("CL", "");
87    SetEnvironmentVariable("LINK", "");
88
89    if (argc > 1 && *argv[1] == '-') {
90	switch (*(argv[1]+1)) {
91	case 'c':
92	    if (argc != 3) {
93		chars = snprintf(msg, sizeof(msg) - 1,
94		        "usage: %s -c <compiler option>\n"
95			"Tests for whether cl.exe supports an option\n"
96			"exitcodes: 0 == no, 1 == yes, 2 == error\n", argv[0]);
97		WriteFile(GetStdHandle(STD_ERROR_HANDLE), msg, chars,
98			&dwWritten, NULL);
99		return 2;
100	    }
101	    return CheckForCompilerFeature(argv[2]);
102	case 'l':
103	    if (argc < 3) {
104		chars = snprintf(msg, sizeof(msg) - 1,
105	       		"usage: %s -l <linker option> ?<mandatory option> ...?\n"
106			"Tests for whether link.exe supports an option\n"
107			"exitcodes: 0 == no, 1 == yes, 2 == error\n", argv[0]);
108		WriteFile(GetStdHandle(STD_ERROR_HANDLE), msg, chars,
109			&dwWritten, NULL);
110		return 2;
111	    }
112	    return CheckForLinkerFeature(&argv[2], argc-2);
113	case 'f':
114	    if (argc == 2) {
115		chars = snprintf(msg, sizeof(msg) - 1,
116			"usage: %s -f <string> <substring>\n"
117			"Find a substring within another\n"
118			"exitcodes: 0 == no, 1 == yes, 2 == error\n", argv[0]);
119		WriteFile(GetStdHandle(STD_ERROR_HANDLE), msg, chars,
120			&dwWritten, NULL);
121		return 2;
122	    } else if (argc == 3) {
123		/*
124		 * If the string is blank, there is no match.
125		 */
126
127		return 0;
128	    } else {
129		return IsIn(argv[2], argv[3]);
130	    }
131	case 's':
132	    if (argc == 2) {
133		chars = snprintf(msg, sizeof(msg) - 1,
134			"usage: %s -s <substitutions file> <file>\n"
135			"Perform a set of string map type substutitions on a file\n"
136			"exitcodes: 0\n",
137			argv[0]);
138		WriteFile(GetStdHandle(STD_ERROR_HANDLE), msg, chars,
139			&dwWritten, NULL);
140		return 2;
141	    }
142	    return SubstituteFile(argv[2], argv[3]);
143	case 'V':
144	    if (argc != 4) {
145		chars = snprintf(msg, sizeof(msg) - 1,
146		    "usage: %s -V filename matchstring\n"
147		    "Extract a version from a file:\n"
148		    "eg: pkgIndex.tcl \"package ifneeded http\"",
149		    argv[0]);
150		WriteFile(GetStdHandle(STD_ERROR_HANDLE), msg, chars,
151		    &dwWritten, NULL);
152		return 0;
153	    }
154	    s = GetVersionFromFile(argv[2], argv[3], *(argv[1]+2) - '0');
155	    if (s && *s) {
156		printf("%s\n", s);
157		return 0;
158	    } else
159		return 1; /* Version not found. Return non-0 exit code */
160
161	case 'Q':
162	    if (argc != 3) {
163		chars = snprintf(msg, sizeof(msg) - 1,
164		    "usage: %s -Q path\n"
165		    "Emit the fully qualified path\n"
166		    "exitcodes: 0 == no, 1 == yes, 2 == error\n", argv[0]);
167		WriteFile(GetStdHandle(STD_ERROR_HANDLE), msg, chars,
168		    &dwWritten, NULL);
169		return 2;
170	    }
171	    return QualifyPath(argv[2]);
172
173	case 'L':
174	    if (argc != 3) {
175		chars = snprintf(msg, sizeof(msg) - 1,
176		    "usage: %s -L keypath\n"
177		    "Emit the fully qualified path of directory containing keypath\n"
178		    "exitcodes: 0 == success, 1 == not found, 2 == error\n", argv[0]);
179		WriteFile(GetStdHandle(STD_ERROR_HANDLE), msg, chars,
180		    &dwWritten, NULL);
181		return 2;
182	    }
183	    return LocateDependency(argv[2]);
184	}
185    }
186    chars = snprintf(msg, sizeof(msg) - 1,
187	    "usage: %s -c|-f|-l|-Q|-s|-V ...\n"
188	    "This is a little helper app to equalize shell differences between WinNT and\n"
189	    "Win9x and get nmake.exe to accomplish its job.\n",
190	    argv[0]);
191    WriteFile(GetStdHandle(STD_ERROR_HANDLE), msg, chars, &dwWritten, NULL);
192    return 2;
193}
194
195static int
196CheckForCompilerFeature(
197    const char *option)
198{
199    STARTUPINFO si;
200    PROCESS_INFORMATION pi;
201    SECURITY_ATTRIBUTES sa;
202    DWORD threadID;
203    char msg[300];
204    BOOL ok;
205    HANDLE hProcess, h, pipeThreads[2];
206    char cmdline[100];
207
208    hProcess = GetCurrentProcess();
209
210    ZeroMemory(&pi, sizeof(PROCESS_INFORMATION));
211    ZeroMemory(&si, sizeof(STARTUPINFO));
212    si.cb = sizeof(STARTUPINFO);
213    si.dwFlags   = STARTF_USESTDHANDLES;
214    si.hStdInput = INVALID_HANDLE_VALUE;
215
216    ZeroMemory(&sa, sizeof(SECURITY_ATTRIBUTES));
217    sa.nLength = sizeof(SECURITY_ATTRIBUTES);
218    sa.lpSecurityDescriptor = NULL;
219    sa.bInheritHandle = FALSE;
220
221    /*
222     * Create a non-inheritible pipe.
223     */
224
225    CreatePipe(&Out.pipe, &h, &sa, 0);
226
227    /*
228     * Dupe the write side, make it inheritible, and close the original.
229     */
230
231    DuplicateHandle(hProcess, h, hProcess, &si.hStdOutput, 0, TRUE,
232	    DUPLICATE_SAME_ACCESS | DUPLICATE_CLOSE_SOURCE);
233
234    /*
235     * Same as above, but for the error side.
236     */
237
238    CreatePipe(&Err.pipe, &h, &sa, 0);
239    DuplicateHandle(hProcess, h, hProcess, &si.hStdError, 0, TRUE,
240	    DUPLICATE_SAME_ACCESS | DUPLICATE_CLOSE_SOURCE);
241
242    /*
243     * Base command line.
244     */
245
246    lstrcpy(cmdline, "cl.exe -nologo -c -TC -Zs -X -Fp.\\_junk.pch ");
247
248    /*
249     * Append our option for testing
250     */
251
252    lstrcat(cmdline, option);
253
254    /*
255     * Filename to compile, which exists, but is nothing and empty.
256     */
257
258    lstrcat(cmdline, " .\\nul");
259
260    ok = CreateProcess(
261	    NULL,	    /* Module name. */
262	    cmdline,	    /* Command line. */
263	    NULL,	    /* Process handle not inheritable. */
264	    NULL,	    /* Thread handle not inheritable. */
265	    TRUE,	    /* yes, inherit handles. */
266	    DETACHED_PROCESS, /* No console for you. */
267	    NULL,	    /* Use parent's environment block. */
268	    NULL,	    /* Use parent's starting directory. */
269	    &si,	    /* Pointer to STARTUPINFO structure. */
270	    &pi);	    /* Pointer to PROCESS_INFORMATION structure. */
271
272    if (!ok) {
273	DWORD err = GetLastError();
274	int chars = snprintf(msg, sizeof(msg) - 1,
275		"Tried to launch: \"%s\", but got error [%u]: ", cmdline, err);
276
277	FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS|
278		FORMAT_MESSAGE_MAX_WIDTH_MASK, 0L, err, 0, (LPSTR)&msg[chars],
279		(300-chars), 0);
280	WriteFile(GetStdHandle(STD_ERROR_HANDLE), msg, lstrlen(msg), &err,NULL);
281	return 2;
282    }
283
284    /*
285     * Close our references to the write handles that have now been inherited.
286     */
287
288    CloseHandle(si.hStdOutput);
289    CloseHandle(si.hStdError);
290
291    WaitForInputIdle(pi.hProcess, 5000);
292    CloseHandle(pi.hThread);
293
294    /*
295     * Start the pipe reader threads.
296     */
297
298    pipeThreads[0] = CreateThread(NULL, 0, ReadFromPipe, &Out, 0, &threadID);
299    pipeThreads[1] = CreateThread(NULL, 0, ReadFromPipe, &Err, 0, &threadID);
300
301    /*
302     * Block waiting for the process to end.
303     */
304
305    WaitForSingleObject(pi.hProcess, INFINITE);
306    CloseHandle(pi.hProcess);
307
308    /*
309     * Wait for our pipe to get done reading, should it be a little slow.
310     */
311
312    WaitForMultipleObjects(2, pipeThreads, TRUE, 500);
313    CloseHandle(pipeThreads[0]);
314    CloseHandle(pipeThreads[1]);
315
316    /*
317     * Look for the commandline warning code in both streams.
318     *  - in MSVC 6 & 7 we get D4002, in MSVC 8 we get D9002.
319     */
320
321    return !(strstr(Out.buffer, "D4002") != NULL
322             || strstr(Err.buffer, "D4002") != NULL
323             || strstr(Out.buffer, "D9002") != NULL
324             || strstr(Err.buffer, "D9002") != NULL
325             || strstr(Out.buffer, "D2021") != NULL
326             || strstr(Err.buffer, "D2021") != NULL);
327}
328
329static int
330CheckForLinkerFeature(
331    char **options,
332    int count)
333{
334    STARTUPINFO si;
335    PROCESS_INFORMATION pi;
336    SECURITY_ATTRIBUTES sa;
337    DWORD threadID;
338    char msg[300];
339    BOOL ok;
340    HANDLE hProcess, h, pipeThreads[2];
341    int i;
342    char cmdline[255];
343
344    hProcess = GetCurrentProcess();
345
346    ZeroMemory(&pi, sizeof(PROCESS_INFORMATION));
347    ZeroMemory(&si, sizeof(STARTUPINFO));
348    si.cb = sizeof(STARTUPINFO);
349    si.dwFlags   = STARTF_USESTDHANDLES;
350    si.hStdInput = INVALID_HANDLE_VALUE;
351
352    ZeroMemory(&sa, sizeof(SECURITY_ATTRIBUTES));
353    sa.nLength = sizeof(SECURITY_ATTRIBUTES);
354    sa.lpSecurityDescriptor = NULL;
355    sa.bInheritHandle = TRUE;
356
357    /*
358     * Create a non-inheritible pipe.
359     */
360
361    CreatePipe(&Out.pipe, &h, &sa, 0);
362
363    /*
364     * Dupe the write side, make it inheritible, and close the original.
365     */
366
367    DuplicateHandle(hProcess, h, hProcess, &si.hStdOutput, 0, TRUE,
368	    DUPLICATE_SAME_ACCESS | DUPLICATE_CLOSE_SOURCE);
369
370    /*
371     * Same as above, but for the error side.
372     */
373
374    CreatePipe(&Err.pipe, &h, &sa, 0);
375    DuplicateHandle(hProcess, h, hProcess, &si.hStdError, 0, TRUE,
376	    DUPLICATE_SAME_ACCESS | DUPLICATE_CLOSE_SOURCE);
377
378    /*
379     * Base command line.
380     */
381
382    lstrcpy(cmdline, "link.exe -nologo ");
383
384    /*
385     * Append our option for testing.
386     */
387
388    for (i = 0; i < count; i++) {
389	lstrcat(cmdline, " \"");
390	lstrcat(cmdline, options[i]);
391	lstrcat(cmdline, "\"");
392    }
393
394    ok = CreateProcess(
395	    NULL,	    /* Module name. */
396	    cmdline,	    /* Command line. */
397	    NULL,	    /* Process handle not inheritable. */
398	    NULL,	    /* Thread handle not inheritable. */
399	    TRUE,	    /* yes, inherit handles. */
400	    DETACHED_PROCESS, /* No console for you. */
401	    NULL,	    /* Use parent's environment block. */
402	    NULL,	    /* Use parent's starting directory. */
403	    &si,	    /* Pointer to STARTUPINFO structure. */
404	    &pi);	    /* Pointer to PROCESS_INFORMATION structure. */
405
406    if (!ok) {
407	DWORD err = GetLastError();
408	int chars = snprintf(msg, sizeof(msg) - 1,
409		"Tried to launch: \"%s\", but got error [%u]: ", cmdline, err);
410
411	FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS|
412		FORMAT_MESSAGE_MAX_WIDTH_MASK, 0L, err, 0, (LPSTR)&msg[chars],
413		(300-chars), 0);
414	WriteFile(GetStdHandle(STD_ERROR_HANDLE), msg, lstrlen(msg), &err,NULL);
415	return 2;
416    }
417
418    /*
419     * Close our references to the write handles that have now been inherited.
420     */
421
422    CloseHandle(si.hStdOutput);
423    CloseHandle(si.hStdError);
424
425    WaitForInputIdle(pi.hProcess, 5000);
426    CloseHandle(pi.hThread);
427
428    /*
429     * Start the pipe reader threads.
430     */
431
432    pipeThreads[0] = CreateThread(NULL, 0, ReadFromPipe, &Out, 0, &threadID);
433    pipeThreads[1] = CreateThread(NULL, 0, ReadFromPipe, &Err, 0, &threadID);
434
435    /*
436     * Block waiting for the process to end.
437     */
438
439    WaitForSingleObject(pi.hProcess, INFINITE);
440    CloseHandle(pi.hProcess);
441
442    /*
443     * Wait for our pipe to get done reading, should it be a little slow.
444     */
445
446    WaitForMultipleObjects(2, pipeThreads, TRUE, 500);
447    CloseHandle(pipeThreads[0]);
448    CloseHandle(pipeThreads[1]);
449
450    /*
451     * Look for the commandline warning code in the stderr stream.
452     */
453
454    return !(strstr(Out.buffer, "LNK1117") != NULL ||
455	    strstr(Err.buffer, "LNK1117") != NULL ||
456	    strstr(Out.buffer, "LNK4044") != NULL ||
457	    strstr(Err.buffer, "LNK4044") != NULL ||
458	    strstr(Out.buffer, "LNK4224") != NULL ||
459	    strstr(Err.buffer, "LNK4224") != NULL);
460}
461
462static DWORD WINAPI
463ReadFromPipe(
464    LPVOID args)
465{
466    pipeinfo *pi = (pipeinfo *) args;
467    char *lastBuf = pi->buffer;
468    DWORD dwRead;
469    BOOL ok;
470
471  again:
472    if (lastBuf - pi->buffer + CHUNK > STATICBUFFERSIZE) {
473	CloseHandle(pi->pipe);
474	return (DWORD)-1;
475    }
476    ok = ReadFile(pi->pipe, lastBuf, CHUNK, &dwRead, 0L);
477    if (!ok || dwRead == 0) {
478	CloseHandle(pi->pipe);
479	return 0;
480    }
481    lastBuf += dwRead;
482    goto again;
483
484    return 0;  /* makes the compiler happy */
485}
486
487static int
488IsIn(
489    const char *string,
490    const char *substring)
491{
492    return (strstr(string, substring) != NULL);
493}
494
495/*
496 * GetVersionFromFile --
497 * 	Looks for a match string in a file and then returns the version
498 * 	following the match where a version is anything acceptable to
499 * 	package provide or package ifneeded.
500 */
501
502static const char *
503GetVersionFromFile(
504    const char *filename,
505    const char *match,
506    int numdots)
507{
508    static char szBuffer[100];
509    char *szResult = NULL;
510    FILE *fp = fopen(filename, "rt");
511
512    if (fp != NULL) {
513	/*
514	 * Read data until we see our match string.
515	 */
516
517	while (fgets(szBuffer, sizeof(szBuffer), fp) != NULL) {
518	    LPSTR p, q;
519
520	    p = strstr(szBuffer, match);
521	    if (p != NULL) {
522		/*
523		 * Skip to first digit after the match.
524		 */
525
526		p += strlen(match);
527		while (*p && !isdigit((unsigned char)*p)) {
528		    ++p;
529		}
530
531		/*
532		 * Find ending whitespace.
533		 */
534
535		q = p;
536		while (*q && (strchr("0123456789.ab", *q)) && (((!strchr(".ab", *q)
537			    && !strchr("ab", q[-1])) || --numdots))) {
538		    ++q;
539		}
540
541		*q = 0;
542		szResult = p;
543		break;
544	    }
545	}
546	fclose(fp);
547    }
548    return szResult;
549}
550
551/*
552 * List helpers for the SubstituteFile function
553 */
554
555typedef struct list_item_t {
556    struct list_item_t *nextPtr;
557    char * key;
558    char * value;
559} list_item_t;
560
561/* insert a list item into the list (list may be null) */
562static list_item_t *
563list_insert(list_item_t **listPtrPtr, const char *key, const char *value)
564{
565    list_item_t *itemPtr = (list_item_t *)malloc(sizeof(list_item_t));
566    if (itemPtr) {
567	itemPtr->key = strdup(key);
568	itemPtr->value = strdup(value);
569	itemPtr->nextPtr = NULL;
570
571	while(*listPtrPtr) {
572	    listPtrPtr = &(*listPtrPtr)->nextPtr;
573	}
574	*listPtrPtr = itemPtr;
575    }
576    return itemPtr;
577}
578
579static void
580list_free(list_item_t **listPtrPtr)
581{
582    list_item_t *tmpPtr, *listPtr = *listPtrPtr;
583    while (listPtr) {
584	tmpPtr = listPtr;
585	listPtr = listPtr->nextPtr;
586	free(tmpPtr->key);
587	free(tmpPtr->value);
588	free(tmpPtr);
589    }
590}
591
592/*
593 * SubstituteFile --
594 *	As windows doesn't provide anything useful like sed and it's unreliable
595 *	to use the tclsh you are building against (consider x-platform builds -
596 *	eg compiling AMD64 target from IX86) we provide a simple substitution
597 *	option here to handle autoconf style substitutions.
598 *	The substitution file is whitespace and line delimited. The file should
599 *	consist of lines matching the regular expression:
600 *	  \s*\S+\s+\S*$
601 *
602 *	Usage is something like:
603 *	  nmakehlp -S << $** > $@
604 *        @PACKAGE_NAME@ $(PACKAGE_NAME)
605 *        @PACKAGE_VERSION@ $(PACKAGE_VERSION)
606 *        <<
607 */
608
609static int
610SubstituteFile(
611    const char *substitutions,
612    const char *filename)
613{
614    static char szBuffer[1024], szCopy[1024];
615    list_item_t *substPtr = NULL;
616    FILE *fp, *sp;
617
618    fp = fopen(filename, "rt");
619    if (fp != NULL) {
620
621	/*
622	 * Build a list of substutitions from the first filename
623	 */
624
625	sp = fopen(substitutions, "rt");
626	if (sp != NULL) {
627	    while (fgets(szBuffer, sizeof(szBuffer), sp) != NULL) {
628		unsigned char *ks, *ke, *vs, *ve;
629		ks = (unsigned char*)szBuffer;
630		while (ks && *ks && isspace(*ks)) ++ks;
631		ke = ks;
632		while (ke && *ke && !isspace(*ke)) ++ke;
633		vs = ke;
634		while (vs && *vs && isspace(*vs)) ++vs;
635		ve = vs;
636		while (ve && *ve && !(*ve == '\r' || *ve == '\n')) ++ve;
637		*ke = 0, *ve = 0;
638		list_insert(&substPtr, (char*)ks, (char*)vs);
639	    }
640	    fclose(sp);
641	}
642
643	/* debug: dump the list */
644#ifndef NDEBUG
645	{
646	    int n = 0;
647	    list_item_t *p = NULL;
648	    for (p = substPtr; p != NULL; p = p->nextPtr, ++n) {
649		fprintf(stderr, "% 3d '%s' => '%s'\n", n, p->key, p->value);
650	    }
651	}
652#endif
653
654	/*
655	 * Run the substitutions over each line of the input
656	 */
657
658	while (fgets(szBuffer, sizeof(szBuffer), fp) != NULL) {
659	    list_item_t *p = NULL;
660	    for (p = substPtr; p != NULL; p = p->nextPtr) {
661		char *m = strstr(szBuffer, p->key);
662		if (m) {
663		    char *cp, *op, *sp;
664		    cp = szCopy;
665		    op = szBuffer;
666		    while (op != m) *cp++ = *op++;
667		    sp = p->value;
668		    while (sp && *sp) *cp++ = *sp++;
669		    op += strlen(p->key);
670		    while (*op) *cp++ = *op++;
671		    *cp = 0;
672		    memcpy(szBuffer, szCopy, sizeof(szCopy));
673		}
674	    }
675	    printf("%s", szBuffer);
676	}
677
678	list_free(&substPtr);
679    }
680    fclose(fp);
681    return 0;
682}
683
684BOOL FileExists(LPCTSTR szPath)
685{
686#ifndef INVALID_FILE_ATTRIBUTES
687    #define INVALID_FILE_ATTRIBUTES ((DWORD)-1)
688#endif
689    DWORD pathAttr = GetFileAttributes(szPath);
690    return (pathAttr != INVALID_FILE_ATTRIBUTES &&
691	    !(pathAttr & FILE_ATTRIBUTE_DIRECTORY));
692}
693
694
695/*
696 * QualifyPath --
697 *
698 *	This composes the current working directory with a provided path
699 *	and returns the fully qualified and normalized path.
700 *	Mostly needed to setup paths for testing.
701 */
702
703static int
704QualifyPath(
705    const char *szPath)
706{
707    char szCwd[MAX_PATH + 1];
708
709    GetFullPathName(szPath, sizeof(szCwd)-1, szCwd, NULL);
710    printf("%s\n", szCwd);
711    return 0;
712}
713
714/*
715 * Implements LocateDependency for a single directory. See that command
716 * for an explanation.
717 * Returns 0 if found after printing the directory.
718 * Returns 1 if not found but no errors.
719 * Returns 2 on any kind of error
720 * Basically, these are used as exit codes for the process.
721 */
722static int LocateDependencyHelper(const char *dir, const char *keypath)
723{
724    HANDLE hSearch;
725    char path[MAX_PATH+1];
726    size_t dirlen;
727    int keylen, ret;
728    WIN32_FIND_DATA finfo;
729
730    if (dir == NULL || keypath == NULL)
731	return 2; /* Have no real error reporting mechanism into nmake */
732    dirlen = strlen(dir);
733    if ((dirlen + 3) > sizeof(path))
734	return 2;
735    strncpy(path, dir, dirlen);
736    strncpy(path+dirlen, "\\*", 3);	/* Including terminating \0 */
737    keylen = strlen(keypath);
738
739#if 0 /* This function is not available in Visual C++ 6 */
740    /*
741     * Use numerics 0 -> FindExInfoStandard,
742     * 1 -> FindExSearchLimitToDirectories,
743     * as these are not defined in Visual C++ 6
744     */
745    hSearch = FindFirstFileEx(path, 0, &finfo, 1, NULL, 0);
746#else
747    hSearch = FindFirstFile(path, &finfo);
748#endif
749    if (hSearch == INVALID_HANDLE_VALUE)
750	return 1; /* Not found */
751
752    /* Loop through all subdirs checking if the keypath is under there */
753    ret = 1; /* Assume not found */
754    do {
755	int sublen;
756	/*
757	 * We need to check it is a directory despite the
758	 * FindExSearchLimitToDirectories in the above call. See SDK docs
759	 */
760	if ((finfo.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == 0)
761	    continue;
762	sublen = strlen(finfo.cFileName);
763	if ((dirlen+1+sublen+1+keylen+1) > sizeof(path))
764	    continue;		/* Path does not fit, assume not matched */
765	strncpy(path+dirlen+1, finfo.cFileName, sublen);
766	path[dirlen+1+sublen] = '\\';
767	strncpy(path+dirlen+1+sublen+1, keypath, keylen+1);
768	if (FileExists(path)) {
769	    /* Found a match, print to stdout */
770	    path[dirlen+1+sublen] = '\0';
771	    QualifyPath(path);
772	    ret = 0;
773	    break;
774	}
775    } while (FindNextFile(hSearch, &finfo));
776    FindClose(hSearch);
777    return ret;
778}
779
780/*
781 * LocateDependency --
782 *
783 *	Locates a dependency for a package.
784 *        keypath - a relative path within the package directory
785 *          that is used to confirm it is the correct directory.
786 *	The search path for the package directory is currently only
787 *      the parent and grandparent of the current working directory.
788 *      If found, the command prints
789 *         name_DIRPATH=<full path of located directory>
790 *      and returns 0. If not found, does not print anything and returns 1.
791 */
792static int LocateDependency(const char *keypath)
793{
794    size_t i;
795    int ret;
796    static const char *paths[] = {"..", "..\\..", "..\\..\\.."};
797
798    for (i = 0; i < (sizeof(paths)/sizeof(paths[0])); ++i) {
799	ret = LocateDependencyHelper(paths[i], keypath);
800	if (ret == 0)
801	    return ret;
802    }
803    return ret;
804}
805
806
807/*
808 * Local variables:
809 *   mode: c
810 *   c-basic-offset: 4
811 *   fill-column: 78
812 *   indent-tabs-mode: t
813 *   tab-width: 8
814 * End:
815 */
816