1 /* 2 * CDDL HEADER START 3 * 4 * This file and its contents are supplied under the terms of the 5 * Common Development and Distribution License ("CDDL"), version 1.0. 6 * You may only use this file in accordance with the terms of version 7 * 1.0 of the CDDL. 8 * 9 * A full copy of the text of the CDDL should have accompanied this 10 * source. A copy of the CDDL is also available via the Internet at 11 * http://www.illumos.org/license/CDDL. 12 * 13 * CDDL HEADER END 14 */ 15 /* 16 * Copyright (c) 2015 by Delphix. All rights reserved. 17 */ 18 19 #include <err.h> 20 #include <stdio.h> 21 #include <stdlib.h> 22 #include <errno.h> 23 #include <fcntl.h> 24 #include <strings.h> 25 #include <unistd.h> 26 #include <stropts.h> 27 #include <sys/debug.h> 28 #include <sys/tihdr.h> 29 #include "connstat.h" 30 31 int 32 mibopen(const char *proto) 33 { 34 int saved; 35 int fd; 36 37 fd = open("/dev/arp", O_RDWR); 38 if (fd == -1) { 39 return (-1); 40 } 41 42 if (ioctl(fd, I_PUSH, proto) == -1) { 43 saved = errno; 44 (void) close(fd); 45 errno = saved; 46 return (-1); 47 } 48 49 return (fd); 50 } 51 52 int 53 conn_walk(int fd, connstat_proto_t *proto, conn_walk_state_t *state) 54 { 55 struct strbuf cbuf, dbuf; 56 struct opthdr *hdr; 57 int flags, r, err = 0; 58 struct { 59 struct T_optmgmt_req req; 60 struct opthdr hdr; 61 } req; 62 union { 63 struct T_optmgmt_ack ack; 64 uint8_t space[sizeof (struct T_optmgmt_ack) + 65 sizeof (struct opthdr) * 2]; 66 } ack; 67 68 bzero(&cbuf, sizeof (cbuf)); 69 bzero(&dbuf, sizeof (dbuf)); 70 71 req.req.PRIM_type = T_OPTMGMT_REQ; 72 req.req.OPT_offset = (caddr_t)&req.hdr - (caddr_t)&req; 73 req.req.OPT_length = sizeof (req.hdr); 74 req.req.MGMT_flags = T_CURRENT; 75 76 req.hdr.level = proto->csp_miblevel; 77 req.hdr.name = 0; 78 req.hdr.len = 0; 79 80 cbuf.buf = (caddr_t)&req; 81 cbuf.len = sizeof (req); 82 83 if (putmsg(fd, &cbuf, NULL, 0) == -1) { 84 warn("failed to request connection info: putmsg"); 85 return (-1); 86 } 87 88 /* 89 * Each reply consists of a control part for one fixed structure or 90 * table, as defined in mib2.h. The format is a T_OPTMGMT_ACK 91 * containing an opthdr structure. The level and name identify the 92 * entry, and len is the size of the data part of the message. 93 */ 94 for (;;) { 95 cbuf.buf = (caddr_t)&ack; 96 cbuf.maxlen = sizeof (ack); 97 flags = 0; 98 99 /* 100 * We first do a getmsg() for the control part so that we 101 * can allocate a properly sized buffer to read the data 102 * part. 103 */ 104 do { 105 r = getmsg(fd, &cbuf, NULL, &flags); 106 } while (r < 0 && errno == EINTR); 107 108 if (r < 0) { 109 warn("failed to fetch further connection info"); 110 err = -1; 111 break; 112 } else if ((r & MORECTL) != 0) { 113 warnx("failed to fetch full control message"); 114 err = -1; 115 break; 116 } 117 118 if (cbuf.len < sizeof (struct T_optmgmt_ack) || 119 ack.ack.PRIM_type != T_OPTMGMT_ACK || 120 ack.ack.MGMT_flags != T_SUCCESS || 121 ack.ack.OPT_length < sizeof (struct opthdr)) { 122 warnx("cannot process invalid message from getmsg()"); 123 err = -1; 124 break; 125 } 126 127 /* LINTED E_BAD_PTR_CAST_ALIGN */ 128 hdr = (struct opthdr *)((caddr_t)&ack + ack.ack.OPT_offset); 129 if (r == 0 && hdr->level == 0 && hdr->name == 0) { 130 /* 131 * snmpcom_req() has sent us the final End-Of-Data 132 * message, so there's nothing further to read. 133 */ 134 break; 135 } 136 137 /* Only data should remain. */ 138 VERIFY3S(r, ==, MOREDATA); 139 140 /* Allocate a buffer to hold the data portion of the message */ 141 if ((dbuf.buf = realloc(dbuf.buf, hdr->len)) == NULL) { 142 warn("failed to realloc() buffer"); 143 err = -1; 144 break; 145 } 146 dbuf.maxlen = hdr->len; 147 dbuf.len = 0; 148 flags = 0; 149 150 do { 151 r = getmsg(fd, NULL, &dbuf, &flags); 152 } while (r < 0 && errno == EINTR); 153 154 if (r < 0) { 155 warn("failed to fetch connection data: getmsg()"); 156 err = -1; 157 break; 158 } else if (r != 0) { 159 warnx("failed to fetch all data: " 160 "getmsg() returned %d", r); 161 err = -1; 162 break; 163 } 164 165 if ((state->cws_flags & CS_IPV4) && 166 hdr->name == proto->csp_mibv4name) { 167 proto->csp_v4walk(&dbuf, state); 168 } else if ((state->cws_flags & CS_IPV6) && 169 hdr->name == proto->csp_mibv6name) { 170 proto->csp_v6walk(&dbuf, state); 171 } 172 } 173 174 free(dbuf.buf); 175 176 return (err); 177 } 178