huwcampbell.com
Reactos
1#include <string.h>
2#include <stdlib.h>
3#include <stdio.h>
4
5#ifndef max
6#define max(a, b) ((a) > (b) ? (a) : (b))
7#endif
8
9int
10write_if_change(char* outbuf, char* filename)
11{
12 FILE* out;
13 unsigned int end;
14 char* cmpbuf;
15 unsigned int stat;
16
17 out = fopen(filename, "rb");
18 if (out == NULL)
19 {
20 out = fopen(filename, "wb");
21 if (out == NULL)
22 {
23 fprintf(stderr, "Unable to create output file\n");
24 return(1);
25 }
26 fputs(outbuf, out);
27 fclose(out);
28 return(0);
29 }
30
31 fseek(out, 0, SEEK_END);
32 end = ftell(out);
33 cmpbuf = malloc(end);
34 if (cmpbuf == NULL)
35 {
36 fprintf(stderr, "Out of memory\n");
37 fclose(out);
38 return(1);
39 }
40
41 fseek(out, 0, SEEK_SET);
42 stat = fread(cmpbuf, 1, end, out);
43 if (stat != end)
44 {
45 fprintf(stderr, "Failed to read data\n");
46 fclose(out);
47 free(cmpbuf);
48 return(1);
49 }
50 if (end == strlen(outbuf) && memcmp(cmpbuf, outbuf, end) == 0)
51 {
52 fclose(out);
53 free(cmpbuf);
54 return(0);
55 }
56
57 fclose(out);
58 free(cmpbuf);
59 out = fopen(filename, "wb");
60 if (out == NULL)
61 {
62 fprintf(stderr, "Unable to create output file\n");
63 return(1);
64 }
65
66 stat = fwrite(outbuf, 1, strlen(outbuf), out);
67 if (strlen(outbuf) != stat)
68 {
69 fprintf(stderr, "Unable to write output file\n");
70 fclose(out);
71 return(1);
72 }
73 fclose(out);
74 return(0);
75}
76
77int
78main(int argc, char* argv[])
79{
80 int include_tests;
81 unsigned int i;
82 char* outbuf;
83 char* s;
84 char config[512];
85
86 if (argc == 1)
87 {
88 fprintf(stderr, "Not enough arguments\n");
89 return(1);
90 }
91
92 outbuf = malloc(256 * 1024);
93 if (outbuf == NULL)
94 {
95 fprintf(stderr, "Out of memory 1\n");
96 return(1);
97 }
98
99 s = outbuf;
100 s = s + sprintf(s, "/* Automatically generated, ");
101 s = s + sprintf(s, "Edit the Makefile to change configuration */\n");
102 s = s + sprintf(s, "#ifndef __INCLUDE_CONFIG_H\n");
103 s = s + sprintf(s, "#define __INCLUDE_CONFIG_H\n");
104 strcpy(config, "");
105 include_tests = 0;
106 for (i = 2; i < argc; i++)
107 {
108 if (strcmp(argv[i], "REGTESTS") == 0)
109 {
110 include_tests = 1;
111 }
112 else
113 {
114 s = s + sprintf(s, "#ifndef %s\n", argv[i]);
115 s = s + sprintf(s, "#define %s\n", argv[i]);
116 s = s + sprintf(s, "#endif /* %s */\n", argv[i]);
117 }
118 strcat(config, argv[i]);
119 if (i != (argc - 1))
120 {
121 strcat(config, " ");
122 }
123 }
124 if (include_tests)
125 {
126 s = s + sprintf(s, "#ifndef __ASM__\n");
127 s = s + sprintf(s, "extern void PrepareTests();\n");
128 s = s + sprintf(s, "#define PREPARE_TESTS PrepareTests();\n");
129 s = s + sprintf(s, "#endif /* __ASM__ */\n");
130 }
131 else
132 {
133 s = s + sprintf(s, "#define PREPARE_TESTS\n");
134 }
135 s = s + sprintf(s, "#define CONFIG \"%s\"\n", config);
136 s = s + sprintf(s, "#endif /* __INCLUDE_CONFIG_H */\n");
137
138 return(write_if_change(outbuf, argv[1]));
139}