summaryrefslogtreecommitdiff
path: root/printutils/lpr.c
blob: 865071011f3a1f1548df5ba969876ff35974a483 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
/* vi: set sw=4 ts=4: */
/*
 * Copyright 2008 Walter Harms (WHarms@bfs.de)
 *
 * Licensed under the GPL v2, see the file LICENSE in this tarball.
 */
#include "libbb.h"
#include "lpr.h"

static char *mygethostname31(void)
{
	char *s = xzalloc(32);
	if (gethostname(s, 31) < 0)
		bb_perror_msg_and_die("gethostname");
	/* gethostname() does not guarantee NUL-termination. xzalloc does. */
	return s;
}

static int xmkstemp(char *temp_name)
{
	int fd;

	fd = mkstemp(temp_name);
	if (fd < 0)
		bb_perror_msg_and_die("mkstemp");
	return fd;
}

/* lpd server sends NUL byte on success.
 * Else read the errormessage and exit.
 */
static void get_response(int server_sock, const char *emsg)
{
	char buf = '\0';

	safe_read(server_sock, &buf, 1);
	if (buf != '\0') {
		bb_error_msg("%s. Server said:", emsg);
		fputc(buf, stderr);
		logmode = 0; /* no errors from bb_copyfd_eof() */
		bb_copyfd_eof(server_sock, STDERR_FILENO);
		xfunc_die();
	}
}

int lpr_main(int argc, char *argv[]) MAIN_EXTERNALLY_VISIBLE;
int lpr_main(int argc, char *argv[])
{
	struct netprint netprint;
	char temp_name[] = "/tmp/lprXXXXXX"; /* for mkstemp */
	char *strings[5];
	const char *netopt;
	const char *jobtitle;
	const char *hostname;
	const char *jobclass;
	char *username;
	int pid1000;
	int server_sock, tmp_fd;
	unsigned opt;
	enum {
		VERBOSE = 1 << 0,
		USE_HEADER = 1 << 1, /* -h: want banner printed */
		USE_MAIL = 1 << 2, /* -m: send mail back to user */
		OPT_U = 1 << 3, /* -U username */
		OPT_J = 1 << 4, /* -J title: the job title for the banner page */
		OPT_C = 1 << 5, /* -C class: job "class" (? supposedly printed on banner) */
		OPT_P = 1 << 6, /* -P queue[@host[:port]] */
		/* if no -P is given use $PRINTER, then "lp@localhost:515" */
	};

	/* Set defaults, parse options */
	hostname = mygethostname31();
	netopt = NULL;
	username = getenv("LOGNAME");
	if (username == NULL)
		username = (char*)"user"; /* TODO: getpwuid(getuid())->pw_name? */
	opt = getopt32(argv, "VhmU:J:C:P:",
			&username, &jobtitle, &jobclass, &netopt);
	argv += optind;
	parse_prt(netopt, &netprint);
	username = xstrndup(username, 31);

	/* For stdin we need to save it to a tempfile,
	 * otherwise we can't know the size. */
	tmp_fd = -1;
	if (!*argv) {
		if (jobtitle == NULL)
			jobtitle = (char *) bb_msg_standard_input;

		tmp_fd = xmkstemp(temp_name);
		if (bb_copyfd_eof(STDIN_FILENO, tmp_fd) < 0)
			goto del_temp_file;
		/* TODO: we actually can have deferred write errors... */
		close(tmp_fd);
		argv--; /* back off from NULL */
		*argv = temp_name;
	}

	if (opt & VERBOSE)
		puts("connect to server");
	server_sock = xconnect_stream(netprint.lsa);

	/* "Receive a printer job" command */
	fdprintf(server_sock, "\x2" "%s\n", netprint.queue);
	get_response(server_sock, "set queue failed");

	pid1000 = getpid() % 1000;
	while (*argv) {
		char dfa_name[sizeof("dfAnnn") + 32];
		struct stat st;
		char **sptr;
		unsigned size;
		int fd;

		fd = xopen(*argv, O_RDONLY);

		/* "The name ... should start with ASCII "dfA",
		 * followed by a three digit job number, followed
		 * by the host name which has constructed the file." */
		snprintf(dfa_name, sizeof(dfa_name),
			"dfA%03u%s", pid1000, hostname);
		pid1000 = (pid1000 + 1) % 1000;

		/*
		 * Generate control file contents
		 */
		/* H HOST, P USER, l DATA_FILE_NAME, J JOBNAME */
		strings[0] = xasprintf("H%.32s\n" "P%.32s\n" "l%.32s\n"
			"J%.99s\n",
			hostname, username, dfa_name,
			(opt & OPT_J) ? jobtitle : *argv);
		sptr = &strings[1];
		/* C CLASS - printed on banner page (if L cmd is also given) */
		if (opt & OPT_C) /* [1] */
			*sptr++ = xasprintf("C%.32s\n", jobclass);
		/* M WHOM_TO_MAIL */
		if (opt & USE_MAIL) /* [2] */
			*sptr++ = xasprintf("M%.32s\n", username);
		/* H USER - print banner page, with given user's name */
		if (opt & USE_HEADER) /* [3] */
			*sptr++ = xasprintf("L%.32s\n", username);
		*sptr = NULL; /* [4] max */

		/* RFC 1179: "LPR servers MUST be able
		 * to receive the control file subcommand first
		 * and SHOULD be able to receive the data file
		 * subcommand first".
		 * Ok, we'll send control file first. */
		size = 0;
		sptr = strings;
		while (*sptr)
			size += strlen(*sptr++);
		if (opt & VERBOSE)
			puts("send control file");
		/* 2: "Receive control file" subcommand */
		fdprintf(server_sock, "\x2" "%u c%s\n", size, dfa_name + 1);
		sptr = strings;
		while (*sptr) {
			xwrite(server_sock, *sptr, strlen(*sptr));
			free(*sptr);
			sptr++;
		}
		free(strings);
		/* "Once all of the contents have
		 * been delivered, an octet of zero bits is sent as
		 * an indication that the file being sent is complete.
		 * A second level of acknowledgement processing
		 * must occur at this point." */
		xwrite(server_sock, "", 1);
		get_response(server_sock, "send control file failed");

		/* Sending data */
		st.st_size = 0; /* paranoia */
		fstat(fd, &st);
		if (opt & VERBOSE)
			puts("send data file");
		/* 3: "Receive data file" subcommand */
		fdprintf(server_sock, "\x3" "%"OFF_FMT"u %s\n", st.st_size, dfa_name);
		/* TODO: if file shrank and we wrote less than st.st_size,
		 * pad output with NUL bytes? Otherwise server won't know
		 * that we are done. */
		if (bb_copyfd_size(fd, server_sock, st.st_size) < 0)
			xfunc_die();
		close(fd);
		xwrite(server_sock, "", 1);
		get_response(server_sock, "send file failed");

		argv++;
	}

	if (ENABLE_FEATURE_CLEAN_UP)
		close(server_sock);

	if (tmp_fd >= 0) {
 del_temp_file:
		unlink(temp_name);
	}

	return 0;
}