Respond immediately to ponder on/off commands
[uci2wb.git] / UCI2WB.c
1 /************************* UCI2WB by H.G.Muller ****************************/\r
2 \r
3 #define VERSION "1.10"\r
4 \r
5 #include <stdio.h>\r
6 #include <stdlib.h>\r
7 #ifdef WIN32\r
8 #  include <windows.h>\r
9 #  include <io.h>\r
10    HANDLE process;\r
11    DWORD thread_id;\r
12 #else\r
13 #  include <pthread.h>\r
14 #  include <signal.h>\r
15 #  define NO_ERROR 0\r
16 #  include <sys/time.h>\r
17    int GetTickCount() // with thanks to Tord\r
18    { struct timeval t; gettimeofday(&t, NULL); return t.tv_sec*1000 + t.tv_usec/1000; }\r
19 #endif\r
20 #include <fcntl.h>\r
21 #include <string.h>\r
22 \r
23 // Set VARIANTS for in WinBoard variant feature. (With -s option this will always be reset to use "shogi".)\r
24 #  define VARIANTS "normal,xiangqi"\r
25 \r
26 #define DPRINT if(debug) printf\r
27 \r
28 #define WHITE 0\r
29 #define BLACK 1\r
30 #define NONE  2\r
31 #define ANALYZE 3\r
32 \r
33 char move[2000][10], checkOptions[8192], iniPos[256], hashOpt[20], pause, pondering, suspended, ponder, post, hasHash, c, sc='c', *suffix, *variants;\r
34 int mps, tc, inc, sTime, depth, myTime, hisTime, stm, computer = NONE, memory, oldMem=0, cores, moveNr, lastDepth, lastScore, startTime, debug;\r
35 int statDepth, statScore, statNodes, statTime, currNr, size, collect, nr, sm, inex, on[500]; char currMove[20], moveMap[500][10]; // for analyze mode\r
36 \r
37 FILE *toE, *fromE;\r
38 int pid;\r
39 \r
40 #ifdef WIN32\r
41 WinPipe(HANDLE *hRd, HANDLE *hWr)\r
42 {\r
43   SECURITY_ATTRIBUTES saAttr;\r
44 \r
45   /* Set the bInheritHandle flag so pipe handles are inherited. */\r
46   saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);\r
47   saAttr.bInheritHandle = TRUE;\r
48   saAttr.lpSecurityDescriptor = NULL;\r
49 \r
50   /* Create a pipe */\r
51   return CreatePipe(hRd, hWr, &saAttr, 0);\r
52 }\r
53 #endif\r
54 \r
55 #define INIT 0\r
56 #define WAKEUP 1\r
57 #define PAUSE 2\r
58 \r
59 void\r
60 Sync (int action)\r
61 {\r
62 #ifdef WIN32\r
63         static HANDLE hWr, hRd; DWORD d; char c;\r
64         switch(action) {\r
65             case INIT:   WinPipe(&hRd, &hWr); break;\r
66             case WAKEUP: WriteFile(hWr, "\n", 1, &d, NULL); break;\r
67             case PAUSE:  ReadFile(hRd, &c, 1, &d, NULL);\r
68         }\r
69 #else\r
70         static int syncPipe[2]; char c;\r
71         switch(action) {\r
72             case INIT:   pipe(syncPipe); break;\r
73             case WAKEUP: write(syncPipe[1], "\n", 1); break;\r
74             case PAUSE:  read(syncPipe[0], &c, 1);\r
75         }\r
76 #endif\r
77 }\r
78 \r
79 void\r
80 StartSearch(char *ponder)\r
81 {       // send the 'go' command to engine. Suffix by ponder.\r
82         int nr = moveNr + (ponder[0] != 0); // we ponder for one move ahead!\r
83         fprintf(toE, "\ngo btime %d wtime %d", stm == BLACK ^ sc=='s' ? myTime : hisTime, stm == WHITE ^ sc=='s' ? myTime : hisTime);\r
84         DPRINT(    "\n# go btime %d wtime %d", stm == BLACK ^ sc=='s' ? myTime : hisTime, stm == WHITE ^ sc=='s' ? myTime : hisTime);\r
85         if(sTime > 0) { fprintf(toE, " movetime %d", sTime); DPRINT(" movetime %d", sTime); } else\r
86         if(mps) { fprintf(toE, " movestogo %d", mps*(nr/(2*mps)+1)-nr/2); DPRINT(" movestogo %d", mps*(nr/(2*mps)+1)-nr/2); }\r
87         if(inc && !suffix) { fprintf(toE, " winc %d binc %d", inc, inc); DPRINT(" winc %d binc %d", inc, inc); }\r
88         if(depth > 0) { fprintf(toE, " depth %d", depth); DPRINT(" depth %d", depth); }\r
89         if(suffix) { fprintf(toE, suffix, inc); DPRINT(suffix, inc); }\r
90         fprintf(toE, "%s\n", ponder);\r
91         DPRINT("%s\n", ponder);\r
92 }\r
93 \r
94 void\r
95 StopPonder(int pondering)\r
96 {\r
97         if(!pondering) return;\r
98         pause = 1;\r
99         fprintf(toE, "stop\n"); fflush(toE); DPRINT("# stop\n"); // note: 'pondering' remains set until engine acknowledges 'stop' with 'bestmove'\r
100         Sync(PAUSE); // wait for engine to acknowledge 'stop' with 'bestmove'.\r
101 }\r
102 \r
103 void\r
104 LoadPos(int moveNr)\r
105 {\r
106         int j;\r
107         fprintf(toE, "%s moves", iniPos);\r
108         DPRINT(    "# %s moves", iniPos);\r
109         for(j=0; j<moveNr; j++) { fprintf(toE, " %s", move[j]); DPRINT(" %s", move[j]); }\r
110 }\r
111 \r
112 void\r
113 StartPonder()\r
114 {\r
115         if(!move[moveNr][0]) return; // no ponder move\r
116         LoadPos(moveNr+1);\r
117         pondering = 1; lastDepth = 1;\r
118         StartSearch(" ponder");\r
119 }\r
120 \r
121 char *Convert(char *pv)\r
122 {   // convert Shogi coordinates to WB\r
123     char *p, *q, c;\r
124     static char buf[10000];\r
125     if(sc != 's') return pv;\r
126     p = pv; q = buf;\r
127     while(c = *p++) {\r
128         if(c >= '0' && c <= '9' || c >= 'a' && c <= 'z') *q++ = 'a'+'0'+size - c; else *q++ = c;\r
129     }\r
130     *q++ = 0;\r
131     return buf;\r
132 }\r
133 \r
134 void\r
135 Move4GUI(char *m)\r
136 {\r
137     if(sc == 's') {\r
138       // convert USI move to WB format\r
139       m[2] = 'a'+'0'+size - m[2];\r
140       m[3] = 'a'+'0'+size - m[3];\r
141       if(m[1] == '*') { // drop\r
142         m[1] = '@';\r
143       } else {\r
144         m[0] = 'a'+'0'+size - m[0];\r
145         m[1] = 'a'+'0'+size - m[1];\r
146         if((stm == WHITE ? (m[1]>'0'+size-size/3 || m[3]>'0'+size-size/3)\r
147                                 : (m[1] <= '0'+size/3 || m[3] <= '0'+size/3)) && m[4] != '+')\r
148              m[4] = '=', m[5] = 0;\r
149       }\r
150     }\r
151 }\r
152 \r
153 void *\r
154 Engine2GUI()\r
155 {\r
156     char line[1024], command[256];\r
157 \r
158     while(1) {\r
159         int i=0, x; char *p, dummy;\r
160 \r
161         fflush(stdout); fflush(toE);\r
162         while((line[i] = x = fgetc(fromE)) != EOF && line[i] != '\n') i++;\r
163         line[++i] = 0;\r
164         if(x == EOF) exit(0);\r
165         DPRINT("# engine said: %s", line), fflush(stdout);\r
166         sscanf(line, "%s", command);\r
167         if(!strcmp(command, "bestmove")) {\r
168             if(pause) { pondering = pause = 0; Sync(WAKEUP); continue; } // bestmove was reply to ponder miss or analysis result; ignore.\r
169             // move was a move to be played\r
170             if(strstr(line+9, "resign")) { printf("resign\n"); computer = NONE; }\r
171             if(strstr(line+9, "(none)") || strstr(line+9, "null") ||\r
172                strstr(line+9, "0000")) { printf("%s\n", lastScore < -99999 ? "resign" : "1/2-1/2 {stalemate}"); computer = NONE; }\r
173             sscanf(line, "bestmove %s", move[moveNr++]);\r
174             myTime -= (GetTickCount() - startTime)*1.02 + inc; // update own clock, so we can give correct wtime, btime with ponder\r
175             if(mps && ((moveNr+1)/2) % mps == 0) myTime += tc; if(sTime) myTime = sTime; // new session or move starts\r
176             // first start a new ponder search, if pondering is on and we have a move to ponder on\r
177             if(p = strstr(line+9, "ponder")) {\r
178               sscanf(p+7, "%s", move[moveNr]);\r
179               if(computer != NONE && ponder) {\r
180                 DPRINT("# ponder on %s\n", move[moveNr]);\r
181                 StartPonder();\r
182               }\r
183               p[-1] = '\n'; *p = 0; // strip off ponder move\r
184             } else move[moveNr][0] = 0;\r
185             Move4GUI(line+9);\r
186             printf("move %s\n", line+9); // send move to GUI\r
187             if(lastScore == 100001 && iniPos[0] != 'f') { printf("%s {mate}\n", stm == WHITE ? "1-0" : "0-1"); computer = NONE; }\r
188             stm = WHITE+BLACK - stm;\r
189         }\r
190         else if(!strcmp(command, "info")) {\r
191             int d=0, s=0, t=0, n=0;\r
192             char *pv;\r
193             if(sscanf(line+5, "string times @ %c", &dummy) == 1) { printf("# %s", line+12); continue; }\r
194             if(collect && (pv = strstr(line+5, "currmove "))) {\r
195                 if(p = strstr(line+5, "currmovenumber ")) {\r
196                     n = atoi(p+15);\r
197                     if(collect == 1 && n != 1) continue; // wait for move 1\r
198                     if(collect + (n == 1) > 2) { // done collecting\r
199                         if(inex && collect == 2) printf("%d 0 0 0 OK to exclude\n", lastDepth);\r
200                         collect = 3; continue;\r
201                     }\r
202                     collect = 2; on[nr=n] = 1; sscanf(pv+9, "%s", moveMap[n]); continue; // store move\r
203                 }\r
204             }\r
205             if(!post) continue;\r
206             if(sscanf(line+5, "string %c", &dummy) == 1) printf("%d 0 0 0 %s", lastDepth, line+12); else {\r
207                 if(p = strstr(line+4, " depth "))      sscanf(p+7, "%d", &d), statDepth = d;\r
208                 if(p = strstr(line+4, " score cp "))   sscanf(p+10, "%d", &s), statScore = s; else\r
209                 if(p = strstr(line+4, " score mate ")) sscanf(p+12, "%d", &s), s += s>0 ? 100000 : -100000, statScore = s; else\r
210                 if(p = strstr(line+4, " score "))      sscanf(p+7, "%d", &s), statScore = s;\r
211                 if(p = strstr(line+4, " nodes "))      sscanf(p+7, "%d", &n), statNodes = n;\r
212                 if(p = strstr(line+4, " time "))       sscanf(p+6, "%d", &t), t /= 10, statTime = t;\r
213                 if(p = strstr(line+4, " currmove "))   sscanf(p+10,"%s", currMove);\r
214                 if(p = strstr(line+4, " currmovenumber ")) sscanf(p+16,"%d", &currNr);\r
215                 if(pv = strstr(line+4, " pv ")) // convert PV info to WB thinking output\r
216                     printf("%3d  %6d %6d %10d %s", lastDepth=d, lastScore=s, t, n, Convert(pv+4));\r
217             }\r
218         }\r
219         else if(!strcmp(command, "option")) { // USI option: extract data fields\r
220             char name[80], type[80], buf[1024], val[256], *q;\r
221             int min=0, max=1e9;\r
222             if(p = strstr(line+6, " type ")) sscanf(p+1, "type %s", type), *p = '\n';\r
223             if(p = strstr(line+6, " min "))  sscanf(p+1, "min %d", &min), *p = '\n';\r
224             if(p = strstr(line+6, " max "))  sscanf(p+1, "max %d", &max), *p = '\n';\r
225             if(p = strstr(line+6, " default "))  sscanf(p+1, "default %[^\n]*", val), *p = '\n';\r
226             if(p = strstr(line+6, " name ")) sscanf(p+1, "name %[^\n]*", name);\r
227             if(!strcmp(name, "Hash") || !strcmp(name, "USI_Hash")) {\r
228                 memory = oldMem = atoi(val); hasHash = 1; \r
229                 strcpy(hashOpt, name);\r
230                 continue;\r
231             }\r
232             // pass on engine-defined option as WB option feature\r
233             if(!strcmp(type, "filename")) type[4] = 0;\r
234             sprintf(buf, "feature option=\"%s -%s", name, type); q = buf + strlen(buf);\r
235             if(     !strcmp(type, "file")\r
236                  || !strcmp(type, "string")) sprintf(q, " %s\"\n", val);\r
237             else if(!strcmp(type, "spin"))   sprintf(q, " %d %d %d\"\n", atoi(val), min, max);\r
238             else if(!strcmp(type, "check"))  sprintf(q, " %d\"\n", strcmp(val, "true") ? 0 : 1), strcat(checkOptions, name);\r
239             else if(!strcmp(type, "button")) sprintf(q, "\"\n");\r
240             else if(!strcmp(type, "combo")) {\r
241                 if(p = strstr(line+6, " default "))  sscanf(p+1, "default %s", type); // current setting\r
242                 min = 0; p = line+6;\r
243                 while(p = strstr(p, " var ")) {\r
244                     sscanf(p += 5, "%s", val); // next choice\r
245                     sprintf(buf + strlen(buf), "%s%s%s", min++ ? " /// " : " ", strcmp(type, val) ? "" : "*", val);\r
246                 }\r
247                 strcat(q, "\"\n");\r
248             }\r
249             else buf[0] = 0; // ignore unrecognized option types\r
250             if(buf[0]) printf("%s", buf);\r
251         }\r
252         else if(!strcmp(command, "id")) {\r
253             char name[256];\r
254             if(sscanf(line, "id name %[^\n]", name) == 1) printf("feature myname=\"%s (U%cI2WB)\"\n", name, sc-32);\r
255         }\r
256         else if(!strcmp(command, "readyok")) { pause = 0; Sync(WAKEUP); } // resume processing of GUI commands\r
257         else if(sscanf(command, "u%ciok", &c)==1 && c==sc)   printf("feature smp=1 memory=%d done=1\n", hasHash), Sync(WAKEUP); // done with options\r
258     }\r
259 }\r
260 \r
261 void\r
262 Move4Engine(char *m)\r
263 {\r
264     if(sc == 's') {\r
265       // convert input move to USI format\r
266       if(m[1] == '@') { // drop\r
267         m[1] = '*';\r
268       } else {\r
269         m[0] = 'a'+'0'+size - m[0];\r
270         m[1] = 'a'+'0'+size - m[1];\r
271       }\r
272       m[2] = 'a'+'0'+size - m[2];\r
273       m[3] = 'a'+'0'+size - m[3];\r
274       if(m[4] == '=') m[4] = 0; // no '=' in USI format!\r
275       else if(m[4]) m[4] = '+'; // cater to WB 4.4 bug :-(\r
276     }\r
277 }\r
278 \r
279 void\r
280 GUI2Engine()\r
281 {\r
282     char line[256], command[256], *p, *q, *r;\r
283 \r
284     while(1) {\r
285         int i, x;\r
286 \r
287         if((computer == stm || computer == ANALYZE) && !suspended) {\r
288             DPRINT("# start search\n");\r
289             LoadPos(moveNr); // load position\r
290             // and set engine thinking (note USI swaps colors!)\r
291             startTime = GetTickCount();\r
292             if(computer == ANALYZE) {\r
293                 fprintf(toE, "\ngo infinite"); DPRINT("\n# go infinite");\r
294                 if(sm & 1) { // some moves are disabled\r
295                     fprintf(toE, " searchmoves"); DPRINT(" searchmoves");\r
296                     for(i=1; i<nr; i++) if(on[i]) { fprintf(toE, " %s", moveMap[i]); DPRINT(" %s", moveMap[i]); }\r
297                 }\r
298                 fprintf(toE, "\n"); DPRINT("\n");\r
299             // code for searchmoves goes here\r
300             } else StartSearch("");\r
301         }\r
302       nomove:\r
303         fflush(toE); fflush(stdout);\r
304         i = 0; while((x = getchar()) != EOF && (line[i] = x) != '\n') i++;\r
305         line[++i] = 0; if(x == EOF) { printf("# EOF\n"); exit(-1); }\r
306         sscanf(line, "%s", command);\r
307         if(!strcmp(command, "new")) {\r
308             computer = BLACK; moveNr = 0; depth = -1; move[0][0] = 0;\r
309             stm = WHITE; strcpy(iniPos, "position startpos");\r
310             if(memory != oldMem && hasHash) fprintf(toE, "setoption name %s value %d\n", hashOpt, memory);\r
311             oldMem = memory;\r
312             // we can set other options here\r
313             pause = 1; // wait for option settings to take effect\r
314             fprintf(toE, "isready\n");\r
315             fprintf(toE, "u%cinewgame\n", sc); fflush(toE);\r
316             Sync(PAUSE); // wait for readyok\r
317         }\r
318         else if(!strcmp(command, "usermove")) {\r
319             sscanf(line, "usermove %s", command); // strips off linefeed\r
320             Move4Engine(command);\r
321             stm = WHITE+BLACK - stm; collect = (computer == ANALYZE); sm = 0;\r
322             // when pondering we either continue the ponder search as normal search, or abort it\r
323             if(pondering || computer == ANALYZE) {\r
324                 if(pondering && !strcmp(command, move[moveNr])) { // ponder hit\r
325                     pondering = 0; moveNr++; startTime = GetTickCount(); // clock starts running now\r
326                     fprintf(toE, "ponderhit\n"); DPRINT("# ponderhit\n");\r
327                     goto nomove;\r
328                 }\r
329                 StopPonder(1);\r
330             }\r
331             strcpy(move[moveNr++], command); // possibly overwrites ponder move\r
332         }\r
333         else if(!strcmp(command, "level")) {\r
334             int sec = 0;\r
335             sscanf(line, "level %d %d:%d %d", &mps, &tc, &sec, &inc) == 4 ||\r
336             sscanf(line, "level %d %d %d", &mps, &tc, &inc);\r
337             tc = (60*tc + sec)*1000; inc *= 1000; sTime = 0;\r
338         }\r
339         else if(!strcmp(command, "option")) {\r
340             char name[80], *p;\r
341             if(sscanf(line+7, "UCI2WB debug output=%d", &debug) == 1) ; else\r
342             if(p = strchr(line, '=')) {\r
343                 *p++ = 0;\r
344                 if(strstr(checkOptions, line+7)) sprintf(p, "%s\n", atoi(p) ? "true" : "false");\r
345                 fprintf(toE, "setoption name %s value %s", line+7, p); DPRINT("# setoption name %s value %s", line+7, p);\r
346             } else { fprintf(toE, "setoption name %s\n", line+7); DPRINT("# setoption name %s\n", line+7); }\r
347         }\r
348         else if(!strcmp(command, "protover")) {\r
349             if(!variants) variants = sc=='s' ? "shogi,5x5+5_shogi" : VARIANTS;\r
350             printf("feature variants=\"%s\" setboard=1 usermove=1 debug=1 ping=1 reuse=0 exclude=1 pause=1 done=0\n", variants);\r
351             printf("feature option=\"UCI2WB debug output -check %d\"\n", debug);\r
352             fprintf(toE, "u%ci\n", sc); fflush(toE); // this prompts UCI engine for options\r
353             Sync(PAUSE); // wait for uciok\r
354         }\r
355         else if(!strcmp(command, "setboard")) {\r
356                 if(strstr(line+9, " b ")) stm = BLACK;\r
357                 if(p = strchr(line+9, '[')) { char c;\r
358                     *p++ = 0; q = strchr(p, ']'); *q = 0; r = q + 4; \r
359                     if(sc == 's') q[2] = 'w' + 'b' - q[2], strcpy(r=q+3, " 1\n"); // Shogi: reverse color\r
360                     else r = strchr(strchr(q+4, ' ') + 1, ' '); // skip to second space (after e.p. square)\r
361                     *r = 0; sprintf(command, "%s%s %s %s", line+9, q+1, p, r+1);\r
362                 } else strcpy(command, line+9);\r
363                 sprintf(iniPos, "%s%sfen %s", iniPos[0]=='p' ? "position " : "", sc=='s' ? "s" : "", command);\r
364                 iniPos[strlen(iniPos)-1] = sm = 0; collect = (computer == ANALYZE);\r
365         }\r
366         else if(!strcmp(command, "variant")) {\r
367                 if(!strcmp(line+8, "shogi\n")) size = 9, strcpy(iniPos, "position startpos");\r
368                 if(!strcmp(line+8, "5x5+5_shogi\n")) size = 5, strcpy(iniPos, "position startpos");\r
369                 if(!strcmp(line+8, "xiangqi\n")) strcpy(iniPos, "fen rnbakabnr/9/1c5c1/p1p1p1p1p/9/9/P1P1P1P1P/1C5C1/9/RNBAKABNR r");\r
370         }\r
371         else if(!strcmp(command, "undo") && (i=1) || !strcmp(command, "remove") && (i=2)) {\r
372             if(pondering || computer == ANALYZE) StopPonder(1);\r
373             moveNr = moveNr > i ? moveNr - i : 0; collect = (computer == ANALYZE); sm = 0;\r
374         }\r
375         else if(!strcmp(command, ".")) {\r
376             printf("stat01: %d %d %d %d 100 %s\n", statTime, statNodes, statDepth, 100-currNr, currMove);\r
377             goto nomove;\r
378         }\r
379         else if(!strcmp(command+2, "clude") && collect > 2) { // include or exclude\r
380             int all = !strcmp(line+8, "all"), in = command[1] == 'n';\r
381             inex = 1; line[strlen(line)-1] = sm = 0; // strip LF and clear sm flag\r
382             for(i=1; i<nr; i++) { if(!strcmp(line+8, moveMap[i]) || all) on[i] = in; sm |= on[i]+1; } // sm: 2 = enabled, 1 = disabled\r
383             if(!(sm & 2)) goto nomove; // no moves enabled; continue current search\r
384             if(computer == ANALYZE) StopPonder(1); // abort old analysis\r
385         }\r
386         else if(!strcmp(command, "pause")) {\r
387             if(computer == stm) myTime -= GetTickCount() - startTime;\r
388             suspended = 1 + pondering, StopPonder(1); // remember if we were pondering, and stop search ignoring bestmove\r
389         }\r
390         else if(!strcmp(command, "resume")) {\r
391             if(suspended == 2) StartPonder(); // restart interrupted ponder search\r
392             suspended = 0; // causes thinking to start in normal way if on move or analyzing\r
393         }\r
394         else if(!strcmp(command, "xboard")) ;\r
395         else if(!strcmp(command, "analyze"))computer = ANALYZE, collect = 1, sm = 0;\r
396         else if(!strcmp(command, "exit"))   computer = NONE, StopPonder(1);\r
397         else if(!strcmp(command, "force"))  computer = NONE, StopPonder(pondering);\r
398         else if(!strcmp(command, "go"))     computer = stm;\r
399         else if(!strcmp(command, "time"))   sscanf(line+4, "%d", &myTime),  myTime  *= 10;\r
400         else if(!strcmp(command, "otim"))   sscanf(line+4, "%d", &hisTime), hisTime *= 10;\r
401         else if(!strcmp(command, "post"))   post = 1;\r
402         else if(!strcmp(command, "nopost")) post = 0;\r
403         else if(!strcmp(command, "easy"))   ponder = 0, StopPonder(pondering);\r
404         else if(!strcmp(command, "hard"))   ponder = 1, StartPonder();\r
405         else if(!strcmp(command, "ping"))   pause = 1, fprintf(toE, "isready\n"), fflush(toE), Sync(PAUSE), printf("pong %s", line+5);\r
406         else if(!strcmp(command, "memory")) sscanf(line, "memory %d", &memory);\r
407         else if(!strcmp(command, "cores"))  sscanf(line, "cores %d", &cores);\r
408         else if(!strcmp(command, "sd"))     sscanf(line, "sd %d", &depth);\r
409         else if(!strcmp(command, "st"))     sscanf(line, "st %d", &sTime), sTime = 1000*sTime - 30, inc = 0;\r
410         else if(!strcmp(command, "quit"))   fprintf(toE, "quit\n"), fflush(toE), exit(0);\r
411     }\r
412 }\r
413 \r
414 int\r
415 StartEngine(char *cmdLine, char *dir)\r
416 {\r
417 #ifdef WIN32\r
418   HANDLE hChildStdinRd, hChildStdinWr,\r
419     hChildStdoutRd, hChildStdoutWr;\r
420   BOOL fSuccess;\r
421   PROCESS_INFORMATION piProcInfo;\r
422   STARTUPINFO siStartInfo;\r
423   DWORD err;\r
424 \r
425   /* Create a pipe for the child's STDOUT. */\r
426   if (! WinPipe(&hChildStdoutRd, &hChildStdoutWr)) return GetLastError();\r
427 \r
428   /* Create a pipe for the child's STDIN. */\r
429   if (! WinPipe(&hChildStdinRd, &hChildStdinWr)) return GetLastError();\r
430 \r
431   SetCurrentDirectory(dir); // go to engine directory\r
432 \r
433   /* Now create the child process. */\r
434   siStartInfo.cb = sizeof(STARTUPINFO);\r
435   siStartInfo.lpReserved = NULL;\r
436   siStartInfo.lpDesktop = NULL;\r
437   siStartInfo.lpTitle = NULL;\r
438   siStartInfo.dwFlags = STARTF_USESTDHANDLES;\r
439   siStartInfo.cbReserved2 = 0;\r
440   siStartInfo.lpReserved2 = NULL;\r
441   siStartInfo.hStdInput = hChildStdinRd;\r
442   siStartInfo.hStdOutput = hChildStdoutWr;\r
443   siStartInfo.hStdError = hChildStdoutWr;\r
444 \r
445   fSuccess = CreateProcess(NULL,\r
446                            cmdLine,        /* command line */\r
447                            NULL,           /* process security attributes */\r
448                            NULL,           /* primary thread security attrs */\r
449                            TRUE,           /* handles are inherited */\r
450                            DETACHED_PROCESS|CREATE_NEW_PROCESS_GROUP,\r
451                            NULL,           /* use parent's environment */\r
452                            NULL,\r
453                            &siStartInfo, /* STARTUPINFO pointer */\r
454                            &piProcInfo); /* receives PROCESS_INFORMATION */\r
455 \r
456   if (! fSuccess) return GetLastError();\r
457 \r
458 //  if (0) { // in the future we could trigger this by an argument\r
459 //    SetPriorityClass(piProcInfo.hProcess, GetWin32Priority(appData.niceEngines));\r
460 //  }\r
461 \r
462   /* Close the handles we don't need in the parent */\r
463   CloseHandle(piProcInfo.hThread);\r
464   CloseHandle(hChildStdinRd);\r
465   CloseHandle(hChildStdoutWr);\r
466 \r
467   process = piProcInfo.hProcess;\r
468   pid = piProcInfo.dwProcessId;\r
469   fromE = (FILE*) _fdopen( _open_osfhandle((long)hChildStdoutRd, _O_TEXT|_O_RDONLY), "r");\r
470   toE   = (FILE*) _fdopen( _open_osfhandle((long)hChildStdinWr, _O_WRONLY), "w");\r
471 #else\r
472     char *argv[10], *p, buf[200];\r
473     int i, toEngine[2], fromEngine[2];\r
474 \r
475     if (dir && dir[0] && chdir(dir)) { perror(dir); exit(1); }\r
476     pipe(toEngine); pipe(fromEngine); // create two pipes\r
477 \r
478     if ((pid = fork()) == 0) { // Child\r
479         dup2(toEngine[0], 0);   close(toEngine[0]);   close(toEngine[1]);   // stdin from toE pipe\r
480         dup2(fromEngine[1], 1); close(fromEngine[0]); close(fromEngine[1]); // stdout into fromE pipe\r
481         dup2(1, fileno(stderr)); // stderr into frome pipe\r
482 \r
483         strcpy(buf, cmdLine); p = buf;\r
484         for (i=0;;) { argv[i++] = p; p = strchr(p, ' '); if (p == NULL) break; *p++ = 0; }\r
485         argv[i] = NULL;\r
486         execvp(argv[0], argv); // startup engine\r
487         \r
488         perror(argv[0]); exit(1); // could not start engine; quit.\r
489     }\r
490     signal(SIGPIPE, SIG_IGN);\r
491     close(toEngine[0]); close(fromEngine[1]); // close engine ends of pipes in adapter\r
492     \r
493     fromE = (FILE*) fdopen(fromEngine[0], "r"); // make into high-level I/O\r
494     toE   = (FILE*) fdopen(toEngine[1], "w");\r
495 #endif\r
496   return NO_ERROR;\r
497 }\r
498 \r
499 main(int argc, char **argv)\r
500 {\r
501         char *dir = NULL, *p, *q; int e;\r
502 \r
503         if(argc == 2 && !strcmp(argv[1], "-v")) { printf("UCI2WB " VERSION " by H.G.Muller\n"); exit(0); }\r
504         if(argc > 1 && !strcmp(argv[1], "debug")) { debug = 1; argc--; argv++; }\r
505         if(argc > 1 && !strcmp(argv[1], "-var")) { variants = argv[2]; argc-=2; argv+=2; }\r
506         if(argc > 1 && argv[1][0] == '-') { sc = argv[1][1]; argc--; argv++; }\r
507         if(argc < 2) { printf("usage is: U%cI2WB [debug] [-s] <engine.exe> [<engine directory>]\n", sc-32); exit(-1); }\r
508         if(argc > 2) dir = argv[2];\r
509         if(argc > 3) suffix = argv[3];\r
510 \r
511         // spawn engine proc\r
512         if(StartEngine(argv[1], dir) != NO_ERROR) { perror(argv[1]), exit(-1); }\r
513 \r
514         Sync(INIT);\r
515 \r
516         // create separate thread to handle engine->GUI traffic\r
517 #ifdef WIN32\r
518         CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) Engine2GUI, (LPVOID) NULL, 0, &thread_id);\r
519 #else\r
520         { pthread_t t; signal(SIGINT, SIG_IGN); pthread_create(&t, NULL, Engine2GUI, NULL); }\r
521 #endif\r
522 \r
523         // handle GUI->engine traffic in original thread\r
524         GUI2Engine();\r
525 }\r