| 5 | 
  | 
 * Routines for local rtrace execution | 
| 6 | 
  | 
 */ | 
| 7 | 
  | 
 | 
| 8 | 
+ | 
#include <signal.h> | 
| 9 | 
+ | 
#include <sys/time.h> | 
| 10 | 
+ | 
#include <string.h> | 
| 11 | 
+ | 
 | 
| 12 | 
  | 
#include "rholo.h" | 
| 13 | 
  | 
#include "random.h" | 
| 14 | 
  | 
#include "paths.h" | 
| 15 | 
  | 
#include "selcall.h" | 
| 16 | 
< | 
#include <signal.h> | 
| 13 | 
< | 
#include <sys/time.h> | 
| 16 | 
> | 
#include "rtprocess.h" | 
| 17 | 
  | 
 | 
| 18 | 
  | 
#ifndef MAXPROC | 
| 19 | 
  | 
#define MAXPROC         64 | 
| 23 | 
  | 
 | 
| 24 | 
  | 
static char     pfile[] = TEMPLATE;             /* persist file name */ | 
| 25 | 
  | 
 | 
| 26 | 
< | 
static int      rtpd[MAXPROC][3];               /* process descriptors */ | 
| 26 | 
> | 
static SUBPROC  rtpd[MAXPROC];          /* process descriptors */ | 
| 27 | 
  | 
static float    *rtbuf = NULL;                  /* allocated i/o buffer */ | 
| 28 | 
  | 
static int      maxqlen = 0;                    /* maximum packets per queue */ | 
| 29 | 
  | 
 | 
| 30 | 
  | 
static PACKET   *pqueue[MAXPROC];               /* packet queues */ | 
| 31 | 
  | 
static int      pqlen[MAXPROC];                 /* packet queue lengths */ | 
| 32 | 
  | 
 | 
| 33 | 
+ | 
static int bestout(void); | 
| 34 | 
+ | 
static int slots_avail(void); | 
| 35 | 
+ | 
static void queue_packet(PACKET *p); | 
| 36 | 
+ | 
static PACKET * get_packets(int poll); | 
| 37 | 
+ | 
static void killpersist(void); | 
| 38 | 
  | 
 | 
| 39 | 
+ | 
 | 
| 40 | 
  | 
int | 
| 41 | 
< | 
start_rtrace()                  /* start rtrace process */ | 
| 41 | 
> | 
start_rtrace(void)                      /* start rtrace process */ | 
| 42 | 
  | 
{ | 
| 43 | 
  | 
        static char     buf1[8]; | 
| 44 | 
  | 
        int     rmaxpack = 0; | 
| 59 | 
  | 
                rtargv[rtargc++] = "-I-"; | 
| 60 | 
  | 
                rtargv[rtargc++] = "-h-"; | 
| 61 | 
  | 
                rtargv[rtargc++] = "-ld-"; | 
| 62 | 
+ | 
                rtargv[rtargc++] = "-co-"; | 
| 63 | 
  | 
                sprintf(buf1, "%d", RPACKSIZ); | 
| 64 | 
  | 
                rtargv[rtargc++] = "-x"; rtargv[rtargc++] = buf1; | 
| 65 | 
  | 
                rtargv[rtargc++] = "-y"; rtargv[rtargc++] = "0"; | 
| 76 | 
  | 
        } | 
| 77 | 
  | 
        maxqlen = 0; | 
