7 |
|
|
8 |
|
#include <signal.h> |
9 |
|
#include <sys/time.h> |
10 |
+ |
#include <string.h> |
11 |
|
|
12 |
|
#include "rholo.h" |
13 |
|
#include "random.h" |
23 |
|
|
24 |
|
static char pfile[] = TEMPLATE; /* persist file name */ |
25 |
|
|
26 |
< |
static SUBPROC rtpd[MAXPROC]; /* 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 */ |
28 |
> |
static int maxqlen; /* 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"; |
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)); |
83 |
< |
if (maxqlen == 0) { |
82 |
> |
n = psiz/(RPACKSIZ*6*sizeof(float)) + 1; |
83 |
> |
if (!maxqlen) { |
84 |
|
if (!(maxqlen = n)) |
85 |
|
error(INTERNAL, |
86 |
|
"bad pipe buffer size assumption"); |
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].w, (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; |
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 |
|
} |
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].r, (char *)rtbuf, |
273 |
< |
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]); |
317 |
< |
if (rv > 0) |
318 |
< |
status = rv; |
319 |
< |
} |
327 |
> |
status = close_processes(rtpd, nprocs); |
328 |
> |
nprocs = 0; |
329 |
|
free((void *)rtbuf); |
330 |
|
rtbuf = NULL; |
331 |
|
maxqlen = 0; |