bulk upload of source

1 files changed, 96 insertions, 0 deletions

diff --git a/update.c b/update.c
new file mode 100644
index 0000000..a7dcc24
--- /dev/null
+++ b/update.c
@@ -0,0 +1,96 @@
+#include <unistd.h>
+#include <fcntl.h>
+#include <alloca.h>
+#include "ninitfeatures.h"
+
+static int infd=-1, outfd=-1;
+static uint32t env_free=0;
+#include "contrib/put_env.h"
+
+static void die_xx(char *s) {
+  char *aa[3]; 
+  INIT_ARGS3(aa, "/sbin/ninit","",0);
+  write(2,"\nupdate: ",9);
+  write(2,s,str_len(s));
+  write(2,"\n",1);
+  while (1) {
+    execve(*aa,aa,environ);	/* really panic !!! */
+    nano_sleep(0,800000000);
+  }
+}
+static void safe_read(void *buf, int len) {
+  if (read(infd,buf,len) != len) die_xx("error reading");
+}
+static void safe_write() {
+  if (write(outfd,"1",1) != 1) die_xx("error writing");
+}
+
+int main(int Argc, char **Argv) {
+  uint32t len,i,k;
+  char **argv, **env, *s;
+
+  s = Argv[0];
+  s[str_rchr(s, '/')] = 0;
+  chdir(s);
+
+  infd=open("../in",O_RDWR);
+  outfd=open("../out",O_RDWR|O_NONBLOCK);
+  fcntl(infd,F_SETFD,FD_CLOEXEC);
+  fcntl(outfd,F_SETFD,FD_CLOEXEC);
+
+  if (infd<0 || outfd<0) {
+    write(outfd,"0",1);
+    die_xx("pipes error");
+  }
+  write(outfd,"7",1);
+  
+  safe_read(&len,sizeof(len)); 
+  argv = alloca((len+5) * sizeof(char *));
+  if (argv==0) die_xx("out of memory");
+  safe_write();
+
+  for (i=0; i<len; i++) {	/* argv */
+    safe_read(&k, sizeof(k));
+    if (k) {
+      if ((s=alloca(k+1))==0) die_xx("out of memory");
+      safe_read(s,k);
+      s[k] = 0;
+    } else s = "";
+    argv[i] = s;
+    safe_write();	
+  }
+  argv[i] = 0;
+
+  /* ----- environ ----- */
+  safe_read(&len,sizeof(len)); 
+  if (len==0) { safe_write(); goto ready; }
+
+  env_free=len+1;
+  if (environ==0) environ = Argv+Argc;
+  for (k=0; environ[k]; ) k++;
+  env = alloca((len+k+5) * sizeof(char *));
+  if (env==0) die_xx("out of memory");
+
+  byte_copy(env, (k+1)*sizeof(char *), environ);
+  environ=env;
+  safe_write();
+
+  for (i=0; i<len; i++) {	/* environ */
+    safe_read(&k, sizeof(k));
+    if (k==0) environ[0] = 0;
+    else {
+      if ((s=alloca(k+1))==0) die_xx("out of memory");
+      safe_read(s,k);
+      s[k] = 0;
+      put_env(s);
+    }
+    safe_write();	
+  }
+  
+ ready:
+  safe_read(&k,sizeof(k));	/* k = "doit" */
+  chdir("/");
+  execve(*argv, argv, environ);
+  die_xx("PANIC");
+  return 1;
+}