| 78 | 
  | 
        for (nprocs = 0; nprocs < ncprocs; nprocs++) {  /* spawn children */ | 
| 79 | 
< | 
                psiz = open_process(rtpd[nprocs], rtargv); | 
| 79 | 
> | 
                psiz = open_process(&rtpd[nprocs], rtargv); | 
| 80 | 
  | 
                if (psiz <= 0) | 
| 81 | 
  | 
                        error(SYSTEM, "cannot start rtrace process"); | 
| 82 | 
  | 
                n = psiz/(RPACKSIZ*6*sizeof(float)); | 
| 97 | 
  | 
 | 
| 98 | 
  | 
 | 
| 99 | 
  | 
static int | 
| 100 | 
< | 
bestout()                       /* get best process to process packet */ | 
| 100 | 
> | 
bestout(void)                   /* get best process to process packet */ | 
| 101 | 
  | 
{ | 
| 102 | 
  | 
        int     cnt; | 
| 103 | 
< | 
        register int    pn, i; | 
| 103 | 
> | 
        int     pn, i; | 
| 104 | 
  | 
 | 
| 105 | 
  | 
        pn = 0;                 /* find shortest queue */ | 
| 106 | 
  | 
        for (i = 1; i < nprocs; i++) | 
| 114 | 
  | 
                if (pqlen[i] == pqlen[pn]) | 
| 115 | 
  | 
                        cnt++; | 
| 116 | 
  | 
                                /* break ties fairly */ | 
| 117 | 
< | 
        if ((cnt = random() % cnt)) | 
| 117 | 
> | 
        if ((cnt = irandom(cnt))) | 
| 118 | 
  | 
                for (i = pn; i < nprocs; i++) | 
| 119 | 
  | 
                        if (pqlen[i] == pqlen[pn] && !cnt--) | 
| 120 | 
  | 
                                return(i); | 
| 122 | 
  | 
} | 
| 123 | 
  | 
 | 
| 124 | 
  | 
 | 
| 125 | 
< | 
int | 
| 126 | 
< | 
slots_avail()                   /* count packet slots available */ | 
| 125 | 
> | 
static int | 
| 126 | 
> | 
slots_avail(void)                       /* count packet slots available */ | 
| 127 | 
  | 
{ | 
| 128 | 
< | 
        register int    nslots = 0; | 
| 129 | 
< | 
        register int    i; | 
| 128 | 
> | 
        int     nslots = 0; | 
| 129 | 
> | 
        int     i; | 
| 130 | 
  | 
 | 
| 131 | 
  | 
        for (i = nprocs; i--; ) | 
| 132 | 
  | 
                nslots += maxqlen - pqlen[i]; | 
| 134 | 
  | 
} | 
| 135 | 
  | 
 | 
| 136 | 
  | 
 | 
| 137 | 
< | 
queue_packet(p)                 /* queue up a beam packet */ | 
| 138 | 
< | 
register PACKET *p; | 
| 137 | 
> | 
static void | 
| 138 | 
> | 
queue_packet(                   /* queue up a beam packet */ | 
| 139 | 
> | 
        PACKET  *p | 
| 140 | 
> | 
) | 
| 141 | 
  | 
{ | 
| 142 | 
  | 
        int     pn, n; | 
| 143 | 
  | 
                                /* determine process to write to */ | 
| 146 | 
  | 
                                /* write out the packet */ | 
| 147 | 
  | 
        packrays(rtbuf, p); | 
| 148 | 
  | 
        if ((n = p->nr) < RPACKSIZ)     /* add flush block? */ | 
| 149 | 
< | 
                bzero((char *)(rtbuf+6*n++), 6*sizeof(float)); | 
| 150 | 
< | 
        if (writebuf(rtpd[pn][1], (char *)rtbuf, 6*sizeof(float)*n) < 0) | 
| 149 | 
> | 
                memset((char *)(rtbuf+6*n++), '\0', 6*sizeof(float)); | 
| 150 | 
> | 
        if (writebuf(rtpd[pn].w, rtbuf, 6*sizeof(float)*n) < 0) | 
| 151 | 
  | 
                error(SYSTEM, "write error in queue_packet"); | 
| 152 | 
  | 
        p->next = NULL; | 
| 153 | 
  | 
        if (!pqlen[pn]++)       /* add it to the end of the queue */ | 
| 154 | 
  | 
                pqueue[pn] = p; | 
| 155 | 
  | 
        else { | 
| 156 | 
< | 
                register PACKET *rpl = pqueue[pn]; | 
| 156 | 
> | 
                PACKET  *rpl = pqueue[pn]; | 
| 157 | 
  | 
                while (rpl->next != NULL) | 
| 158 | 
  | 
                        rpl = rpl->next; | 
| 159 | 
  | 
                rpl->next = p; | 
| 161 | 
  | 
} | 
| 162 | 
  | 
 | 
| 163 | 
  | 
 | 
| 164 | 
< | 
PACKET * | 
| 165 | 
< | 
get_packets(poll)               /* read packets from rtrace processes */ | 
| 166 | 
< | 
int     poll; | 
| 164 | 
> | 
static PACKET * | 
| 165 | 
> | 
get_packets(            /* read packets from rtrace processes */ | 
| 166 | 
> | 
        int     poll | 
| 167 | 
> | 
) | 
| 168 | 
  | 
{ | 
| 169 | 
  | 
        static struct timeval   tpoll;  /* zero timeval struct */ | 
| 170 | 
  | 
        fd_set  readset, errset; | 
| 171 | 
  | 
        PACKET  *pldone = NULL, *plend; | 
| 172 | 
< | 
        register PACKET *p; | 
| 172 | 
> | 
        PACKET  *p; | 
| 173 | 
  | 
        int     n, nr; | 
| 174 | 
< | 
        register int    pn; | 
| 174 | 
> | 
        int     pn; | 
| 175 | 
  | 
        float   *bp; | 
| 176 | 
  | 
                                        /* prepare select call */ | 
| 177 | 
  | 
        FD_ZERO(&readset); FD_ZERO(&errset); n = 0; | 
| 178 | 
  | 
        for (pn = nprocs; pn--; ) { | 
| 179 | 
  | 
                if (pqlen[pn]) | 
| 180 | 
< | 
                        FD_SET(rtpd[pn][0], &readset); | 
| 181 | 
< | 
                FD_SET(rtpd[pn][0], &errset); | 
| 182 | 
< | 
                if (rtpd[pn][0] >= n) | 
| 183 | 
< | 
                        n = rtpd[pn][0] + 1; | 
| 180 | 
> | 
                        FD_SET(rtpd[pn].r, &readset); | 
| 181 | 
> | 
                FD_SET(rtpd[pn].r, &errset); | 
| 182 | 
> | 
                if (rtpd[pn].r >= n) | 
| 183 | 
> | 
                        n = rtpd[pn].r + 1; | 
| 184 | 
  | 
        } | 
| 185 | 
  | 
                                        /* make the call */ | 
| 186 | 
  | 
        n = select(n, &readset, (fd_set *)NULL, &errset, | 
| 194 | 
  | 
                return(NULL); | 
| 195 | 
  | 
                                        /* make read call(s) */ | 
| 196 | 
  | 
        for (pn = 0; pn < nprocs; pn++) { | 
| 197 | 
< | 
                if (!FD_ISSET(rtpd[pn][0], &readset) && | 
| 198 | 
< | 
                                !FD_ISSET(rtpd[pn][0], &errset)) | 
| 197 | 
> | 
                if (!FD_ISSET(rtpd[pn].r, &readset) && | 
| 198 | 
> | 
                                !FD_ISSET(rtpd[pn].r, &errset)) | 
| 199 | 
  | 
                        continue; | 
| 200 | 
  | 
        reread: | 
| 201 | 
< | 
                n = read(rtpd[pn][0], (char *)rtbuf, | 
| 201 | 
> | 
                n = read(rtpd[pn].r, (char *)rtbuf, | 
| 202 | 
  | 
                                4*sizeof(float)*RPACKSIZ*pqlen[pn]); | 
| 203 | 
  | 
                if (n < 0) { | 
| 204 | 
< | 
                        if (errno == EINTR | errno == EAGAIN) | 
| 204 | 
> | 
                        if ((errno == EINTR) | (errno == EAGAIN)) | 
| 205 | 
  | 
                                goto reread; | 
| 206 | 
  | 
                        error(SYSTEM, "read error in get_packets"); | 
| 207 | 
  | 
                } | 
| 213 | 
  | 
                                nr++;                   /* add flush block */ | 
| 214 | 
  | 
                        n -= 4*sizeof(float)*nr; | 
| 215 | 
  | 
                        if (n < 0) {                    /* get remainder */ | 
| 216 | 
< | 
                                n += readbuf(rtpd[pn][0], | 
| 216 | 
> | 
                                n += readbuf(rtpd[pn].r, | 
| 217 | 
  | 
                                                (char *)(bp+4*nr)+n, -n); | 
| 218 | 
  | 
                                if (n) | 
| 219 | 
  | 
                                        goto eoferr; | 
| 237 | 
  | 
        return(pldone);                         /* return finished packets */ | 
| 238 | 
  | 
eoferr: | 
| 239 | 
  | 
        error(USER, "rtrace process died"); | 
| 240 | 
+ | 
        return NULL; /* pro forma return */ | 
| 241 | 
  | 
} | 
| 242 | 
  | 
 | 
| 243 | 
  | 
 | 
| 244 | 
  | 
PACKET * | 
| 245 | 
< | 
do_packets(pl)                  /* queue a packet list, return finished */ | 
| 246 | 
< | 
register PACKET *pl; | 
| 245 | 
> | 
do_packets(                     /* queue a packet list, return finished */ | 
| 246 | 
> | 
        PACKET  *pl | 
| 247 | 
> | 
) | 
| 248 | 
  | 
{ | 
| 249 | 
< | 
        register PACKET *p; | 
| 249 | 
> | 
        PACKET  *p; | 
| 250 | 
  | 
                                        /* consistency check */ | 
| 251 | 
  | 
        if (nprocs < 1) | 
| 252 | 
  | 
                error(CONSISTENCY, "do_packets called with no active process"); | 
| 260 | 
  | 
 | 
| 261 | 
  | 
 | 
| 262 | 
  | 
PACKET * | 
| 263 | 
< | 
flush_queue()                   /* empty all rtrace queues */ | 
| 263 | 
> | 
flush_queue(void)                       /* empty all rtrace queues */ | 
| 264 | 
  | 
{ | 
| 265 | 
  | 
        PACKET  *rpdone = NULL; | 
| 266 | 
< | 
        register PACKET *rpl; | 
| 266 | 
> | 
        PACKET  *rpl = NULL; | 
| 267 | 
  | 
        float   *bp; | 
| 268 | 
< | 
        register PACKET *p; | 
| 268 | 
> | 
        PACKET  *p; | 
| 269 | 
  | 
        int     i, n, nr; | 
| 270 | 
  | 
 | 
| 271 | 
  | 
        for (i = 0; i < nprocs; i++) | 
| 282 | 
  | 
                                if (rpl->nr < RPACKSIZ) | 
| 283 | 
  | 
                                        nr++;           /* add flush block */ | 
| 284 | 
  | 
                        } | 
| 285 | 
< | 
                        n = readbuf(rtpd[i][0], (char *)rtbuf, | 
| 271 | 
< | 
                                        4*sizeof(float)*nr); | 
| 285 | 
> | 
                        n = readbuf(rtpd[i].r, rtbuf, 4*sizeof(float)*nr); | 
| 286 | 
  | 
                        if (n < 0) | 
| 287 | 
  | 
                                error(SYSTEM, "read failure in flush_queue"); | 
| 288 | 
  | 
                        bp = rtbuf;                     /* process packets */ | 
| 303 | 
  | 
} | 
| 304 | 
  | 
 | 
| 305 | 
  | 
 | 
| 306 | 
< | 
static | 
| 307 | 
< | 
killpersist()                   /* kill persistent process */ | 
| 306 | 
> | 
static void | 
| 307 | 
> | 
killpersist(void)                       /* kill persistent process */ | 
| 308 | 
  | 
{ | 
| 309 | 
  | 
        FILE    *fp; | 
| 310 | 
  | 
        int     pid; | 
| 318 | 
  | 
 | 
| 319 | 
  | 
 | 
| 320 | 
  | 
int | 
| 321 | 
< | 
end_rtrace()                    /* close rtrace process(es) */ | 
| 321 | 
> | 
end_rtrace(void)                        /* close rtrace process(es) */ | 
| 322 | 
  | 
{ | 
| 323 | 
  | 
        int     status = 0, rv; | 
| 324 | 
  | 
 | 
| 325 | 
  | 
        if (nprocs > 1) | 
| 326 | 
  | 
                killpersist(); | 
| 327 | 
< | 
        while (nprocs > 0) { | 
| 328 | 
< | 
                rv = close_process(rtpd[--nprocs]); | 
| 315 | 
< | 
                if (rv > 0) | 
| 316 | 
< | 
                        status = rv; | 
| 317 | 
< | 
        } | 
| 327 | 
> | 
        status = close_processes(rtpd, nprocs); | 
| 328 | 
> | 
        nprocs = 0; | 
| 329 | 
  | 
        free((void *)rtbuf); | 
| 330 | 
  | 
        rtbuf = NULL; | 
| 331 | 
  | 
        maxqlen = 0; |