xref: /illumos-gate/usr/src/cmd/lp/model/netpr/tcp_misc.c (revision 581cede61ac9c14d8d4ea452562a567189eead78)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 
22 /*
23  * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  */
26 
27 #pragma ident	"%Z%%M%	%I%	%E% SMI"
28 
29 #include <stdio.h>
30 #include <stdlib.h>
31 #include <libintl.h>
32 #include <signal.h>
33 #include <errno.h>
34 #include <string.h>
35 #include <strings.h>
36 #include <syslog.h>
37 #include <unistd.h>
38 #include <sys/types.h>
39 #include <fcntl.h>
40 #include <sys/socket.h>
41 #include <netinet/in.h>
42 #include <arpa/inet.h>
43 #include <netdb.h>
44 #include "netpr.h"
45 #include "netdebug.h"
46 
47 #define	MAX_NLPS	60	/* Max no. loops in while */
48 
49 np_tcpjob_t *
50 create_tcp_job(np_job_t *genjob, int filesize)
51 {
52 	np_tcpjob_t *tcpjob;
53 
54 	if (genjob == NULL)
55 		return (NULL);
56 
57 	tcpjob = (np_tcpjob_t *)malloc(sizeof (np_tcpjob_t));
58 	ASSERT(tcpjob, MALLOC_ERR);
59 	(void) memset(tcpjob, 0, sizeof (np_tcpjob_t));
60 
61 	tcpjob->np_port = "9100";
62 	tcpjob->gen_data = genjob;
63 	tcpjob->gen_data->filesize = filesize;
64 
65 	return (tcpjob);
66 }
67 
68 int
69 tcp_open(char *dest, np_tcpjob_t *tcpjob, int timeout)
70 {
71 	struct hostent *hp;
72 	struct sockaddr_in6 serv_addr;
73 	int	s,
74 		err,
75 		error_num;
76 	unsigned timo = 1;
77 	int retry;
78 	int rtnerr;
79 
80 	/*
81 	 * Get the host address and port number to connect to.
82 	 */
83 	if (dest == NULL) {
84 		return (-1);
85 	}
86 
87 	if ((hp = (getipnodebyname(dest, AF_INET6, AI_DEFAULT,
88 		    &error_num))) == NULL) {
89 		(void) fprintf(stderr,
90 		gettext("Netpr: System call getipnodebyname fails\n"));
91 		syslog(LOG_DEBUG, "System call getipnodebyname fails "
92 		    "getipnodebyname() returned %d", error_num);
93 		return (-1);
94 	}
95 
96 	(void) memset(&serv_addr, 0, sizeof (struct sockaddr_in6));
97 	bcopy(hp->h_addr, (caddr_t)&serv_addr.sin6_addr, hp->h_length);
98 	serv_addr.sin6_family = hp->h_addrtype;
99 	serv_addr.sin6_port = (int)htons(atoi(tcpjob->np_port));
100 
101 	do {
102 
103 	retry = 0;
104 	rtnerr = 0;
105 	/*
106 	 * Try connecting to the printer.
107 	 */
108 	s = socket(PF_INET6, SOCK_STREAM, 0);
109 	if (s < 0) {
110 		(void) fprintf(stderr,
111 		    gettext("Netpr: System call socket fails\n"));
112 		syslog(LOG_DEBUG, "System call socket fails");
113 		rtnerr = -1;
114 	} else {
115 		(void) signal(SIGALRM, null_sighandler);
116 		(void) alarm(timeout);
117 		if (connect(s, (struct sockaddr *)&serv_addr,
118 					sizeof (serv_addr)) < 0) {
119 			err = errno;
120 			(void) alarm(0);
121 			errno = err;
122 
123 			if (errno == ECONNREFUSED && timo <= 16) {
124 				(void) sleep(timo);
125 				timo *= 2;
126 				retry++;
127 			} else {
128 			(void) fprintf(stderr,
129 				gettext("Netpr: Cannot connect to printer\n"));
130 				syslog(LOG_DEBUG, "Cannot connect to printer");
131 				rtnerr = -1;
132 			}
133 			/* The connect on this socket failed; close it */
134 			(void) close(s);
135 		} else
136 			(void) alarm(0);
137 	}
138 
139 	} while (retry);
140 
141 	return ((rtnerr) ? rtnerr : s);
142 }
143 
144 
145 int
146 tcp_print(int sockfd, caddr_t pa, np_tcpjob_t *tcpjob)
147 {
148 	char c;
149 	int xfer;
150 	char buf[BUFSIZ + 1];
151 	int nr = 0;
152 	int ctr = 0;
153 	int msg_given = 0;
154 	int nlps = 0;
155 
156 	if ((xfer = xfer_file(sockfd, pa,
157 		tcpjob->gen_data->filesize, tcpjob->gen_data->timeout)) < 0) {
158 			return (xfer);
159 	}
160 
161 	if ((shutdown(sockfd, 1)) != 0) {
162 		(void) fprintf(stderr,
163 		gettext("Netpr: System error: possible loss of data\n"));
164 			syslog(LOG_DEBUG,
165 				"shutdown error; possible loss of data");
166 		return (E_SYSTEM_ERROR);
167 	}
168 
169 
170 	/* read in single character ack or msg from printer */
171 
172 	(void) memset(buf, 0, BUFSIZ + 1);
173 	while (ctr < BUFSIZ) {
174 		(void) signal(SIGALRM, null_sighandler);
175 		(void) alarm(2);
176 		errno = 0;
177 		nr = read(sockfd, &c, 1);
178 		(void) alarm(0);
179 		if (errno == EINTR) {
180 			if (msg_given == 0) {
181 				tell_lptell(ERRORMSG,
182 				gettext("Printer not responding;" \
183 				"Either warming up or needs attention\n"));
184 				msg_given++;
185 			}
186 
187 			/* if no ACK received, do not loop forever */
188 
189 			if (nlps++ >= MAX_NLPS) {
190 				syslog(LOG_DEBUG, "No final ack received");
191 				break;
192 			}
193 		} else {
194 			if ((buf[ctr++] = c) == '\n' || (nr == 0))
195 				break;
196 		}
197 	}
198 	if (ctr > 1)
199 		syslog(LOG_DEBUG, "Message from tcp printer on read: %s",
200 			buf);
201 
202 	if (msg_given && (nlps < MAX_NLPS)) {
203 		(void) fprintf(stderr, gettext("Printer ok\n"));
204 		tell_lptell(OKMSG, "Current");
205 	}
206 
207 	return (E_SUCCESS);
208 }
209