ViewVC Help

View File | Revision Log | Show Annotations | Download File | Root Listing

root/Development/ray/src/rt/persist.c

Comparing ray/src/rt/persist.c (file contents):
Revision 2.1 by greg, Wed Jan 20 15:19:00 1993 UTC vs.
Revision 2.41 by greg, Thu May 1 15:50:28 2008 UTC

#	Line 1 | Line 1
1	–	/* Copyright (c) 1993 Regents of the University of California */
2	–
1		#ifndef lint
2	<	static char SCCSid[] = "$SunId$ LBL";
2	>	static const char       RCSid[] = "$Id$";
3		#endif
6	–
4		/*
5		* Routines for persistent rtrace and rpict processes.
6	+	*
7	+	*  External symbols declared in ray.h
8		*/
9
10	<	#include "standard.h"
11	<	#include "paths.h"
10	>	#include "copyright.h"
11	>
12	>	#include <string.h>
13		#include <signal.h>
14	–	#include <sys/types.h>
14		#include <sys/stat.h>
15	+	#include <sys/types.h>
16
17	<	extern char     *strcpy(), *index();
17	>	#include "platform.h"
18	>	#ifndef NON_POSIX /* XXX need abstraction for process management */
19	>	#include <sys/wait.h>
20	>	#endif
21
22	<	extern int      headismine;     /* boolean true if header belongs to me */
22	>	#include "rtprocess.h" /* getpid() */
23	>	#include "standard.h"
24	>	#include "random.h"
25	>	#include "ray.h"
26
27	+	#ifdef F_SETLKW
28	+	#include "paths.h"
29	+	#include "selcall.h"
30	+
31	+	#ifndef TIMELIM
32	+	#define TIMELIM         (8*3600)        /* time limit for holding pattern */
33	+	#endif
34	+
35	+	extern int      headismine;     /* boolean true if header belongs to me */
36	+	extern char     *progname;      /* global program name */
37	+	extern char     *errfile;       /* global error file name */
38		static char     *persistfname = NULL;   /* persist file name */
39		static int      persistfd = -1;         /* persist file descriptor */
40	+	static char     inpname[TEMPLEN+1], outpname[TEMPLEN+1], errname[TEMPLEN+1];
41
42	<	static char     inpname[TEMPLEN+1], outpname[TEMPLEN+1];
42	>	typedef void (sighandler_t)(int);
43	>	static sighandler_t sig_io;
44	>	static sighandler_t sig_alrm;
45
46
47	<	pfdetach()              /* release persist (and header) resources */
47	>	extern void
48	>	pfdetach(void)          /* release persist (and header) resources */
49		{
50		if (persistfd >= 0)
51		close(persistfd);
#	Line 32 | Line 53 | pfdetach()             /* release persist (and header) resources
53		persistfname = NULL;
54		inpname[0] = '\0';
55		outpname[0] = '\0';
56	+	errname[0] = '\0';
57		headismine = 0;
58		}
59
60
61	<	pfclean()               /* clean up persist files */
61	>	extern void
62	>	pfclean(void)           /* clean up persist files */
63		{
64		if (persistfd >= 0)
65		close(persistfd);
#	Line 46 | Line 69 | pfclean()              /* clean up persist files */
69		unlink(inpname);
70		if (outpname[0])
71		unlink(outpname);
72	+	if (errname[0])
73	+	unlink(errname);
74		}
75
76
77	<	pflock(lf)              /* place or release exclusive lock on file */
78	<	int     lf;
77	>	extern void
78	>	pflock(         /* place or release exclusive lock on file */
79	>	int     lf
80	>	)
81		{
82		struct flock    fls;
83
#	Line 63 | Line 90 | int    lf;
90		}
91
92
93	<	persistfile(pfn)        /* open persist file and lock it */
94	<	char    *pfn;
93	>	extern void
94	>	persistfile(    /* open persist file and lock it */
95	>	char    *pfn
96	>	)
97		{
98	<	persistfd = open(pfn, O_WRONLY|O_CREAT|O_EXCL, 0666);
98	>	persistfd = open(pfn, O_WRONLY|O_CREAT|O_EXCL, 0644);
99		if (persistfd >= 0) {
100		persistfname = pfn;
101		pflock(1);
#	Line 83 | Line 112 | char   *pfn;
112		}
113
114
115	<	sig_noop() {}
115	>	static int      got_io;
116
117	+	static void sig_io(int i) { got_io++; }
118
119	<	pfhold()                /* holding pattern for idle rendering process */
119	>	static void sig_alrm(int i) { quit(0); }
120	>
121	>
122	>	extern void
123	>	pfhold(void)            /* holding pattern for idle rendering process */
124		{
125	<	char    buf[128];
125	>	sighandler_t    *oldalrm;
126	>	char    buf[512];
127		register int    n;
128		/* close input and output descriptors */
129	<	close(fileno(stdin));
130	<	fflush(stdout);
131	<	close(fileno(stdout));
129	>	close(0);
130	>	close(1);
131	>	if (errfile == NULL)
132	>	close(2);
133		/* create named pipes for input and output */
134	<	if (mknod(mktemp(strcpy(inpname,TEMPLATE)), S_IFIFO|0600) < 0)
134	>	if (mkfifo(mktemp(strcpy(inpname,TEMPLATE)), 0600) < 0)
135		goto createrr;
136	<	if (mknod(mktemp(strcpy(outpname,TEMPLATE)), S_IFIFO|0600) < 0)
136	>	if (mkfifo(mktemp(strcpy(outpname,TEMPLATE)), 0600) < 0)
137		goto createrr;
138	<	sprintf(buf, "%d\n%s\n%s\n", getpid(), inpname, outpname);
139	<	if (lseek(persistfd, 0L, 0) < 0)
140	<	error(SYSTEM, "seek error on persist file in pfhold");
138	>	if (errfile == NULL &&
139	>	mkfifo(mktemp(strcpy(errname,TEMPLATE)), 0600) < 0)
140	>	goto createrr;
141	>	sprintf(buf, "%s %d\n%s\n%s\n%s\n", progname, getpid(),
142	>	inpname, outpname, errname);
143		n = strlen(buf);
144		if (write(persistfd, buf, n) < n)
145	<	error(SYSTEM, "error writing persist file in pfhold");
146	<	/* wait for someone to signal us */
147	<	signal(SIGALRM, sig_noop);
148	<	pflock(0);
149	<	pause();
150	<	pflock(1);
145	>	error(SYSTEM, "error writing persist file");
146	>	lseek(persistfd, (off_t)0, SEEK_SET);
147	>	/* wait TIMELIM for someone to signal us */
148	>	got_io = 0;
149	>	signal(SIGIO, sig_io);
150	>	oldalrm = signal(SIGALRM, sig_alrm);
151	>	alarm(TIMELIM);
152	>	pflock(0);                      /* unlock persist file for attach */
153	>	while (!got_io)
154	>	pause();                /* wait for attach */
155	>	alarm(0);                       /* turn off alarm */
156	>	signal(SIGALRM, oldalrm);
157	>	signal(SIGIO, SIG_DFL);
158	>	pflock(1);                      /* grab persist file back */
159		/* someone wants us; reopen stdin and stdout */
160	<	if (freopen(inpname, "r", stdin) == NULL)
161	<	goto openerr;
162	<	if (freopen(outpname, "w", stdout) == NULL)
163	<	goto openerr;
160	>	close(0);
161	>	if (open(inpname, O_RDONLY) != 0)
162	>	error(INTERNAL, "unexpected stdin file number");
163	>	clearerr(stdin);
164	>	close(1);
165	>	if (open(outpname, O_WRONLY) != 1)
166	>	error(INTERNAL, "unexpected stdout file number");
167	>	sleep(3);               /* give them a chance to open their pipes */
168	>	if (errname[0]) {
169	>	close(2);
170	>	if (open(errname, O_WRONLY) != 2)
171	>	error(INTERNAL, "unexpected stderr file number");
172	>	unlink(errname);
173	>	errname[0] = '\0';
174	>	}
175		unlink(inpname);
176		inpname[0] = '\0';
177		unlink(outpname);
#	Line 122 | Line 179 | pfhold()               /* holding pattern for idle rendering proces
179		return;
180		createrr:
181		error(SYSTEM, "cannot create named pipes in pfhold");
125	–	openerr:
126	–	error(SYSTEM, "cannot open named pipes in pfhold");
182		}
183
184
185	<	io_process()            /* just act as conduits to and from actual process */
185	>	extern void
186	>	io_process(void)                /* just act as go-between for actual process */
187		{
188		register char   *cp;
189		register int    nr, n;
190	<	char    buf[4096], *pfin, *pfout;
191	<	int     pid;
192	<	/* load and close persist file */
193	<	nr = read(persistfd, buf, sizeof(buf));
194	<	pfdetach();
190	>	char    buf[BUFSIZ], *pfin, *pfout, *pferr;
191	>	int     pid, nfds;
192	>	int     fdout, fderr = -1;
193	>	int     status = 0;
194	>	fd_set  readfds, excepfds;
195	>	/* load persist file */
196	>	n = 40;
197	>	while ((nr = read(persistfd, buf, sizeof(buf)-1)) == 0) {
198	>	if (!n--)
199	>	error(USER, "unattended persist file?");
200	>	pflock(0);
201	>	sleep(3+(3*getpid()+random())%13);      /* wait until ready */
202	>	pflock(1);
203	>	}
204		if (nr < 0)
205	<	error(SYSTEM, "cannot read persist file");
206	<	buf[nr] = '\0';
207	<	if ((cp = index(buf, '\n')) == NULL)
205	>	error(SYSTEM, "error reading persist file");
206	>	#ifndef _WIN32 /* XXX we need a replacement for that one */
207	>	ftruncate(persistfd, (off_t)0L);        /* truncate persist file */
208	>	#endif
209	>	pfdetach();                     /* close & release persist file */
210	>	buf[nr] = '\0';                 /* parse what we got */
211	>	if ((cp = strchr(buf, ' ')) == NULL)
212		goto formerr;
213		*cp++ = '\0';
214	<	if ((pid = atoi(buf)) <= 0)
214	>	if ((pid = atoi(cp)) <= 0)
215		goto formerr;
216	<	pfin = cp;
148	<	if ((cp = index(cp, '\n')) == NULL)
216	>	if ((cp = strchr(cp, '\n')) == NULL)
217		goto formerr;
218	+	pfin = ++cp;
219	+	if ((cp = strchr(cp, '\n')) == NULL)
220	+	goto formerr;
221		*cp++ = '\0';
222		pfout = cp;
223	<	if ((cp = index(cp, '\n')) == NULL)
223	>	if ((cp = strchr(cp, '\n')) == NULL)
224		goto formerr;
225		*cp++ = '\0';
226	+	pferr = cp;
227	+	if ((cp = strchr(cp, '\n')) == NULL)
228	+	goto formerr;
229	+	*cp++ = '\0';
230		if (cp-buf != nr)
231		goto formerr;
232	<	/* wake up rendering process */
233	<	if (kill(pid, SIGALRM) < 0)
232	>	if (strcmp(buf, progname)) {
233	>	sprintf(errmsg, "persist file for %s, not %s", buf, progname);
234	>	error(USER, errmsg);
235	>	}
236	>	/* wake up rendering process */
237	>	if (kill(pid, SIGIO) < 0)
238		error(SYSTEM, "cannot signal rendering process in io_process");
239	<	pid = fork();           /* fork i/o process */
239	>	/* fork child feeder process */
240	>	pid = fork();
241		if (pid < 0)
242		error(SYSTEM, "fork failed in io_process");
243	<	/* connect to appropriate pipe */
244	<	if (pid) {                      /* parent passes renderer output */
245	<	if (freopen(pfout, "r", stdin) == NULL)
246	<	error(SYSTEM, "cannot open input pipe in io_process");
247	<	} else {                        /* child passes renderer input */
248	<	if (freopen(pfin, "w", stdout) == NULL)
249	<	error(SYSTEM, "cannot open input pipe in io_process");
243	>	if (pid == 0) {                 /* feeder loop */
244	>	int     fdin;
245	>	close(1);               /* open input pipe */
246	>	if ((fdin = open(pfin, O_WRONLY)) < 0)
247	>	error(SYSTEM, "cannot open feed pipe in io_process");
248	>	/* renderer stdin */
249	>	while ((nr = read(0, cp=buf, sizeof(buf))) > 0) {
250	>	do {
251	>	if ((n = write(fdin, cp, nr)) <= 0)
252	>	goto writerr;
253	>	cp += n;
254	>	} while ((nr -= n) > 0);
255	>	}
256	>	if (nr < 0)
257	>	goto readerr;
258	>	_exit(0);
259		}
260	<	/* pass input to output */
261	<	cp = buf; n = sizeof(buf);
262	<	while ((nr = read(fileno(stdin), cp, n)) > 0) {
263	<	nr += cp-buf;
264	<	if ((n = write(fileno(stdout), buf, nr)) <= 0)
265	<	error(SYSTEM, "write error in io_process");
266	<	cp = buf;
267	<	while (n < nr)
268	<	*cp++ = buf[n++];
269	<	n = sizeof(buf) - (cp-buf);
260	>	close(0);
261	>	/* open output pipes, in order */
262	>	if ((fdout = open(pfout, O_RDONLY)) < 0)
263	>	error(SYSTEM, "cannot open output pipe in io_process");
264	>	if (pferr[0] && (fderr = open(pferr, O_RDONLY)) < 0)
265	>	error(SYSTEM, "cannot open error pipe in io_process");
266	>	for ( ; ; ) {                   /* eater loop */
267	>	FD_ZERO(&readfds);
268	>	FD_ZERO(&excepfds);
269	>	nfds = 0;
270	>	if (fdout >= 0) {
271	>	FD_SET(fdout, &readfds);
272	>	FD_SET(fdout, &excepfds);
273	>	nfds = fdout+1;
274	>	}
275	>	if (fderr >= 0) {
276	>	FD_SET(fderr, &readfds);
277	>	FD_SET(fderr, &excepfds);
278	>	nfds = fderr+1;
279	>	}
280	>	if (nfds == 0)
281	>	break;                  /* all done, exit */
282	>	if (select(nfds, &readfds, NULL, &excepfds, NULL) < 0)
283	>	error(SYSTEM, "error in select call in io_process");
284	>	/* renderer stderr */
285	>	if (fderr >= 0 && (FD_ISSET(fderr, &readfds) ||
286	>	FD_ISSET(fderr, &excepfds))) {
287	>	nr = read(fderr, cp=buf, sizeof(buf));
288	>	if (nr < 0)
289	>	goto readerr;
290	>	if (nr == 0) {
291	>	close(fderr);
292	>	/* close(2);    don't close stderr! */
293	>	fderr = -1;
294	>	} else {
295	>	cp[nr] = '\0';  /* deduce status if we can */
296	>	n = strlen(progname);
297	>	if (!strncmp(cp, progname, n) &&
298	>	cp[n++] == ':' &&
299	>	cp[n++] == ' ') {
300	>	register struct erract  *ep;
301	>	for (ep = erract; ep < erract+NERRS;
302	>	ep++)
303	>	if (ep->pre[0] &&
304	>	!strncmp(cp+n, ep->pre,
305	>	strlen(ep->pre))) {
306	>	status = ep->ec;
307	>	break;
308	>	}
309	>	}
310	>	do {            /* write message */
311	>	if ((n = write(2, cp, nr)) <= 0)
312	>	goto writerr;
313	>	cp += n;
314	>	} while ((nr -= n) > 0);
315	>	}
316	>	}
317	>	/* renderer stdout */
318	>	if (fdout >= 0 && (FD_ISSET(fdout, &readfds) ||
319	>	FD_ISSET(fdout, &excepfds))) {
320	>	nr = read(fdout, cp=buf, sizeof(buf));
321	>	if (nr < 0)
322	>	goto readerr;
323	>	if (nr == 0) {          /* EOF */
324	>	close(fdout);
325	>	close(1);
326	>	fdout = -1;
327	>	} else
328	>	do {            /* write it all */
329	>	if ((n = write(1, cp, nr)) <= 0)
330	>	goto writerr;
331	>	cp += n;
332	>	} while ((nr -= n) > 0);
333	>	}
334		}
335	<	fclose(stdin);
336	<	fclose(stdout);
337	<	if (pid)                /* parent waits for child */
185	<	wait(0);
186	<	exit(0);                /* all done, exit (not quit!) */
335	>	kill(pid, SIGTERM);     /* no more process to feed, so... */
336	>	waitpid(pid, 0, 0);     /* wait for feeder process */
337	>	_exit(status);
338		formerr:
339		error(USER, "format error in persist file");
340	+	readerr:
341	+	error(SYSTEM, "read error in io_process");
342	+	writerr:
343	+	error(SYSTEM, "write error in io_process");
344		}
345	+
346	+	#else
347	+
348	+	extern void pfclean(void) {}
349	+
350	+	#endif

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines (old)
>	Changed lines (new)