Return to dkinstall.c CVS log

Up to [Research Unix] / researchv10no / cmd / asd++

Annotation of researchv10no/cmd/asd++/dkinstall.c, revision 1.1.1.1

1.1       root        1: #include "decl.h"
                      2: #include <signal.h>
                      3: #include <errno.h>
                      4: #include <pwd.h>
                      5: 
                      6: extern "C" {
                      7:        extern int dup2 (int, int);
                      8:        extern int fork();
                      9:        extern int setupshares (int, void(*)() = 0);
                     10:        extern void perror (char*);
                     11:        extern int errno;
                     12:        extern int sys_nerr;
                     13:        extern char *sys_errlist[];
                     14: }
                     15: 
                     16: static const int bufsize = 4096;
                     17: static const int timeout = 60*10;
                     18: static const int wtimeout = 15;
                     19: 
                     20: static void
                     21: alarmcatch(int)
                     22: {
                     23:        signal (SIGALRM, alarmcatch);
                     24: }
                     25: 
                     26: static void
                     27: die (String s)
                     28: {
                     29:        String msg = s;
                     30:        if ((unsigned) errno < sys_nerr)
                     31:                msg += ":" + String(sys_errlist[errno]);
                     32:        msg += "\n";
                     33:        cout << msg;
                     34:        exit (1);
                     35: }
                     36: 
                     37: static void
                     38: twrite (char* buf, int len)
                     39: {
                     40:        alarm (timeout);
                     41:        int n = write (1, buf, len);
                     42:        alarm (0);
                     43:        if (n != len)
                     44:                die ("remote write failed");
                     45: }
                     46: 
                     47: static int
                     48: tread (char* buf, int len)
                     49: {
                     50:        alarm (timeout);
                     51:        int n = read (0, buf, len);
                     52:        alarm (0);
                     53:        if (n < 0)
                     54:                die ("remote read failed");
                     55:        return n;
                     56: }
                     57: 
                     58: static void
                     59: send (char* s)
                     60: {
                     61:        twrite (s, strlen (s));
                     62: }
                     63: 
                     64: static int
                     65: mkfile (String file)
                     66: {
                     67:        unlink (file);
                     68:        int fd = creat (file, 0600);
                     69:        if (fd < 0)
                     70:                die ("temp file creat");
                     71:        if (close (fd) < 0)
                     72:                die ("temp file close");
                     73:        fd = open (file, 2);
                     74:        if (fd < 0)
                     75:                die ("temp file reopen");
                     76:        if (unlink (file) < 0)
                     77:                die ("temp file unlink");
                     78:        return fd;
                     79: }
                     80: 
                     81: main()
                     82: {
                     83:        signal (SIGALRM, alarmcatch);
                     84: 
                     85:        /* announce our presence */
                     86:        send ("asd\n");
                     87: 
                     88:        /* use daemon's share instead of root */
                     89:        struct passwd *pw;
                     90:        pw = getpwnam ("daemon");
                     91:        if (pw)
                     92:                setupshares (pw->pw_uid);
                     93:        char buf[bufsize];
                     94:        int n;
                     95:        checksum cs, rcs;
                     96: 
                     97:        umask (077);
                     98: 
                     99:        /* make a temp file to hold the data */
                    100:        Path tfile = Path(tmpdir) & "asd" + String(dec(getpid()));
                    101:        int infd = mkfile (tfile);
                    102:        int outfd = mkfile (tfile);
                    103: 
                    104:        /* copy the data to the input temp file and checksum it */
                    105:        while ((n = tread (buf, bufsize)) > 0) {
                    106:                cs.combine (buf, n);
                    107:                if (write (infd, buf, n) != n)
                    108:                        die ("temp file write");
                    109:        }
                    110:        lseek (infd, 0, 0);
                    111: 
                    112:        /* Check for normal end of data */
                    113:        if (n < 0)
                    114:                die ("remote read");
                    115: 
                    116:        /* Try to read the checksum */
                    117:        alarm (timeout);
                    118:        n = read (0, rcs);
                    119:        alarm (0);
                    120: 
                    121:        /* Do the checksums match? */
                    122:        if (n != cksize || cs != rcs) {
                    123:                cout << "checksum error\n";
                    124:                exit (1);
                    125:        }
                    126: 
                    127:        /* Assert we got the stuff successfully */
                    128:        twrite ("", 0);
                    129: 
                    130:        /* Run inspkg and put the results in the output temp file */
                    131:        int rc, status;
                    132:        int pid = fork();
                    133: 
                    134:        switch (pid) {
                    135:        case -1:
                    136:                die ("fork");
                    137: 
                    138:        case 0:                 // child
                    139:                dup2 (infd, 0);
                    140:                dup2 (outfd, 1);
                    141:                close (infd);
                    142:                close (outfd);
                    143:                dup2 (1, 2);
                    144:                dup2 (1, 3);
                    145:                execl (inspkg, "inspkg", (char*) 0);
                    146:                die ("execl");
                    147: 
                    148:        default:                // parent
                    149:                do {
                    150:                        alarm (wtimeout);
                    151:                        rc = wait (&status);
                    152:                        alarm (0);
                    153:                        if (rc == -1) {
                    154:                                if (errno == EINTR)
                    155:                                        twrite ("\n", 1);
                    156:                                else
                    157:                                        die ("wait");
                    158:                        }
                    159:                } while (rc != pid);
                    160:        }
                    161: 
                    162:        /* Send the output temp file to the other end */
                    163:        lseek (outfd, 0, 0);
                    164:        while ((n = read (outfd, buf, bufsize)) > 0)
                    165:                twrite (buf, n);
                    166: 
                    167:        /* Check for normal EOF */
                    168:        if (n < 0)
                    169:                die ("reading inspkg output");
                    170: 
                    171:        /* Report the inspkg return code */
                    172:        cout << "return code " << status << "\n";
                    173: 
                    174:        return 0;
                    175: }

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.