M xrxs.c → xrxs.c

@@ -6,9 +6,12 @@ #include <9p.h>
 #include <stdio.h>
 #include <dirent.h>
 #include <libString.h>
+#include <mp.h>
+#include <libsec.h>
 
 #define CARTSLOC "./carts/"
-
+#define CTL_HASHV 139931
+#define SHUTDOWN_HASHV 11192337284248
 /* clang-format off */
 
 char  clca(char c) { return c >= 'A' && c <= 'Z' ? c + ('a' - 'A') : c; } /* char to lowercase */
@@ -25,6 +28,8 @@ int   ssin(char *s, char *ss) { int a = 0, b = 0; while(s[a]) { if(s[a] == ss[b]) { if(!ss[b + 1]) return a - b; b++; } else b = 0; a++; } return -1; } /* string substring index */
 char *ccat(char *dst, char c) { int len = slen(dst); dst[len] = c; dst[len + 1] = '\0'; return dst; }
 
 /* clang-format on */
+int chatty9p = 1;
+static int serving = 1;
 
 static char Ebad[] = "something bad happened";
 static char Enomem[] = "no memory";
@@ -59,6 +64,16 @@ r->ofcall.count = count;
   respond(r, nil);
 }
 
+unsigned long long hash(char* str) {
+  unsigned long long h;
+  unsigned char* p;
+
+  h = 0;
+  for (p = (unsigned char*)str; *p != '\0'; p++)
+    h = 37 * h + *p;
+  return h; // or, h % ARRAY_SIZE;
+}
+
 void fswrite(Req* r) {
   void* v;
   Ramfile* rf;
@@ -84,10 +99,45 @@ r->ofcall.count = count;
   respond(r, nil);
 }
 
+void handlectl(Req* r) {
+  char cmd[16];
+  char* c = r->ifcall.data;
+
+  while (*c && *c != ' ') {
+    ccat(cmd, *c);
+  }
+
+  unsigned long long const cmd_hashv = hash(cmd);
+
+  switch (cmd_hashv) {
+    case SHUTDOWN_HASHV:
+      serving = 0;
+      break;
+    default:
+      break;
+  }
+  r->ofcall.count = 16;
+  r->fid->file->dir.length = 16;
+  respond(r, nil);
+}
+
+void xrxswrite(Req* r) {
+  unsigned int filename_hashv = hash(r->ifcall.name);
+  printf(r->ifcall.name);
+  switch (filename_hashv) {
+    case CTL_HASHV:
+      handlectl(r);
+      break;
+    default:
+      fswrite(r);
+      break;
+  }
+}
+
 void fscreate(Req* r) {
   Ramfile* rf;
   File* f;
-
+  printf("eee");
   if (f = createfile(r->fid->file, r->ifcall.name, nil, r->ifcall.perm, nil)) {
     rf = emalloc9p(sizeof *rf);
     f->aux = rf;
@@ -99,9 +149,22 @@ }
   respond(r, Ebad);
 }
 
+void xrxscreate(Req* r) {
+  unsigned int filename_hashv = hash(r->ifcall.name);
+  printf("fff");
+  switch (filename_hashv) {
+    case CTL_HASHV:
+      handlectl(r);
+      break;
+    default:
+      fscreate(r);
+      break;
+  }
+}
+
 void fsopen(Req* r) {
   Ramfile* rf;
-
+  printf("ooo");
   rf = r->fid->file->aux;
 
   if (rf && (r->ifcall.mode & OTRUNC)) {
@@ -166,11 +229,17 @@ rf->ndata = slen(data);
   f->aux = rf;
 }
 
+void xrxswstat(Req* r) {
+  r->fid->omode = OWRITE;
+  respond(r, nil);
+}
+
 Srv fs = {
   .open = fsopen,
   .read = fsread,
-  .write = fswrite,
-  .create = fscreate,
+  .write = xrxswrite,
+  .create = xrxscreate,
+  .wstat = xrxswstat
 };
 
 int threadmaybackground(void) { return 1; }
@@ -182,6 +251,8 @@ char* usocket = nil;
   int i;
   String** cart;
   char cartsloc[256] = "./carts/";
+  char ctlbuf[128] = "some data";
+  Ramfile ctlfile = {ctlbuf, 128};
 
   /* if -m PATH is supplied, mount on PATH */
   /* if -s NAME is supplied, create a socket for the namespace */
@@ -202,6 +273,8 @@
   fs.tree = alloctree(nil, nil, DMDIR | 0777, fsdestroyfile);
   tree = fs.tree;
   createfile(tree->root, "carts", nil, DMDIR | 0555, nil);
+  createfile(tree->root, "ctl", nil, 0600, &ctlfile);
+
   String** carts = listdir("carts/");
   cart = carts;
   while (*cart) {
@@ -210,13 +283,15 @@ cart++;
   }
   fs.foreground = 1;
 
-  if (argc >= 3) {
-    if (mtpt != nil && access(mtpt, AEXIST) < 0 && access(mtpt, AEXIST) < 0)
-      sysfatal("mountpoint %s does not exist", mtpt);
+  while (serving) {
+    if (argc >= 3) {
+      if (mtpt != nil && access(mtpt, AEXIST) < 0 && access(mtpt, AEXIST) < 0)
+        sysfatal("mountpoint %s does not exist", mtpt);
 
-    threadpostmountsrv(&fs, usocket, mtpt, MREPL | MCREATE);
-    threadexits(0);
-  } else {
-    srv(&fs);
+      threadpostmountsrv(&fs, usocket, mtpt, MREPL | MCREATE);
+      threadexits(0);
+    } else {
+      srv(&fs);
+    }
   }
 }