2 * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
3 * Copyright (c) 1996, 1998 by Internet Software Consortium.
5 * Permission to use, copy, modify, and distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
9 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR
12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
15 * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18 #if !defined(LINT) && !defined(CODECENTER)
19 static const char rcsid
[] = "$Id: irp.c,v 1.9.394.2 2008/02/28 05:46:12 marka Exp $";
24 #include "port_before.h"
27 #include <sys/types.h>
28 #include <sys/socket.h>
30 #include <netinet/in.h>
31 #include <arpa/inet.h>
41 #include <isc/memcluster.h>
49 #include "port_after.h"
53 static void irp_close(struct irs_acc
*);
59 (sizeof (*(su)) - sizeof ((su)->sun_path) + strlen((su)->sun_path))
66 /* send errors to syslog if true. */
67 int irp_log_errors
= 1;
70 * This module handles the irp module connection to irpd.
72 * The client expects a synchronous interface to functions like
73 * getpwnam(3), so we can't use the ctl_* i/o library on this end of
74 * the wire (it's used in the server).
78 * irs_acc *irs_irp_acc(const char *options);
80 * Initialize the irp module.
83 irs_irp_acc(const char *options
) {
89 if (!(acc
= memget(sizeof *acc
))) {
93 memset(acc
, 0x5e, sizeof *acc
);
94 if (!(irp
= memget(sizeof *irp
))) {
105 acc
->gr_map
= irs_irp_gr
;
110 acc
->pw_map
= irs_irp_pw
;
114 acc
->sv_map
= irs_irp_sv
;
115 acc
->pr_map
= irs_irp_pr
;
116 acc
->ho_map
= irs_irp_ho
;
117 acc
->nw_map
= irs_irp_nw
;
118 acc
->ng_map
= irs_irp_ng
;
119 acc
->close
= irp_close
;
125 irs_irp_connection_setup(struct irp_p
*cxndata
, int *warned
) {
126 if (irs_irp_is_connected(cxndata
)) {
128 } else if (irs_irp_connect(cxndata
) != 0) {
129 if (warned
!= NULL
&& !*warned
) {
130 syslog(LOG_ERR
, "irpd connection failed: %m\n");
141 * int irs_irp_connect(void);
143 * Sets up the connection to the remote irpd server.
147 * 0 on success, -1 on failure.
151 irs_irp_connect(struct irp_p
*pvt
) {
153 struct sockaddr
*addr
;
154 struct sockaddr_in iaddr
;
155 #ifndef NO_SOCKADDR_UN
156 struct sockaddr_un uaddr
;
164 if (pvt
->fdCxn
!= -1) {
169 #ifndef NO_SOCKADDR_UN
170 memset(&uaddr
, 0, sizeof uaddr
);
172 memset(&iaddr
, 0, sizeof iaddr
);
174 irphost
= getenv(IRPD_HOST_ENV
);
175 if (irphost
== NULL
) {
176 irphost
= "127.0.0.1";
179 #ifndef NO_SOCKADDR_UN
180 if (irphost
[0] == '/') {
181 addr
= (struct sockaddr
*)&uaddr
;
182 strncpy(uaddr
.sun_path
, irphost
, sizeof uaddr
.sun_path
);
183 uaddr
.sun_family
= AF_UNIX
;
184 socklen
= SUN_LEN(&uaddr
);
186 uaddr
.sun_len
= socklen
;
191 if (inet_pton(AF_INET
, irphost
, &ipaddr
) != 1) {
192 errno
= EADDRNOTAVAIL
;
197 addr
= (struct sockaddr
*)&iaddr
;
198 socklen
= sizeof iaddr
;
200 iaddr
.sin_len
= socklen
;
202 iaddr
.sin_family
= AF_INET
;
203 iaddr
.sin_port
= htons(IRPD_PORT
);
204 iaddr
.sin_addr
.s_addr
= ipaddr
;
208 pvt
->fdCxn
= socket(addr
->sa_family
, SOCK_STREAM
, PF_UNSPEC
);
209 if (pvt
->fdCxn
< 0) {
214 if (connect(pvt
->fdCxn
, addr
, socklen
) != 0) {
219 flags
= fcntl(pvt
->fdCxn
, F_GETFL
, 0);
228 if (fcntl(pvt
->fdCxn
, F_SETFL
, flags
) < 0) {
235 code
= irs_irp_read_response(pvt
, text
, sizeof text
);
236 if (code
!= IRPD_WELCOME_CODE
) {
237 if (irp_log_errors
) {
238 syslog(LOG_WARNING
, "Connection failed: %s", text
);
240 irs_irp_disconnect(pvt
);
248 * int irs_irp_is_connected(struct irp_p *pvt);
252 * Non-zero if streams are setup to remote.
257 irs_irp_is_connected(struct irp_p
*pvt
) {
258 return (pvt
->fdCxn
>= 0);
263 * irs_irp_disconnect(struct irp_p *pvt);
265 * Closes streams to remote.
269 irs_irp_disconnect(struct irp_p
*pvt
) {
270 if (pvt
->fdCxn
!= -1) {
279 irs_irp_read_line(struct irp_p
*pvt
, char *buffer
, int len
) {
280 char *realstart
= &pvt
->inbuffer
[0];
281 char *p
, *start
, *end
;
288 start
= p
= &pvt
->inbuffer
[pvt
->incurr
];
289 end
= &pvt
->inbuffer
[pvt
->inlast
];
291 while (p
!= end
&& *p
!= '\n')
295 /* Found no newline so shift data down if necessary
296 * and append new data to buffer
298 if (start
> realstart
) {
299 memmove(realstart
, start
, end
- start
);
300 pvt
->inlast
= end
- start
;
303 end
= &pvt
->inbuffer
[pvt
->inlast
];
306 spare
= sizeof (pvt
->inbuffer
) - pvt
->inlast
;
309 i
= read(pvt
->fdCxn
, end
, spare
);
313 return (buffpos
> 0 ? buffpos
: -1);
321 while (p
!= end
&& *p
!= '\n')
326 /* full buffer and still no newline */
327 i
= sizeof pvt
->inbuffer
;
329 /* include newline */
335 memcpy(buffer
+ buffpos
, start
, i
);
338 buffer
[buffpos
] = '\0';
348 fprintf(stderr
, "read line: %s\n", buffer
);
354 * int irp_read_response(struct irp_p *pvt);
358 * The number found at the beginning of the line read from
359 * FP. 0 on failure(0 is not a legal response code). The
360 * rest of the line is discarded.
365 irs_irp_read_response(struct irp_p
*pvt
, char *text
, size_t textlen
) {
370 if (irs_irp_read_line(pvt
, line
, sizeof line
) <= 0) {
374 p
= strchr(line
, '\n');
379 if (sscanf(line
, "%d", &code
) != 1) {
381 } else if (text
!= NULL
&& textlen
> 0U) {
383 while (isspace((unsigned char)*p
)) p
++;
384 while (isdigit((unsigned char)*p
)) p
++;
385 while (isspace((unsigned char)*p
)) p
++;
386 strncpy(text
, p
, textlen
- 1);
387 p
[textlen
- 1] = '\0';
394 * char *irp_read_body(struct irp_p *pvt, size_t *size);
396 * Read in the body of a response. Terminated by a line with
397 * just a dot on it. Lines should be terminated with a CR-LF
398 * sequence, but we're nt piccky if the CR is missing.
399 * No leading dot escaping is done as the protcol doesn't
400 * use leading dots anywhere.
404 * Pointer to null-terminated buffer allocated by memget.
405 * *SIZE is set to the length of the buffer.
410 irs_irp_read_body(struct irp_p
*pvt
, size_t *size
) {
413 size_t len
= LINEINCR
;
414 char *buffer
= memget(len
);
421 if (irs_irp_read_line(pvt
, line
, sizeof line
) <= 0 ||
422 strchr(line
, '\n') == NULL
)
425 linelen
= strlen(line
);
427 if (line
[linelen
- 1] != '\n')
430 /* We're not strict about missing \r. Should we be?? */
431 if (linelen
> 2 && line
[linelen
- 2] == '\r') {
432 line
[linelen
- 2] = '\n';
433 line
[linelen
- 1] = '\0';
437 if (linelen
== 2 && line
[0] == '.') {
444 if (linelen
> (len
- (idx
+ 1))) {
445 char *p
= memget(len
+ LINEINCR
);
449 memcpy(p
, buffer
, len
);
455 memcpy(buffer
+ idx
, line
, linelen
);
464 * int irs_irp_get_full_response(struct irp_p *pvt, int *code,
465 * char **body, size_t *bodylen);
467 * Gets the response to a command. If the response indicates
468 * there's a body to follow(code % 10 == 1), then the
469 * body buffer is allcoated with memget and stored in
470 * *BODY. The length of the allocated body buffer is stored
471 * in *BODY. The caller must give the body buffer back to
472 * memput when done. The results code is stored in *CODE.
476 * 0 if a result was read. -1 on some sort of failure.
481 irs_irp_get_full_response(struct irp_p
*pvt
, int *code
, char *text
,
482 size_t textlen
, char **body
, size_t *bodylen
) {
483 int result
= irs_irp_read_response(pvt
, text
, textlen
);
493 /* Code that matches 2xx is a good result code.
494 * Code that matches xx1 means there's a response body coming.
496 if ((result
/ 100) == 2 && (result
% 10) == 1) {
497 *body
= irs_irp_read_body(pvt
, bodylen
);
507 * int irs_irp_send_command(struct irp_p *pvt, const char *fmt, ...);
509 * Sends command to remote connected via the PVT
510 * structure. FMT and args after it are fprintf-like
511 * arguments for formatting.
515 * 0 on success, -1 on failure.
519 irs_irp_send_command(struct irp_p
*pvt
, const char *fmt
, ...) {
526 if (pvt
->fdCxn
< 0) {
531 (void) vsprintf(buffer
, fmt
, ap
);
532 todo
= strlen(buffer
);
534 if (todo
> (int)sizeof(buffer
) - 3) {
535 syslog(LOG_CRIT
, "memory overrun in irs_irp_send_command()");
538 strcat(buffer
, "\r\n");
539 todo
= strlen(buffer
);
542 i
= write(pvt
->fdCxn
, buffer
+ pos
, todo
);
545 fprintf(stderr
, "Wrote: \"");
546 fwrite(buffer
+ pos
, sizeof (char), todo
, stderr
);
547 fprintf(stderr
, "\"\n");
564 * void irp_close(struct irs_acc *this)
569 irp_close(struct irs_acc
*this) {
570 struct irp_p
*irp
= (struct irp_p
*)this->private;
573 irs_irp_disconnect(irp
);
574 memput(irp
, sizeof *irp
);
577 memput(this, sizeof *this);