X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?a=blobdiff_plain;f=server%2Fsrc%2Fclient.c;fp=server%2Fsrc%2Fclient.c;h=bec2f11b4ea5a110231c9a5eca71d4f0b2e93874;hb=779888af6bc05f2395e0c9fc0c53067e0674cb27;hp=0000000000000000000000000000000000000000;hpb=fdbff4437ad74ce1457051d2f743e50672be1ec4;p=reposerve

diff --git a/server/src/client.c b/server/src/client.c
new file mode 100644
index 0000000..bec2f11
--- /dev/null
+++ b/server/src/client.c
@@ -0,0 +1,135 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include "client.h"
+#include "proto.h"
+#include "repo.h"
+
+static int proc_request(struct client *c);
+static int proc_flist(struct client *c, char *data, int datasz);
+static int proc_push(struct client *c, char *data, int datasz);
+static int proc_pull(struct client *c, char *data, int datasz);
+
+const char *repo_path;
+
+int handle_client(struct client *c)
+{
+	char buf[512];
+	int sz, left;
+
+	while((sz = read(c->s, buf, sizeof buf)) > 0) {
+		left = sizeof c->buf - c->bsz;
+		if(sz > left) sz = left;
+		memcpy(c->buf + c->bsz, buf, sz);
+		c->bsz += sz;
+	}
+
+	if(proc_request(c) == -1) {
+		return -1;
+	}
+	return 0;
+}
+
+static int proc_request(struct client *c)
+{
+	char *endp = c->buf;
+	char *reqdata;
+	int reqdata_len;
+
+	while(endp < c->buf + c->bsz) {
+		if(*endp == '\n') break;
+	}
+	if(endp >= c->buf + c->bsz) {
+		return 0;
+	}
+	*endp = 0;
+
+	reqdata = endp + 1;
+	reqdata_len = c->bsz - (reqdata - c->buf);
+
+	c->bsz = 0;
+
+	if(strcmp(c->buf, "flist") == 0) {
+		return proc_flist(c, reqdata, reqdata_len);
+	} else if(strcmp(c->buf, "push") == 0) {
+		return proc_push(c, reqdata, reqdata_len);
+	} else if(strcmp(c->buf, "pull") == 0) {
+		return proc_pull(c, reqdata, reqdata_len);
+	} else if(strcmp(c->buf, "dos") == 0) {
+		c->flags |= CLIENT_DOS;
+		send_ok(c, 0);
+		return 0;
+	}
+
+	send_err(c);
+	return 0;
+}
+
+void send_string(struct client *c, const char *s)
+{
+	write(c->s, s, strlen(s));
+}
+
+void send_ok(struct client *c, int resp_sz)
+{
+	char buf[32];
+	sprintf(buf, "OK %d\n", resp_sz < 0 ? 0 : resp_sz);
+	send_string(c, buf);
+}
+
+void send_err(struct client *c)
+{
+	send_string(c, "ERR\n");
+}
+
+static struct flist *gen_flist(int contents)
+{
+	int i, count;
+	struct flist *flist;
+
+	if(!(flist = flist_create())) {
+		return 0;
+	}
+
+	if(repo_init(repo_path) == -1) {
+		return 0;
+	}
+	count = repo_num_files();
+
+	for(i=0; i<count; i++) {
+		flist_add(flist, repo_file(i)->path, contents);
+	}
+	repo_cleanup();
+
+	return flist;
+}
+
+static int proc_flist(struct client *c, char *data, int datasz)
+{
+	struct flist *flist;
+
+	if(!(flist = gen_flist(0))) {
+		send_err(c);
+		return -1;
+	}
+	flist_finalize(flist);
+
+	send_ok(c, flist->final_size);
+	write(c->s, flist->flist, flist->final_size);
+
+	flist_destroy(flist);
+	return 0;
+}
+
+static int proc_push(struct client *c, char *data, int datasz)
+{
+	send_err(c);
+	return 0;
+}
+
+static int proc_pull(struct client *c, char *data, int datasz)
+{
+	send_err(c);
+	return 0;
+}