recvjob.c revision 1.13
1 /* $NetBSD: recvjob.c,v 1.13 2001/10/09 02:15:38 mjl Exp $ */ 2 3 /* 4 * Copyright (c) 1983, 1993 5 * The Regents of the University of California. All rights reserved. 6 * 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 1. Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 3. All advertising materials mentioning features or use of this software 17 * must display the following acknowledgement: 18 * This product includes software developed by the University of 19 * California, Berkeley and its contributors. 20 * 4. Neither the name of the University nor the names of its contributors 21 * may be used to endorse or promote products derived from this software 22 * without specific prior written permission. 23 * 24 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 25 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 26 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 27 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 28 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 29 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 30 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 31 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 32 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 33 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 34 * SUCH DAMAGE. 35 */ 36 37 #include <sys/cdefs.h> 38 39 #ifndef lint 40 __COPYRIGHT("@(#) Copyright (c) 1983, 1993\n\ 41 The Regents of the University of California. All rights reserved.\n"); 42 #endif /* not lint */ 43 44 #ifndef lint 45 #if 0 46 static char sccsid[] = "@(#)recvjob.c 8.2 (Berkeley) 4/27/95"; 47 #else 48 __RCSID("$NetBSD: recvjob.c,v 1.13 2001/10/09 02:15:38 mjl Exp $"); 49 #endif 50 #endif /* not lint */ 51 52 /* 53 * Receive printer jobs from the network, queue them and 54 * start the printer daemon. 55 */ 56 #include <sys/param.h> 57 #include <sys/mount.h> 58 #include <sys/stat.h> 59 60 #include <unistd.h> 61 #include <signal.h> 62 #include <fcntl.h> 63 #include <dirent.h> 64 #include <syslog.h> 65 #include <stdio.h> 66 #include <stdlib.h> 67 #include <string.h> 68 #include "lp.h" 69 #include "lp.local.h" 70 #include "extern.h" 71 #include "pathnames.h" 72 73 #define ack() (void)write(STDOUT_FILENO, sp, 1); 74 75 static char dfname[NAME_MAX]; /* data files */ 76 static int minfree; /* keep at least minfree blocks available */ 77 static char *sp = ""; 78 static char tfname[NAME_MAX]; /* tmp copy of cf before linking */ 79 80 static int chksize __P((int)); 81 static void frecverr __P((const char *, ...)) 82 __attribute__((__format__(__printf__, 1, 2))); 83 static int noresponse __P((void)); 84 static void rcleanup __P((int)); 85 static int read_number __P((char *)); 86 static int readfile __P((char *, int)); 87 static int readjob __P((void)); 88 89 90 void 91 recvjob(void) 92 { 93 struct stat stb; 94 int status; 95 96 /* 97 * Perform lookup for printer name or abbreviation 98 */ 99 if ((status = cgetent(&bp, printcapdb, printer)) == -2) 100 frecverr("cannot open printer description file"); 101 else if (status == -1) 102 frecverr("unknown printer %s", printer); 103 else if (status == -3) 104 fatal("potential reference loop detected in printcap file"); 105 106 if (cgetstr(bp, "lf", &LF) == -1) 107 LF = _PATH_CONSOLE; 108 if (cgetstr(bp, "sd", &SD) == -1) 109 SD = _PATH_DEFSPOOL; 110 if (cgetstr(bp, "lo", &LO) == -1) 111 LO = DEFLOCK; 112 113 (void)close(2); /* set up log file */ 114 if (open(LF, O_WRONLY|O_APPEND, 0664) < 0) { 115 syslog(LOG_ERR, "%s: %m", LF); 116 (void)open(_PATH_DEVNULL, O_WRONLY); 117 } 118 119 if (chdir(SD) < 0) 120 frecverr("%s: %s: %m", printer, SD); 121 if (stat(LO, &stb) == 0) { 122 if (stb.st_mode & 010) { 123 /* queue is disabled */ 124 putchar('\1'); /* return error code */ 125 exit(1); 126 } 127 } else if (stat(SD, &stb) < 0) 128 frecverr("%s: %s: %m", printer, SD); 129 minfree = 2 * read_number("minfree"); /* scale KB to 512 blocks */ 130 signal(SIGTERM, rcleanup); 131 signal(SIGPIPE, rcleanup); 132 133 if (readjob()) 134 printjob(); 135 } 136 137 /* 138 * Read printer jobs sent by lpd and copy them to the spooling directory. 139 * Return the number of jobs successfully transfered. 140 */ 141 static int 142 readjob(void) 143 { 144 int size, nfiles; 145 char *cp; 146 147 ack(); 148 nfiles = 0; 149 for (;;) { 150 /* 151 * Read a command to tell us what to do 152 */ 153 cp = line; 154 do { 155 if ((size = read(STDOUT_FILENO, cp, 1)) != 1) { 156 if (size < 0) 157 frecverr("%s: Lost connection", 158 printer); 159 return(nfiles); 160 } 161 } while (*cp++ != '\n' && (cp - line + 1) < sizeof(line)); 162 if (cp - line + 1 >= sizeof(line)) 163 frecverr("readjob overflow"); 164 *--cp = '\0'; 165 cp = line; 166 switch (*cp++) { 167 case '\1': /* cleanup because data sent was bad */ 168 rcleanup(0); 169 continue; 170 171 case '\2': /* read cf file */ 172 size = 0; 173 while (*cp >= '0' && *cp <= '9') 174 size = size * 10 + (*cp++ - '0'); 175 if (*cp++ != ' ') 176 break; 177 /* 178 * host name has been authenticated, we use our 179 * view of the host name since we may be passed 180 * something different than what gethostbyaddr() 181 * returns 182 */ 183 (void)strncpy(cp + 6, from, sizeof(line) - strlen(line) - 1); 184 line[sizeof(line) - 1 ] = '\0'; 185 (void)strncpy(tfname, cp, sizeof(tfname) - 1); 186 tfname[sizeof(tfname) - 1 ] = '\0'; 187 tfname[0] = 't'; 188 if (strchr(tfname, '/')) 189 frecverr("readjob: %s: illegal path name", 190 tfname); 191 if (!chksize(size)) { 192 (void)write(STDOUT_FILENO, "\2", 1); 193 continue; 194 } 195 if (!readfile(tfname, size)) { 196 rcleanup(0); 197 continue; 198 } 199 if (link(tfname, cp) < 0) 200 frecverr("%s: %m", tfname); 201 (void)unlink(tfname); 202 tfname[0] = '\0'; 203 nfiles++; 204 continue; 205 206 case '\3': /* read df file */ 207 size = 0; 208 while (*cp >= '0' && *cp <= '9') 209 size = size * 10 + (*cp++ - '0'); 210 if (*cp++ != ' ') 211 break; 212 if (!chksize(size)) { 213 (void)write(STDOUT_FILENO, "\2", 1); 214 continue; 215 } 216 (void)strncpy(dfname, cp, sizeof(dfname) - 1); 217 dfname[sizeof(dfname) - 1] = '\0'; 218 if (strchr(dfname, '/')) 219 frecverr("readjob: %s: illegal path name", 220 dfname); 221 (void)readfile(dfname, size); 222 continue; 223 } 224 frecverr("protocol screwup: %s", line); 225 } 226 } 227 228 /* 229 * Read files send by lpd and copy them to the spooling directory. 230 */ 231 static int 232 readfile(char *file, int size) 233 { 234 char *cp; 235 char buf[BUFSIZ]; 236 int i, j, amt; 237 int fd, err; 238 239 fd = open(file, O_CREAT|O_EXCL|O_WRONLY, FILMOD); 240 if (fd < 0) 241 frecverr("readfile: %s: illegal path name: %m", file); 242 ack(); 243 err = 0; 244 for (i = 0; i < size; i += BUFSIZ) { 245 amt = BUFSIZ; 246 cp = buf; 247 if (i + amt > size) 248 amt = size - i; 249 do { 250 j = read(STDOUT_FILENO, cp, amt); 251 if (j <= 0) 252 frecverr("Lost connection"); 253 amt -= j; 254 cp += j; 255 } while (amt > 0); 256 amt = BUFSIZ; 257 if (i + amt > size) 258 amt = size - i; 259 if (write(fd, buf, amt) != amt) { 260 err++; 261 break; 262 } 263 } 264 (void)close(fd); 265 if (err) 266 frecverr("%s: write error", file); 267 if (noresponse()) { /* file sent had bad data in it */ 268 (void)unlink(file); 269 return(0); 270 } 271 ack(); 272 return(1); 273 } 274 275 static int 276 noresponse(void) 277 { 278 char resp; 279 280 if (read(STDOUT_FILENO, &resp, 1) != 1) 281 frecverr("Lost connection"); 282 if (resp == '\0') 283 return(0); 284 return(1); 285 } 286 287 /* 288 * Check to see if there is enough space on the disk for size bytes. 289 * 1 == OK, 0 == Not OK. 290 */ 291 static int 292 chksize(int size) 293 { 294 int spacefree; 295 struct statfs sfb; 296 297 if (statfs(".", &sfb) < 0) { 298 syslog(LOG_ERR, "%s: %m", "statfs(\".\")"); 299 return (1); 300 } 301 spacefree = sfb.f_bavail * (sfb.f_bsize / 512); 302 size = (size + 511) / 512; 303 if (minfree + size > spacefree) 304 return(0); 305 return(1); 306 } 307 308 static int 309 read_number(fn) 310 char *fn; 311 { 312 char lin[80]; 313 FILE *fp; 314 315 if ((fp = fopen(fn, "r")) == NULL) 316 return (0); 317 if (fgets(lin, 80, fp) == NULL) { 318 fclose(fp); 319 return (0); 320 } 321 fclose(fp); 322 return (atoi(lin)); 323 } 324 325 /* 326 * Remove all the files associated with the current job being transfered. 327 */ 328 static void 329 rcleanup(int signo) 330 { 331 if (tfname[0]) 332 (void)unlink(tfname); 333 if (dfname[0]) 334 do { 335 do { 336 if (strchr(dfname, '/') == 0) 337 (void)unlink(dfname); 338 } while (dfname[2]-- != 'A'); 339 dfname[2] = 'z'; 340 } while (dfname[0]-- != 'd'); 341 dfname[0] = '\0'; 342 } 343 344 #ifdef __STDC__ 345 #include <stdarg.h> 346 #else 347 #include <varargs.h> 348 #endif 349 350 static void 351 #ifdef __STDC__ 352 frecverr(const char *msg, ...) 353 #else 354 frecverr(msg, va_alist) 355 char *msg; 356 va_dcl 357 #endif 358 { 359 extern char fromb[]; 360 va_list ap; 361 #ifdef __STDC__ 362 va_start(ap, msg); 363 #else 364 va_start(ap); 365 #endif 366 rcleanup(0); 367 syslog(LOG_ERR, "%s", fromb); 368 vsyslog(LOG_ERR, msg, ap); 369 va_end(ap); 370 putchar('\1'); /* return error code */ 371 exit(1); 372 } 373
Indexes created Sun Sep 28 07:09:56 GMT 2025