2 * The Submit Daemon: High-Level Part of the Protocol
4 * (c) 2007 Martin Mares <mj@ucw.cz>
8 #include "lib/mempool.h"
9 #include "lib/simple-lists.h"
10 #include "lib/stkstring.h"
11 #include "sherlock/object.h"
12 #include "sherlock/objread.h"
18 /*** REQUESTS AND REPLIES ***/
21 read_error_cb(struct obj_read_state *st UNUSED, byte *msg)
23 client_error("Request parse error: %s", msg);
27 read_request(struct conn *c)
32 c->pool = mp_new(1024);
33 c->request = obj_new(c->pool);
34 c->reply = obj_new(c->pool);
36 struct obj_read_state st;
37 obj_read_start(&st, c->request);
38 st.error_callback = read_error_cb;
43 int l = bgets_nodie(&c->rx_fb, line, sizeof(line));
45 client_error("Request line too long");
51 client_error("Truncated request");
56 if (size >= max_request_size)
57 client_error("Request too long");
58 obj_read_attr(&st, line[0], line+1);
65 write_reply(struct conn *c)
67 if (!obj_find_attr(c->reply, '-') && !obj_find_attr(c->reply, '+'))
68 obj_set_attr(c->reply, '+', "OK");
72 if (msg = obj_find_aval(c->reply, '-'))
73 log(L_DEBUG, ">> -%s", msg);
74 else if (msg = obj_find_aval(c->reply, '+'))
75 log(L_DEBUG, ">> +%s", msg);
77 log(L_DEBUG, ">> ???");
79 obj_write(&c->tx_fb, c->reply, BUCKET_TYPE_PLAIN);
80 bputc(&c->tx_fb, '\n');
85 err(struct conn *c, byte *msg)
87 obj_set_attr(c->reply, '-', msg);
93 copy_attrs(struct odes *dest, struct odes *src)
95 for (struct oattr *a = src->attrs ; a; a=a->next)
96 for (struct oattr *aa = a; aa; aa=aa->same)
97 obj_add_attr(dest, aa->attr, aa->val);
101 cmd_status(struct conn *c)
103 uns verbose = obj_find_anum(c->request, 'V', 0);
106 CLIST_FOR_EACH(struct task *, t, task_list)
108 struct odes *to = task_status_find_task(c, t, 1);
109 struct odes *tr = obj_add_son(c->reply, 'T' + OBJ_ATTR_SON);
111 CLIST_FOR_EACH(simp_node *, p, t->parts)
113 struct odes *po = task_status_find_part(to, p->s, 1);
114 struct odes *pr = obj_add_son(tr, 'P' + OBJ_ATTR_SON);
116 uns current_ver = obj_find_anum(po, 'V', 0);
117 for (struct oattr *v = obj_find_attr(po, 'V' + OBJ_ATTR_SON); v; v=v->same)
119 struct odes *vo = v->son;
120 uns ver = obj_find_anum(vo, 'V', 0);
121 if (ver == current_ver || verbose)
122 obj_add_son_ref(pr, 'V' + OBJ_ATTR_SON, vo);
130 static struct fastbuf *
131 read_attachment(struct conn *c)
133 uns size = obj_find_anum(c->request, 'S', 0);
134 if (size > max_attachment_size)
136 err(c, "Submission too large");
139 obj_set_attr(c->reply, '+', "Go on");
141 obj_set_attr(c->reply, '+', NULL);
143 // This is less efficient than bbcopy(), but we want our own error handling.
144 struct fastbuf *fb = bopen_tmp(4096);
149 uns cnt = bread(&c->rx_fb, buf, MIN(remains, (uns)sizeof(buf)));
153 client_error("Truncated attachment");
155 bwrite(fb, buf, cnt);
163 cmd_submit(struct conn *c)
165 byte *tname = obj_find_aval(c->request, 'T');
168 err(c, "No task specified");
171 struct task *task = task_find(tname);
174 err(c, "No such task");
178 byte *pname = obj_find_aval(c->request, 'P');
181 simp_node *s = clist_head(&task->parts);
185 else if (!part_exists_p(task, pname))
187 err(c, "No such task part");
191 byte *ext = obj_find_aval(c->request, 'X');
192 if (!ext || !ext_exists_p(task, ext))
194 err(c, "Missing or invalid extension");
198 struct fastbuf *fb = read_attachment(c);
202 // FIXME: Check contest time
203 // FIXME: Keep history of submitted tasks
206 struct odes *tasko = task_status_find_task(c, task, 1);
207 struct odes *parto = task_status_find_part(tasko, pname, 1);
208 uns current_ver = obj_find_anum(parto, 'V', 0);
210 uns replaced_ver = 0;
211 for (struct oattr *a = obj_find_attr(parto, 'V' + OBJ_ATTR_SON); a; a=a->same)
213 uns ver = obj_find_anum(a->son, 'V', 0);
214 byte *ext = obj_find_aval(a->son, 'X');
216 last_ver = MAX(last_ver, ver);
217 if (ver == current_ver)
219 task_delete_part(c->user, tname, pname, ext, ver);
220 obj_set_attr(a->son, 'S', "replaced");
221 replaced_ver = current_ver;
224 struct odes *vero = obj_add_son(parto, 'V' + OBJ_ATTR_SON);
225 obj_set_attr_num(vero, 'V', ++last_ver);
226 obj_set_attr_num(vero, 'T', time(NULL));
227 obj_set_attr(vero, 'S', "submitted");
228 obj_set_attr(vero, 'X', ext);
230 // FIXME: remove old versions from the status file?
231 task_submit_part(c->user, tname, pname, ext, last_ver, fb);
232 obj_set_attr_num(parto, 'V', last_ver);
233 task_unlock_status(c, 1);
235 log(L_INFO, "User %s submitted task %s%s (version %d%s)",
237 (strcmp(tname, pname) ? stk_printf("/%s", pname) : ""),
239 (replaced_ver ? stk_printf(", replaced %d", replaced_ver) : ""));
242 /*** COMMAND MUX ***/
245 execute_command(struct conn *c)
247 byte *cmd = obj_find_aval(c->request, '!');
250 err(c, "Missing command");
254 log(L_DEBUG, "<< %s", cmd);
255 if (!strcasecmp(cmd, "SUBMIT"))
257 else if (!strcasecmp(cmd, "STATUS"))
260 err(c, "Unknown command");
264 process_command(struct conn *c)
266 if (!read_request(c))
273 /*** INITIAL HANDSHAKE ***/
276 execute_init(struct conn *c)
278 byte *user = obj_find_aval(c->request, 'U');
281 err(c, "Missing user");
285 !strcmp(user, c->cert_name) ||
286 c->rule->allow_admin && !strcmp(c->cert_name, "admin"))
288 if (!user_exists_p(user))
290 err(c, "Unknown user");
293 log(L_INFO, "Logged in %s", user);
297 err(c, "Permission denied");
298 log(L_ERROR, "Unauthorized attempt to log in as %s", user);
301 c->user = xstrdup(user);
305 process_init(struct conn *c)
307 if (!read_request(c))
311 return !obj_find_attr(c->reply, '-');