f8adcc88c2d1dabfbc476a24b43ec611f97e00f5
[uci2wb.git] / UCI2WB.c
1 /****************************************************************************/\r
2 /*                           UCI2WB by H.G.Muller                           */\r
3 /*                                                                          */\r
4 /* UCI2WB is an adapter to run engines that communicate in various dialects */\r
5 /* of the Universal Chess Interface in a GUI that supports XBoard protocol  */\r
6 /* (CECP). It supports UCI (when used for Xiangqi: the 'Cyclone dialect'),  */\r
7 /* as well as USI and UCCI when used with the flags -s or -x, respectively. */\r
8 /* This version of UCI2WB is released under the GNU General Public License, */\r
9 /* of which you should have received a copy together with this file.        */\r
10 /****************************************************************************/\r
11 \r
12 #define VERSION "4.1"\r
13 \r
14 #include <stdio.h>\r
15 #include <stdlib.h>\r
16 #ifdef WIN32\r
17 #  include <windows.h>\r
18 #  include <io.h>\r
19    HANDLE process;\r
20    DWORD thread_id;\r
21    void Bury(int s) { if(WaitForSingleObject(process, 1000*s+50) != WAIT_OBJECT_0) TerminateProcess(process, 0); }\r
22 #else\r
23 #  include <pthread.h>\r
24 #  include <signal.h>\r
25 #  include <unistd.h>\r
26 #  define NO_ERROR 0\r
27 #  include <sys/time.h>\r
28 #  include <sys/wait.h>\r
29    int GetTickCount() // with thanks to Tord\r
30    { struct timeval t; gettimeofday(&t, NULL); return t.tv_sec*1000 + t.tv_usec/1000; }\r
31 //#  include <unistd.h>\r
32    int Sleep(int msec) { return usleep(1000*msec); }\r
33    int pid; void Bury(int msec) { Sleep(msec+50); if(waitpid(-1, NULL, WNOHANG) <= 0) kill(pid, SIGKILL); }\r
34 #endif\r
35 #include <fcntl.h>\r
36 #include <string.h>\r
37 #include <ctype.h>\r
38 \r
39 // Set VARIANTS for in WinBoard variant feature. (With -s option this will always be reset to use "shogi".)\r
40 #define VARIANTS ",normal,xiangqi"\r
41 #define STDVARS "chess,chess960,crazyhouse,3check,giveaway,suicide,losers,atomic,seirawan,shogi,xiangqi"\r
42 #define EGT ",gaviotaTbPath,syzygyPath,nalimovPath,robbotripleBaseDirectory,robbototalBaseDirectory,bitbases path,"\r
43 #define XQPOS "position fen rnbakabnr/9/1c5c1/p1p1p1p1p/9/9/P1P1P1P1P/1C5C1/9/RNBAKABNR"\r
44 \r
45 #define DPRINT if(debug) printf\r
46 #define EPRINT(X) { char f[999]; sprintf X; DPRINT("%s", f); fprintf(toE, "%s", f + 2*(*f == '#')); /* strip optional # prefix */ }\r
47 \r
48 #define WHITE 0\r
49 #define BLACK 1\r
50 #define NONE  2\r
51 #define ANALYZE 3\r
52 \r
53 char move[2000][10], iniPos[256], hashOpt[20], suspended, ponder, post, hasHash, c, sc=' ', sc2, suffix[81], varOpt, searching, *binary;\r
54 int mps, tc, inc, sTime, depth, myTime, hisTime, stm, computer = NONE, memory, oldMem=0, cores, moveNr, lastDepth, lastScore, startTime, debug, flob;\r
55 int statDepth, statScore, statNodes, statTime, currNr, size, collect, nr, sm, inex, on[500], frc, byo = -1, namOpt, comp;\r
56 char currMove[20], moveMap[500][10], /* for analyze mode */ canPonder[20], threadOpt[20], varList[8000], anaOpt[20], checkOptions[8192] = "Ponder";\r
57 char pvs[99][999], board[100];  // XQ board for UCCI\r
58 char *nameWord = "name ", *valueWord = "value ", *wTime = "w", *bTime = "b", *wInc = "winc", *bInc = "binc", newGame; // keywords that differ in UCCI\r
59 int unit = 1, drawOffer, scores[99], mpvSP, maxDepth, ponderAlways, newCnt, priority, killDelay;\r
60 \r
61 FILE *toE, *fromE, *fromF;\r
62 \r
63 char *strcasestr (char *p, char *q) { while(*p) { char *r=p++, *s=q; while(tolower(*r++) == tolower(*s) && *s) s++; if(!*s) return p-1; } return NULL; }\r
64 \r
65 #ifdef WIN32\r
66 WinPipe(HANDLE *hRd, HANDLE *hWr)\r
67 {\r
68   SECURITY_ATTRIBUTES saAttr;\r
69 \r
70   /* Set the bInheritHandle flag so pipe handles are inherited. */\r
71   saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);\r
72   saAttr.bInheritHandle = TRUE;\r
73   saAttr.lpSecurityDescriptor = NULL;\r
74 \r
75   /* Create a pipe */\r
76   return CreatePipe(hRd, hWr, &saAttr, 0);\r
77 }\r
78 #endif\r
79 \r
80 #define INIT 0\r
81 #define WAKEUP 1\r
82 #define PAUSE 2\r
83 \r
84 void\r
85 Sync (int action)\r
86 {\r
87 #ifdef WIN32\r
88         static HANDLE hWr, hRd; DWORD d; char c;\r
89         switch(action) {\r
90             case INIT:   WinPipe(&hRd, &hWr); break;\r
91             case WAKEUP: WriteFile(hWr, "\n", 1, &d, NULL); break;\r
92             case PAUSE:  ReadFile(hRd, &c, 1, &d, NULL);\r
93         }\r
94 #else\r
95         static int syncPipe[2], res;    char c;\r
96         switch(action) {\r
97             case INIT:   res = pipe(syncPipe); break;\r
98             case WAKEUP: res = write(syncPipe[1], "\n", 1); break;\r
99             case PAUSE:  res = read(syncPipe[0], &c, 1);\r
100         }\r
101         if(res < 0) printf("tellusererror UCI2WB: bad sync pipe\n"), exit(0);\r
102 #endif\r
103 }\r
104 \r
105 void\r
106 FromFEN(char *fen)\r
107 {       int i=0;\r
108         while(*fen) {\r
109             char c = *fen++;\r
110             if(c >= 'A') board[i++] = c; else\r
111             if(c == '/') i++; else\r
112             if(c == ' ') break; else\r
113             while(c-- > '0' && i < 99) board[i++] = 0;\r
114             if(i >= 99) break;\r
115         }\r
116 }\r
117 \r
118 char *\r
119 ToFEN(int stm)\r
120 {\r
121         int i, n=0; static char fen[200]; char *p = fen;\r
122         for(i=0; i<99; i++) {\r
123             char c = board[i];\r
124             if(c >= 'A')  { if(n) *p++ = '0' + n; n = 0;  *p++ = c; } else n ++;\r
125             if(i%10 == 8) { if(n) *p++ = '0' + n; n = -1; *p++ = '/'; }\r
126         }\r
127         sprintf(p-1, " %c - - 0 1", stm);\r
128         return fen;\r
129 }\r
130 \r
131 int\r
132 Sqr(char *m, int j)\r
133 {\r
134         int n = m[j] - 'a' + 10*('9' - m[j+1]);\r
135         if(n < 0) n = 0; else if(n > 99) n = 99; return n;\r
136 }\r
137 \r
138 int\r
139 Play(int nr)\r
140 {\r
141         int i, last = -1;\r
142         FromFEN(iniPos + 13); // FEN part\r
143         for(i=0; i<nr; i++) {\r
144             int from=Sqr(move[i], 0), to=Sqr(move[i], 2);\r
145             if(board[to] || (board[from]|32)  == 'p' && move[i][1] != move[i][3]) last = i;\r
146             board[to] = board[from]; board[from] = 0;\r
147         }\r
148         return last;\r
149 }\r
150 \r
151 void\r
152 StartSearch(char *ponder)\r
153 {       // send the 'go' command to engine. Suffix by ponder.\r
154         int x = (ponder[0] != 0);                   // during ponder stm is the opponent\r
155         int black = (stm == BLACK ^ x ^ sc == 's'); // set if our color is what the engine calls black\r
156         int nr = moveNr + x;                        // we ponder for one move ahead!\r
157         int t = (flob ? inc + myTime/40 : 1000*byo*(byo>0)); // byoyomi time \r
158         if(sc == 'x') black = 1; else drawOffer = 0;// in UCCI 'black' refers to us and 'white' to opponent\r
159         if(!x && drawOffer) ponder = " draw", drawOffer = 0; //pass draw offer only when not pondering\r
160         EPRINT((f, "# go%s %stime %d %stime %d", ponder, bTime, (black ? myTime : hisTime) - t, wTime, (!black ? myTime : hisTime) - t))\r
161         if(sTime > 0) EPRINT((f, " movetime %d", sTime)) else\r
162         if(mps) EPRINT((f, " movestogo %d", mps*(nr/(2*mps)+1)-nr/2))\r
163         if(flob || byo >= 0) sprintf(suffix, " byoyomi %d", t); // for engines running purely on byoyomi\r
164         if((inc || !mps && !sTime && sc == 'x') && !*suffix) EPRINT((f, " %s %d %s %d", wInc, inc, bInc, inc))\r
165         if(depth > 0) EPRINT((f, " depth %d", depth))\r
166         if(*suffix) EPRINT((f, suffix, inc))\r
167         EPRINT((f, "\n")); maxDepth = mpvSP = 0;\r
168 }\r
169 \r
170 void\r
171 StopSearch(int discard)\r
172 {\r
173         if(!searching) return;\r
174         if(discard) searching = 0; // this causes bestmove to be ignored\r
175         EPRINT((f, "# stop\n")) fflush(toE); // note: 'pondering' remains set until engine acknowledges 'stop' with 'bestmove'\r
176 }\r
177 \r
178 void\r
179 LoadPos(int moveNr)\r
180 {\r
181         int j, lastCapt = 0; char *pos = iniPos, buf[200], stm;\r
182         if(sc == 'x') { // UCCI: send only reversible moves\r
183             lastCapt = Play(moveNr); // find last capture (returns -1 if none!)\r
184             Play(++lastCapt);        // reconstruct board after last capture\r
185             stm = (!strstr(iniPos+13, " b ") ^ lastCapt & 1 ? 'w' :  'b');\r
186             sprintf(buf, "position fen %s", ToFEN(stm)); pos = buf; // send it as FEN (with "position" in UCCI!)\r
187         }\r
188         EPRINT((f, "# %s moves", pos))\r
189         for(j=lastCapt; j<moveNr; j++) EPRINT((f, " %s", move[j]))\r
190         EPRINT((f, "\n"))\r
191 }\r
192 \r
193 void\r
194 StartPonder(int moveNr)\r
195 {\r
196         if(!move[moveNr][0]) return; // no ponder move\r
197         LoadPos(moveNr+1);\r
198         searching = 1; lastDepth = 1;\r
199         DPRINT("# ponder on %s\n", move[moveNr]);\r
200         StartSearch(" ponder");\r
201 }\r
202 \r
203 void\r
204 Analyze(char *val)\r
205 {\r
206     if(*anaOpt) EPRINT((f, "# setoption %s%s %s%s\n", nameWord, anaOpt, valueWord, val));\r
207 }\r
208 \r
209 char *Convert(char *pv)\r
210 {   // convert Shogi coordinates to WB\r
211     char *p, *q, c;\r
212     static char buf[10000];\r
213     if(sc != 's') return pv;\r
214     p = pv; q = buf;\r
215     while(c = *p++) {\r
216         if(c >= '0' && c <= '9' || c >= 'a' && c <= 'z') *q++ = 'a'+'0'+size - c; else *q++ = c;\r
217     }\r
218     *q++ = 0;\r
219     return buf;\r
220 }\r
221 \r
222 void\r
223 Move4GUI(char *m)\r
224 {\r
225     if(sc == 's') {\r
226       // convert USI move to WB format\r
227       m[2] = 'a'+'0'+size - m[2];\r
228       m[3] = 'a'+'0'+size - m[3];\r
229       if(m[1] == '*') { // drop\r
230         m[1] = '@';\r
231       } else {\r
232         m[0] = 'a'+'0'+size - m[0];\r
233         m[1] = 'a'+'0'+size - m[1];\r
234         if((stm == WHITE ? (m[1]>'0'+size-size/3 || m[3]>'0'+size-size/3)\r
235                                 : (m[1] <= '0'+size/3 || m[3] <= '0'+size/3)) && m[4] != '+')\r
236              m[4] = '=', m[5] = 0;\r
237       }\r
238     }\r
239 }\r
240 \r
241 int\r
242 ReadLine (FILE *f, char *line, int len)\r
243 {\r
244     int x, i = 0;\r
245     while((x = fgetc(f)) != EOF && (line[i] = x) != '\n') i+=(i<len); line[++i] = 0;\r
246     return (x != EOF);\r
247 }\r
248 \r
249 void\r
250 HandleEngineOutput()\r
251 {\r
252     char line[1024], command[256]; static char egts[999];\r
253 \r
254     while(1) {\r
255         int i=0; char *p, dummy;\r
256 \r
257         fflush(stdout); fflush(toE);\r
258         if(fromF && !ReadLine(fromF, line, 1022))  fromF = 0, printf("# end fake\n");\r
259         if(!fromF && !ReadLine(fromE, line, 1022)) printf("tellusererror UCI2WB: %s died on me\n", binary), exit(0);\r
260         DPRINT("# engine said: %s", line), fflush(stdout);\r
261         if(sscanf(line, "%s", command) != 1) continue;\r
262         if(!strcmp(command, "bestmove")) {\r
263             if(searching == 1) { searching = 0; printf("%d 0 0 0 UCI violation! Engine moves during ponder\n", lastDepth+1); return; } // ignore ponder search\r
264             else if(searching != 3) { searching = 0; return; } // ponder miss or analysis result; ignore.\r
265             // move was a move to be played\r
266             if(p = strstr(line+8, " draw")) *p = 0, printf("offer draw\n"); // UCCI\r
267             if(strstr(line+9, "resign")) { printf("resign\n"); computer = NONE; }\r
268             if(strstr(line+9, "win")) { printf("%s {claim}\n", stm== WHITE ? "1-0" :"0-1"); computer = NONE; } // USI\r
269             if(strstr(line+9, "(none)") || strstr(line+9, "null") ||\r
270                strstr(line+9, "0000")) { printf("%s\n", lastScore < -99999 ? "resign" : "1/2-1/2 {stalemate}"); computer = NONE; }\r
271             sscanf(line, "bestmove %s", move[moveNr++]);\r
272             myTime -= (GetTickCount() - startTime)*1.02 - inc; // update own clock, so we can give correct wtime, btime with ponder\r
273             if(mps && ((moveNr+1)/2) % mps == 0) myTime += tc; if(sTime) myTime = sTime; // new session or move starts\r
274             stm = WHITE+BLACK - stm; searching = 0;\r
275             // first start a new ponder search, if pondering is on and we have a move to ponder on\r
276             if(p = strstr(line+9, "ponder")) {\r
277               sscanf(p+7, "%s", move[moveNr]);\r
278               if(computer != NONE && ponder) StartPonder(moveNr);\r
279               p[-1] = '\n'; *p = 0; // strip off ponder move\r
280             } else move[moveNr][0] = 0;\r
281             Move4GUI(line+9);\r
282             printf("move %s\n", line+9); // send move to GUI\r
283             if(move[moveNr][0]) printf("Hint: %s\n", move[moveNr]);\r
284             if(lastScore == 100001 && iniPos[0] != 'f') { printf("%s {mate}\n", stm == BLACK ? "1-0" : "0-1"); computer = NONE; }\r
285             fflush(stdout); return;\r
286         }\r
287         else if(!strcmp(command, "info")) {\r
288             int d=0, s=0, t=(GetTickCount() - startTime)/10, n=1;\r
289             char *pv, varName[80];\r
290             if(sscanf(line+5, "string times @ %c", &dummy) == 1) { printf("# %s", line+12); continue; }\r
291             if(sscanf(line+5, "string variant %s", varName) == 1) {\r
292                 if(!strstr(STDVARS, varName)) {\r
293                     int files = 8, ranks = 8, hand = 0; char parent[80];\r
294                     if(p = strstr(line+18, " files ")) sscanf(p+7, "%d", &files);\r
295                     if(p = strstr(line+18, " ranks ")) sscanf(p+7, "%d", &ranks);\r
296                     if(p = strstr(line+18, " pocket ")) sscanf(p+8, "%d", &hand);\r
297                     if(p = strstr(line+18, " template ")) sscanf(p+10, "%s", parent); else strcpy(parent, "fairy");\r
298                     if(p = strstr(line+18, " startpos "))\r
299                         printf("setup (-) %dx%d+%d_%s %s", files, ranks, hand, parent, p+10);\r
300                 }\r
301                 continue;\r
302             }\r
303             if(!post) continue;\r
304             if(sscanf(line+5, "string %c", &dummy) == 1) printf("%d 0 0 0 %s", lastDepth, line+12); else {\r
305                 if(p = strstr(line+4, " depth "))      sscanf(p+7, "%d", &d), statDepth = d;\r
306                 if(p = strstr(line+4, " score cp "))   sscanf(p+10, "%d", &s), statScore = s; else\r
307                 if(p = strstr(line+4, " score mate ")) sscanf(p+12, "%d", &s), s += s>0 ? 100000 : -100000, statScore = s; else\r
308                 if(p = strstr(line+4, " score "))      sscanf(p+7, "%d", &s), statScore = s;\r
309                 if(p = strstr(line+4, " nodes "))      sscanf(p+7, "%d", &n), statNodes = n;\r
310                 if(p = strstr(line+4, " time "))       sscanf(p+6, "%d", &t), t /= 10, statTime = t;\r
311                 if(pv = strstr(line+4, " pv ")) { // convert PV info to WB thinking output\r
312                   if(d > maxDepth) maxDepth = d, mpvSP = 0; else if(d < maxDepth) continue; // ignore depth regressions\r
313                   if(p = strstr(line+4, " upperbound ")) strcat(p, "?\n"); else\r
314                   if(p = strstr(line+4, " lowerbound ")) strcat(p, "!\n");\r
315                   for(i=0; i<mpvSP; i++) if(s == scores[i] && !strcmp(pvs[i], pv+4)) break; // check if duplicat\r
316                   if(i >= mpvSP) strncpy(pvs[mpvSP], pv+4, 998), scores[mpvSP++] = s,       // emit as thinking output if not\r
317                     printf("%3d  %6d %6d %10d %s", lastDepth=d, lastScore=s, t, n, Convert(pv+4));\r
318                 } else if(s == -100000) lastScore = s; // when checkmated score is valid even without PV (which might not come)\r
319             }\r
320             if(collect && (pv = strstr(line+4, " currmove "))) {\r
321                 sscanf(pv+10,"%s", currMove);\r
322                 if(p = strstr(line+4, " currmovenumber ")) {\r
323                     n = currNr = atoi(p+16);\r
324                     if(collect == 1 && n != 1) continue; // wait for move 1\r
325                     if(collect + (n == 1) > 2) { // done collecting\r
326                         if(inex && collect == 2) printf("%d 0 0 0 OK to exclude\n", lastDepth);\r
327                         collect = 3; continue;\r
328                     }\r
329                     collect = 2; on[nr=n] = 1; strcpy(moveMap[n], currMove); continue; // store move\r
330                 }\r
331             }\r
332         }\r
333         else if(!strcmp(command, "option")) { // USI option: extract data fields\r
334             char name[80], type[80], buf[1024], val[256], *q;\r
335             int min=0, max=1e9; *val = 0;\r
336             if(p = strstr(line+6, " type ")) sscanf(p+1, "type %s", type), *p = '\n';\r
337             if(p = strstr(line+6, " min "))  sscanf(p+1, "min %d", &min), *p = '\n';\r
338             if(p = strstr(line+6, " max "))  sscanf(p+1, "max %d", &max), *p = '\n';\r
339             if(p = strstr(line+6, " default "))  sscanf(p+1, "default %[^\n]*", val), *p = '\n';\r
340             if(!(p = strstr(line+6, " name "))) p = line+1; sscanf(p+6, "%[^\n]", name); // 'name' is omitted in UCCI\r
341             if(!strcasecmp(name, "UCI_Chess960")) { frc=2; continue; }\r
342             if(!strcasecmp(name, "UCI_Variant")) { if(p = strstr(line+6, " var ")) strcpy(varList, p); varOpt = 1; continue; }\r
343             if(!strcasecmp(name, "UCI_Opponent")) { namOpt = 1; continue; }\r
344             if(!strcasecmp(name+2, "I_AnalyseMode")) { strcpy(anaOpt, name); continue; }\r
345             if(frc< 0 && (strstr(name, "960") || strcasestr(name, "frc")) && !strcmp(type, "check")) {\r
346                 EPRINT((f, "# setoption name %s value true\n", name)) strcpy(val, "true"); // set non-standard suspected FRC options\r
347             }\r
348             if(!strcasecmp(name, "Threads")) { strcpy(threadOpt, name); continue; }\r
349             if(!strcasecmp(name, "Ponder") || !strcasecmp(name, "USI_Ponder")) { strcpy(canPonder, name); continue; }\r
350             if(!strcasecmp(name, "Hash") || !strcasecmp(name, "USI_Hash") || !strcasecmp(name, "hashsize")) {\r
351                 memory = oldMem = atoi(val); hasHash = 1; \r
352                 strcpy(hashOpt, name);\r
353                 continue;\r
354             }\r
355             if(!strcasecmp(name, "newgame") && !strcmp(type, "button")) { newGame++; continue; }\r
356             if(!strcasecmp(name, "usemillisec")) { unit = (!strcmp(val, "false") ? 2 : 1); continue; }\r
357             sprintf(buf, ",%s,", name); if(p = strcasestr(EGT, buf)) { // collect EGT formats\r
358                 strcpy(buf, p); for(p=buf; *++p >='a';){} if(*p == ' ') strcpy(buf, ",scorpio"); *p = 0; strcat(egts, buf); continue; // clip at first non-lower-case\r
359             }\r
360             // pass on engine-defined option as WB option feature\r
361             if(!strcmp(type, "filename")) type[4] = 0;\r
362             else if(sc == 'c' && !strcmp(type, "string")) { // in UCI try to guess which strings are file or directory names\r
363                 if(strcasestr(name, "file")) strcpy(type, "file"); else\r
364                 if(strcasestr(name, "path") || strcasestr(name, "directory") || strcasestr(name, "folder")) strcpy(type, "path");\r
365             }\r
366             sprintf(buf, "feature option=\"%s -%s", name, type); q = buf + strlen(buf);\r
367             if(     !strcmp(type, "file")\r
368                  || !strcmp(type, "string")) sprintf(q, " %s\"\n", val);\r
369             else if(!strcmp(type, "spin"))   sprintf(q, " %d %d %d\"\n", atoi(val), min, max);\r
370             else if(!strcmp(type, "check"))  sprintf(q, " %d\"\n", strcmp(val, "true") ? 0 : 1), strcat(checkOptions, name);\r
371             else if(!strcmp(type, "button")) sprintf(q, "\"\n");\r
372             else if(!strcmp(type, "combo")) {\r
373                 if(p = strstr(line+6, " default "))  sscanf(p+1, "default %s", type); // current setting\r
374                 min = 0; p = line+6;\r
375                 while(p = strstr(p, " var ")) {\r
376                     sscanf(p += 5, "%s", val); // next choice\r
377                     sprintf(buf + strlen(buf), "%s%s%s", min++ ? " /// " : " ", strcmp(type, val) ? "" : "*", val);\r
378                 }\r
379                 strcat(q, "\"\n");\r
380 \r
381             }\r
382             else buf[0] = 0; // ignore unrecognized option types\r
383             if(buf[0]) printf("%s", buf);\r
384         }\r
385         else if(!strcmp(command, "id")) {\r
386             static char name[256], version[256];\r
387             if(sscanf(line, "id name %[^\n]", name) == 1) printf("feature myname=\"%s (U%cI2WB)\"\n", name, sc-32);\r
388             if(sscanf(line, "id version %[^\n]", version) == 1 && *name) printf("feature myname=\"%s %s (U%cI2WB)\"\n", name, version, sc-32);\r
389         }\r
390         else if(!strcmp(command, "readyok")) return; // resume processing of GUI commands\r
391         else if(sc == 'x'&& !strcmp(command, "ucciok") || sscanf(command, "u%ciok", &c)==1 && c==sc) {\r
392             char *p = varList, *q = p;\r
393             while(*q && *q != '\n')  if(!strncmp(q, " var ", 5)) *p++ = ',', q +=5; // replace var keywords by commas\r
394                                 else if(!strncmp(q-1, " chess ", 7)) strcpy(p, "normal"), p += 6, q += 5; // 'chess' is called 'normal' in CECP\r
395                                 else *p++ = *q++; // copy other variant names unmodified\r
396             *p = 0;\r
397             if(frc) sprintf(p, ",normal,fischerandom"), printf("feature oocastle=%d\n", frc<0); // unannounced FRC uses O-O castling\r
398             if(!*varList) strcpy(varList, sc=='s' ? ",shogi,5x5+5_shogi" : VARIANTS); // without clue guess liberally\r
399             printf("feature variants=\"%s\"\n", varList+1); // from UCI_Variant combo and/or UCI_Chess960 check options\r
400             if(*egts) printf("feature egt=\"%s\"\n", egts+1);\r
401             printf("feature smp=1 memory=%d done=1\n", hasHash);\r
402             if(unit == 2) { unit = 1; EPRINT((f, "# setoption usemillisec true\n")) }\r
403             fflush(stdout); return; // done with options\r
404         }\r
405     }\r
406 }\r
407 \r
408 void\r
409 Move4Engine(char *m)\r
410 {\r
411     if(sc == 's') {\r
412       // convert input move to USI format\r
413       if(m[1] == '@') { // drop\r
414         m[1] = '*';\r
415       } else {\r
416         m[0] = 'a'+'0'+size - m[0];\r
417         m[1] = 'a'+'0'+size - m[1];\r
418       }\r
419       m[2] = 'a'+'0'+size - m[2];\r
420       m[3] = 'a'+'0'+size - m[3];\r
421       if(m[4] == '=') m[4] = 0; // no '=' in USI format!\r
422       else if(m[4]) m[4] = '+'; // cater to WB 4.4 bug :-(\r
423     }\r
424 }\r
425 \r
426 void DoCommand ();\r
427 char mySide;\r
428 volatile char queue[10000], *qStart, *qEnd;\r
429 \r
430 void\r
431 LaunchSearch()\r
432 {\r
433     int i;\r
434 \r
435     if(suspended || searching) return;\r
436 \r
437         if(computer == stm || computer == ANALYZE && sm != 1) {\r
438             DPRINT("# start search\n");\r
439             LoadPos(moveNr); fflush(stdout); // load position\r
440             // and set engine thinking (note USI swaps colors!)\r
441             startTime = GetTickCount(); mySide = stm; // remember side we last played for\r
442             if(computer == ANALYZE) {\r
443                 EPRINT((f, "# go infinite")); maxDepth = mpvSP = 0;\r
444                 if(sm & 1) { // some moves are disabled\r
445                     EPRINT((f, " searchmoves"))\r
446                     for(i=1; i<nr; i++) if(on[i]) EPRINT((f, " %s", moveMap[i]))\r
447                 }\r
448                 EPRINT((f, "\n")) searching = 2; // suppresses spurious commands during analysis starting new searches\r
449             } else searching = 3, StartSearch(""); // request suspending of input processing while thinking\r
450         } else if(ponderAlways && computer == NONE) move[moveNr][0] = 0, StartPonder(moveNr-1);\r
451         else if(BLACK+WHITE-stm == computer && ponder && moveNr) StartPonder(moveNr);\r
452 }\r
453 \r
454 void\r
455 GUI2Engine()\r
456 {\r
457     char line[256], command[256], *p;\r
458 \r
459     while(1) {\r
460         int difficult;\r
461 \r
462        for(difficult=0; !difficult; ) { // read and handle commands that can (or must) be handled during thinking\r
463         fflush(toE); fflush(stdout);\r
464         if(!ReadLine(stdin, line, 254)) printf("# EOF\n"), sprintf(line, "quit -1\n");\r
465         if(!sscanf(line, "%s", command)) return;\r
466         if(!strcmp(command, "usermove")) { difficult--; break; } // for efficiency during game play, moves, time & otim are tried first\r
467         else if(!strcmp(command, "time"))   sscanf(line+4, "%d", &myTime),  myTime  = (10*myTime)/unit;\r
468         else if(!strcmp(command, "otim"))   sscanf(line+4, "%d", &hisTime), hisTime = (10*hisTime)/unit;\r
469         else if(!strcmp(command, "draw"))   drawOffer = 1; // backlogged anyway, so this can be done instantly\r
470         else if(!strcmp(command, "post"))  post = 1;\r
471         else if(!strcmp(command, "nopost"))post = 0;\r
472         else if(!strcmp(command, ".")) {\r
473             printf("stat01: %d %d %d %d %d %s\n", statTime, statNodes, statDepth, nr-currNr, nr, currMove);\r
474         }\r
475         else if(!strcmp(command, "pause")) {\r
476             if(computer == stm) myTime -= GetTickCount() - startTime;\r
477             suspended = 1 + (searching == 1); // remember if we were pondering, and stop search ignoring bestmove\r
478             StopSearch(1);\r
479         }\r
480         else if(!strcmp(command, "xboard")) ;\r
481         else if(!strcmp(command, "random")) ;\r
482         else if(!strcmp(command, "accepted")) ;\r
483         else if(!strcmp(command, "rejected")) ;\r
484         else if(!strcmp(command, "book")) ;\r
485         else if(!strcmp(command, "ics")) ;\r
486         else if(!strcmp(command, "hint")) ;\r
487         else if(!strcmp(command, "computer")) comp = 1;\r
488         else { //convert easy & hard to "option" after treating their effect on the adapter\r
489           if(!strcmp(command, "easy")) {\r
490             if(*canPonder) ponder = 0, sprintf(command, "option"), sprintf(line, "option %s=0\n", canPonder); else continue;\r
491           }\r
492           else if(!strcmp(command, "hard")) {\r
493             if(*canPonder) ponder = 1, sprintf(command, "option"), sprintf(line, "option %s=1\n", canPonder); else continue;\r
494           }\r
495           if(!strcmp(command, "option")) {\r
496             if(sscanf(line+7, "UCI2WB debug output=%d", &debug) == 1) ; else\r
497             if(sscanf(line+7, "ponder always=%d", &ponderAlways) == 1) ; else\r
498             if(sscanf(line+7, "Floating Byoyomi=%d", &flob) == 1) ; else\r
499             if(sscanf(line+7, "Byoyomi=%d", &byo) == 1) ; else\r
500             difficult = 1;\r
501           }\r
502           else difficult = 1; // difficult command; terminate loop for easy ones\r
503         }\r
504        } // next command\r
505 \r
506         // some commands that should never come during thinking can be safely processed here\r
507         if(difficult < 0) { // used as kludge to signal "usermove" was already matched\r
508             sscanf(line, "usermove %s", command); // strips off linefeed\r
509             Move4Engine(command);\r
510             collect = (computer == ANALYZE); sm = 0;\r
511             // when pondering we either continue the ponder search as normal search, or abort it\r
512             if(searching == 1 && !strcmp(command, move[moveNr])) { // ponder hit\r
513                 char *draw = drawOffer ? " draw" : ""; drawOffer = 0;\r
514                 stm = WHITE+BLACK - stm;         // for acceptance of ponder move (can be safely done out of sync)\r
515                 searching = 3; moveNr++; startTime = GetTickCount(); // clock starts running now\r
516                 EPRINT((f, "# ponderhit%s\n", draw)) fflush(toE); fflush(stdout);\r
517             } else {\r
518                 if(searching) StopSearch(1);     // ponder miss or analysis, as moves won't arrive during thinking\r
519                 p = line+7; while(qEnd < queue+10000 && (*qEnd++ = *p++) != '\n') {}\r
520                 Sync(WAKEUP);                    // queue move for adding it to game (and toggle stm)\r
521             }\r
522         } else\r
523         if(!strcmp(command, "resume")) {\r
524             if(suspended == 2) StartPonder(moveNr); // restart interrupted ponder search\r
525             suspended = 0;  *qEnd++ = '\n'; Sync(WAKEUP); // causes search to start in normal way if on move or analyzing\r
526         } else\r
527       {\r
528         DPRINT("# queue '%s', searching=%d\n", command, searching);\r
529         if(searching == 3) { // command arrived during thinking; order abort for 'instant commands'\r
530             if(!strcmp(command, "?") || !strcmp(command, "quit") ||\r
531                !strcmp(command, "force") || !strcmp(command, "result")) StopSearch(0);\r
532         } else StopSearch(1); // always abort pondering or analysis\r
533 \r
534         // queue command for execution by engine thread\r
535         if(qStart == qEnd) qStart = qEnd = queue;\r
536         p = line; while(qEnd < queue+10000 && (*qEnd++ = *p++) != '\n') {}\r
537         Sync(WAKEUP);\r
538         // when 'stop' doesn't catch engine's attention in reasonable time, so the GUI might kill us:\r
539         if(!strcmp(command, "quit")) { Bury(killDelay); exit(0); } // kill the engine and exit\r
540       }\r
541     }\r
542 }\r
543 \r
544 \r
545 void\r
546 DoCommand ()\r
547 {\r
548     char line[1024], command[256], *p, *q, *r, type[99];\r
549     int i;\r
550 \r
551     p=line; while(qStart < qEnd && (*p++ = *qStart++) != '\n') {} *p = 0;\r
552     sscanf(line, "%s %s", command, type); DPRINT("# command %s\n", command), fflush(stdout);\r
553 \r
554         if(!strcmp(command, "new")) {\r
555             computer = BLACK; moveNr = 0; depth = -1; move[0][0] = 0;\r
556             stm = WHITE; strcpy(iniPos, "position startpos"); frc &= ~1;\r
557             if(newCnt++) return; // prevent a 2nd 'isready' due to reuse=0-violating 'new' preceding 'quit'\r
558             if(memory != oldMem && hasHash) EPRINT((f, "# setoption %s%s %s%d\n", nameWord, hashOpt, valueWord, memory))\r
559             oldMem = memory;\r
560             // we can set other options here\r
561             if(varOpt && strstr(varList, ",normal")) EPRINT((f, "# setoption name UCI_Variant value chess\n"))\r
562             EPRINT((f, "# isready\n")) fflush(toE);\r
563             HandleEngineOutput(); // wait for readyok\r
564             if(sc == 'x') { if(newGame) EPRINT((f, "# setoption newgame\n")) } else // optional in UCCI\r
565             EPRINT((f, "# u%cinewgame\n", sc)) fflush(toE);\r
566         }\r
567         else if(!strcmp(command, "e")) { strcpy(move[moveNr++], type); stm ^= WHITE|BLACK; return; }\r
568         else if(!strcmp(command, "option")) {\r
569             char *p;\r
570             if(p = strchr(line, '=')) {\r
571                 *p++ = 0;\r
572                 if(strstr(checkOptions, line+7)) sprintf(p, "%s\n", atoi(p) ? "true" : "false");\r
573                 EPRINT((f, "# setoption %s%s %s%s", nameWord, line+7, valueWord, p));\r
574             } else EPRINT((f, "# setoption %s%s\n", nameWord, line+7));\r
575         }\r
576         else if(!strcmp(command, "level")) {\r
577             int sec = 0;\r
578             if(sscanf(line, "level %d %d:%d %d", &mps, &tc, &sec, &inc) != 4)\r
579                 sscanf(line, "level %d %d %d", &mps, &tc, &inc);\r
580             tc = (60*tc + sec)*1000; inc *= 1000; sTime = 0; tc /= unit; inc /= unit;\r
581         }\r
582         else if(!strcmp(command, "protover")) {\r
583             printf("feature setboard=1 usermove=1 debug=1 ping=1 name=1 reuse=0 exclude=1 pause=1 sigint=0 sigterm=0 done=0\n");\r
584             printf("feature option=\"UCI2WB debug output -check %d\"\n", debug);\r
585             printf("feature option=\"ponder always -check %d\"\n", ponderAlways);\r
586             if(sc == 's') printf("feature option=\"Floating Byoyomi -check %d\"\nfeature option=\"Byoyomi -spin %d -1 1000\"\n", flob, byo);\r
587             EPRINT((f, sc == 'x' ? "# ucci\n" : "# u%ci\n", sc)) fflush(toE); // prompt UCI engine for options\r
588             HandleEngineOutput(); // wait for uciok\r
589         }\r
590         else if(!strcmp(command, "setboard")) {\r
591                 stm = (strstr(line+9, " b ") ? BLACK : WHITE);\r
592                 if((p = strchr(line+9, '[')) && !varOpt) {\r
593                     *p++ = 0; q = strchr(p, ']'); *q = 0; r = q + 4; \r
594                     if(sc == 's') q[2] = 'w' + 'b' - q[2], strcpy(r=q+3, " 1\n"); // Shogi: reverse color\r
595                     else r = strchr(strchr(q+4, ' ') + 1, ' '); // skip to second space (after e.p. square)\r
596                     *r = 0; sprintf(command, "%s%s %s %s", line+9, q+1, p, r+1);\r
597                 } else strcpy(command, line+9);\r
598                 if(frc == -1 && (p = strchr(command, ' '))) strncpy(p+3, "KQkq", 4); // unannounced FRC\r
599                 sprintf(iniPos, "%s%sfen %s", iniPos[0]=='p' ? "position " : "", sc=='s' ? "s" : "", command);\r
600                 iniPos[strlen(iniPos)-1] = sm = 0; collect = (computer == ANALYZE);\r
601         }\r
602         else if(!strcmp(command, "variant")) {\r
603                 if(varOpt) {\r
604                     EPRINT((f, "# setoption name UCI_Variant value %sucinewgame\nisready\n", line+8))\r
605                     fflush(toE); HandleEngineOutput(); // wait for readyok\r
606                 }\r
607                 if(!strcmp(line+8, "shogi\n")) size = 9, strcpy(iniPos, "position startpos");\r
608                 if(!strcmp(line+8, "5x5+5_shogi\n")) size = 5, strcpy(iniPos, "position startpos");\r
609                 if(!strcmp(line+8, "xiangqi\n")) sprintf(iniPos, "%s %c", XQPOS+9*(sc2 == 'c'), sc2 == 'c' ? 'r' : 'w'); // with -c use Cyclone dialect\r
610                 if(!strcmp(line+8, "fischerandom\n")) { frc |= 1; if(frc > 0) EPRINT((f, "# setoption name UCI_Chess960 value true\n")) }\r
611         }\r
612         else if(!strcmp(command, "undo") && (i=1) || !strcmp(command, "remove") && (i=2)) {\r
613             moveNr = moveNr > i ? moveNr - i : 0; collect = (computer == ANALYZE); sm = 0;\r
614         }\r
615         else if(!strcmp(command+2, "clude") && collect > 2) { // include or exclude\r
616             int all = !strcmp(line+8, "all"), in = command[1] == 'n';\r
617             inex = 1; line[strlen(line)-1] = sm = 0; // strip LF and clear sm flag\r
618             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
619         }\r
620         else if(!strcmp(command, "analyze"))computer = ANALYZE, collect = 1, sm = 0, Analyze("true");\r
621         else if(!strcmp(command, "exit"))   computer = NONE, Analyze("false");\r
622         else if(!strcmp(command, "force"))  computer = NONE;\r
623         else if(!strcmp(command, "go"))     computer = stm;\r
624         else if(!strcmp(command, "ping"))   { /* static int done; if(!done) pause = 1, fprintf(toE, "isready\n"), fflush(toE), printf("# send isready\n"), fflush(stdout), Sync(PAUSE); done = 1;*/ printf("po%s", line+2); }\r
625         else if(!strcmp(command, "memory")) sscanf(line, "memory %d", &memory);\r
626         else if(!strcmp(command, "cores")&& !!*threadOpt) { sscanf(line, "cores %d", &cores); EPRINT((f, "# setoption %s%s %s%d\n", nameWord, threadOpt, valueWord, cores)) }\r
627         else if(!strcmp(command, "egtpath")){\r
628             sscanf(line+8, "%s %[^\n]", type, command);\r
629             if(p = strstr(EGT, type)) strcpy(type, p), p = strchr(type, ','), *p = 0; else strcpy(type, "bitbases path");\r
630             EPRINT((f, "# setoption name %s value %s\n", type, command));\r
631         }\r
632         else if(!strcmp(command, "sd"))     sscanf(line, "sd %d", &depth);\r
633         else if(!strcmp(command, "st"))     sscanf(line, "st %d", &sTime), sTime = 1000*sTime - 30, inc = 0, sTime /= unit;\r
634         else if(!strcmp(command, "name"))   { if(namOpt) EPRINT((f, "# setoption name UCI_Opponent value none none %s %s", comp ? "computer" : "human", line+5)) }\r
635         else if(!strcmp(command, "result")) {\r
636             if(sc == 's') EPRINT((f, "# gameover %s\n", line[8] == '/' ? "draw" : (line[7] == '0') == mySide ? "win" : "lose"))\r
637             computer = NONE;\r
638         }\r
639         else if(!strcmp(command, "quit"))   { EPRINT((f, "# quit\n")) fflush(toE); }\r
640         else printf("Error (unknown command): %s\n", command);\r
641 \r
642         fflush(stdout);\r
643 }\r
644 \r
645 void *\r
646 Engine2GUI()\r
647 {\r
648     if(fromF = fopen("DefectiveEngineOptions.ini", "r")) printf("# fake engine input\n");\r
649     while(1) {\r
650         if(searching > 1) HandleEngineOutput();  // this could leave us (or fall through) pondering\r
651         while(qStart == qEnd && searching) HandleEngineOutput(); // relay ponder output until command arrives\r
652         Sync(PAUSE); // possibly wait for command silently if engine is idle\r
653         DoCommand(); LaunchSearch();\r
654     }\r
655 }\r
656 \r
657 int\r
658 StartEngine(char *cmdLine, char *dir)\r
659 {\r
660 #ifdef WIN32\r
661   HANDLE hChildStdinRd, hChildStdinWr,\r
662     hChildStdoutRd, hChildStdoutWr;\r
663   BOOL fSuccess;\r
664   PROCESS_INFORMATION piProcInfo;\r
665   STARTUPINFO siStartInfo;\r
666   DWORD err;\r
667 \r
668   /* Create a pipe for the child's STDOUT. */\r
669   if (! WinPipe(&hChildStdoutRd, &hChildStdoutWr)) return GetLastError();\r
670 \r
671   /* Create a pipe for the child's STDIN. */\r
672   if (! WinPipe(&hChildStdinRd, &hChildStdinWr)) return GetLastError();\r
673 \r
674   SetCurrentDirectory(dir); // go to engine directory\r
675 \r
676   /* Now create the child process. */\r
677   siStartInfo.cb = sizeof(STARTUPINFO);\r
678   siStartInfo.lpReserved = NULL;\r
679   siStartInfo.lpDesktop = NULL;\r
680   siStartInfo.lpTitle = NULL;\r
681   siStartInfo.dwFlags = STARTF_USESTDHANDLES;\r
682   siStartInfo.cbReserved2 = 0;\r
683   siStartInfo.lpReserved2 = NULL;\r
684   siStartInfo.hStdInput = hChildStdinRd;\r
685   siStartInfo.hStdOutput = hChildStdoutWr;\r
686   siStartInfo.hStdError = hChildStdoutWr;\r
687 \r
688   fSuccess = CreateProcess(NULL,\r
689                            cmdLine,        /* command line */\r
690                            NULL,           /* process security attributes */\r
691                            NULL,           /* primary thread security attrs */\r
692                            TRUE,           /* handles are inherited */\r
693                            DETACHED_PROCESS|CREATE_NEW_PROCESS_GROUP,\r
694                            NULL,           /* use parent's environment */\r
695                            NULL,\r
696                            &siStartInfo, /* STARTUPINFO pointer */\r
697                            &piProcInfo); /* receives PROCESS_INFORMATION */\r
698 \r
699   if (! fSuccess) return GetLastError();\r
700 \r
701   if (priority > 0) { // for now only implement all lowered priorityies the same way\r
702     SetPriorityClass(piProcInfo.hProcess, BELOW_NORMAL_PRIORITY_CLASS);\r
703   }\r
704 \r
705   /* Close the handles we don't need in the parent */\r
706   CloseHandle(piProcInfo.hThread);\r
707   CloseHandle(hChildStdinRd);\r
708   CloseHandle(hChildStdoutWr);\r
709 \r
710   process = piProcInfo.hProcess;\r
711   fromE = (FILE*) _fdopen( _open_osfhandle((long)hChildStdoutRd, _O_TEXT|_O_RDONLY), "r");\r
712   toE   = (FILE*) _fdopen( _open_osfhandle((long)hChildStdinWr, _O_WRONLY), "w");\r
713 #else\r
714     char *argv[10], *p, buf[200];\r
715     int i, toEngine[2], fromEngine[2];\r
716 \r
717     if (dir && dir[0] && chdir(dir)) { perror(dir); exit(1); }\r
718     i = pipe(toEngine) + pipe(fromEngine); // create two pipes\r
719     if(i < 0) printf("tellusererror UCI2WB: no engine pipe\n"), exit(0);\r
720 \r
721     if ((pid = fork()) == 0) { // Child\r
722         dup2(toEngine[0], 0);   close(toEngine[0]);   close(toEngine[1]);   // stdin from toE pipe\r
723         dup2(fromEngine[1], 1); close(fromEngine[0]); close(fromEngine[1]); // stdout into fromE pipe\r
724         dup2(1, fileno(stderr)); // stderr into frome pipe\r
725 \r
726         strcpy(buf, cmdLine); p = buf;\r
727         for (i=0;;) { argv[i++] = p; p = strchr(p, ' '); if (p == NULL) break; *p++ = 0; }\r
728         argv[i] = NULL;\r
729         if(priority) i = nice(priority);\r
730         execvp(argv[0], argv); // startup engine\r
731         \r
732         perror(argv[0]); exit(1); // could not start engine; quit.\r
733     }\r
734     signal(SIGPIPE, SIG_IGN);\r
735     close(toEngine[0]); close(fromEngine[1]); // close engine ends of pipes in adapter\r
736     \r
737     fromE = (FILE*) fdopen(fromEngine[0], "r"); // make into high-level I/O\r
738     toE   = (FILE*) fdopen(toEngine[1], "w");\r
739 #endif\r
740   return NO_ERROR;\r
741 }\r
742 \r
743 int\r
744 main(int argc, char **argv)\r
745 {\r
746         char *dir = NULL;\r
747 \r
748 \r
749         if(argc == 2 && !strcmp(argv[1], "-v")) { printf("UCI2WB " VERSION " by H.G.Muller\n"); exit(0); }\r
750         if(argc > 1 && !strcmp(argv[1], "debug")) { debug = 1; argc--; argv++; }\r
751         if(argc > 1 && !strcmp(argv[1], "-var")) { strcpy(varList+1, argv[2]); *varList = ','; argc-=2; argv+=2; }\r
752         if(argc > 2 && !strcmp(argv[1], "-nice")) { sscanf(argv[2], "%d", &priority); argc-=2; argv+=2; }\r
753         if(argc > 2 && !strcmp(argv[1], "-kill")) { sscanf(argv[2], "%d", &killDelay); argc-=2; argv+=2; }\r
754         if(argc > 1 && argv[1][0] == '-') { sc = argv[1][1]; argc--; argv++; }\r
755         if(argc < 2) { printf("usage is: U%cI2WB [debug] [-s] <engine.exe> [<engine directory>]\n", sc-32); exit(-1); }\r
756         if(argc > 2) dir = argv[2];\r
757         if(argc > 3) strncpy(suffix, argv[3], 80);\r
758 \r
759         if(sc == 'x') nameWord = valueWord = bTime = "", wTime = "opp", bInc = "increment", wInc = "oppincrement", unit = 1000; // switch to UCCI keywords\r
760         else if(sc == 'f' ) frc = -1, sc = 'c';   // UCI for unannounced Chess960\r
761         else if(sc != 's') sc2 = sc, sc = 'c';    // UCI for normal Chess\r
762 \r
763         // spawn engine proc\r
764         if(StartEngine(binary = argv[1], dir) != NO_ERROR) { perror(argv[1]), exit(-1); }\r
765 \r
766         Sync(INIT);\r
767 \r
768         // create separate thread to handle engine->GUI traffic\r
769 #ifdef WIN32\r
770         CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) Engine2GUI, (LPVOID) NULL, 0, &thread_id);\r
771 #else\r
772         { pthread_t t; signal(SIGINT, SIG_IGN); signal(SIGTERM, SIG_IGN); pthread_create(&t, NULL, Engine2GUI, NULL); }\r
773 #endif\r
774 \r
775         // handle GUI->engine traffic in original thread\r
776         GUI2Engine();\r
777         return 0;\r
778 }\r