Simple-Go

Simple-Go/simple-gtp.c
simple-gtp.c
⎇
Raw
1#include "simple-gtp.h"
2
3char* cmd_error(const char* msg)
4{
5 char* ret = malloc(strlen("? ") + strlen(msg) + 1);
6 strcpy(ret, "? ");
7 strcat(ret, msg);
8 return ret;
9}
10
11char* cmd_success(const char* msg)
12{
13 char* ret = malloc(strlen("= ") + strlen(msg) + 2);
14 strcpy(ret, "= ");
15 strcat(ret, msg);
16 return ret;
17}
18
19const char* const known_commands_string = "protocol_version\nname\nversion\nknown_command\nlist_commands\nquit\nboardsize\nclear_board\nkomi\nplay\ngenmove\nshowboard";
20const char* known_commands_array[] = {"protocol_version","name","version","known_command","list_commands","quit","boardsize","clear_board","komi","play","genmove","showboard"};
21int cmd_count = sizeof(known_commands_array)/sizeof(known_commands_array[0]);
22
23char* handle_gtp_cmd(const char* msg, game_state* game)
24{
25 if(!msg || !strlen(msg))
26 {
27 char* ret = calloc(1,1);
28 return ret;
29 }
30
31 char* command = malloc(strlen(msg) + 1);
32 sscanf(msg, "%s", command);
33
34 Vector* arguments = new_vector();
35
36 char* tmp_msg = malloc(strlen(msg)+1);
37 strcpy(tmp_msg, msg);
38
39
40 char* current = strtok(tmp_msg, " ");
41 char* tmp_arg;
42
43 while(current)
44 {
45 tmp_arg = malloc(strlen(current)+1);
46 strcpy(tmp_arg, current);
47 vector_push(arguments, tmp_arg);
48 current = strtok(NULL, " ");
49 }
50
51 char* (*func_ptr)(const char*) = cmd_error;
52 const char* func_args = "unknown command";
53
54 if(strcmp(command, "protocol_version") == 0)
55 {
56 func_ptr = cmd_success;
57 func_args = "2";
58 } else if(strcmp(command, "name") == 0) {
59 func_ptr = cmd_success;
60 func_args = "simple-go";
61 } else if(strcmp(command, "version") == 0) {
62 func_ptr = cmd_success;
63 func_args = "0.1";
64 } else if(strcmp(command, "known_command") == 0) {
65 func_ptr = cmd_success;
66 func_args = "false";
67 for(int i = 0; i < cmd_count; i++)
68 {
69 if(arguments->length > 0 && strcmp(vector_at(arguments, 0), known_commands_array[i]) == 0)
70 {
71 func_args = "true";
72 break;
73 }
74 }
75 } else if(strcmp(command, "list_commands") == 0) {
76 func_ptr = cmd_success;
77 func_args = known_commands_string;
78 } else if(strcmp(command, "boardsize") == 0) {
79 unsigned int size;
80 if(arguments->length == 1 && sscanf(vector_at(arguments,0), "%u", &size) > 0 && 0 < size && size < 26)
81 {
82 func_ptr = cmd_success;
83 func_args = "";
84 delete_board(game->board);
85 game->board = create_board(size);
86 } else {
87 func_ptr = cmd_error;
88 func_args = "unacceptable size";
89 }
90 } else if(strcmp(command, "komi") == 0) {
91 float komi;
92 if(arguments->length == 1 && sscanf(vector_at(arguments,0), "%f", &komi))
93 {
94 func_ptr = cmd_success;
95 func_args = "";
96 game->komi = komi;
97 } else {
98 func_ptr = cmd_error;
99 func_args = "komi not a float";
100 }
101 } else if(strcmp(command, "clear_board") == 0) {
102 unsigned int size = game->board->size;
103 delete_board(game->board);
104 game->board = create_board(size);
105 } else if(strcmp(command, "play") == 0) {
106 char* color = malloc(6);
107 char y;
108 int x;
109 if((arguments->length == 2 || arguments->length == 3) &&
110 (snprintf(color, 6, "%s", (char*)vector_at(arguments,0)) > 0) &&
111 (strcmp(color, "white") == 0 || strcmp(color, "black") == 0) &&
112 (sscanf(vector_at(arguments,1), "%c", &y) > 0) &&
113 ((unsigned char)(tolower(y) - 'a') < game->board->size) &&
114 (sscanf((char*)vector_at(arguments,1)+1, "%d", &x) > 0) &&
115 ((unsigned int)x <= game->board->size) && (unsigned int)x > 0)
116 {
117 if(play_at(game, (unsigned char)(y-'a'), (unsigned int)x-1))
118 {
119 func_ptr = cmd_success;
120 func_args = "";
121 } else {
122 func_ptr = cmd_error;
123 func_args = "illegal move";
124 }
125 } else {
126 func_ptr = cmd_error;
127 func_args = "invalid color or coordinate";
128 }
129 free(color);
130 } else if(strcmp(command, "genmove") == 0) {
131 unsigned int y;
132 unsigned int x;
133 do
134 {
135 y = (unsigned int)rand() % game->board->size;
136 x = (unsigned int)rand() % game->board->size;
137 } while(!play_at(game, y, x));
138
139 }
140
141 free(command);
142 delete_vector(arguments, NULL);
143 free(arguments);
144 free(tmp_msg);
145
146 return func_ptr(func_args);
147}
148