diff options
author | Stef Walter <stef@memberwebs.com> | 2006-01-27 00:45:54 +0000 |
---|---|---|
committer | Stef Walter <stef@memberwebs.com> | 2006-01-27 00:45:54 +0000 |
commit | a4f6f05bd37adf54d84561dab6b11365c0e2b860 (patch) | |
tree | d7c3220f9bbf8d11d12bea6f80e190aa78c6d1af | |
parent | 965ffa71d8352d53b48d08ffd7fee3e4bcafcfa6 (diff) |
Move rrdbot code into right project.
-rw-r--r-- | src/Makefile.am | 11 | ||||
-rw-r--r-- | src/bsnmp/Makefile.am | 4 | ||||
-rw-r--r-- | src/bsnmp/asn1.c | 1000 | ||||
-rw-r--r-- | src/bsnmp/asn1.h | 183 | ||||
-rw-r--r-- | src/bsnmp/snmp.c | 1081 | ||||
-rw-r--r-- | src/bsnmp/snmp.h | 174 | ||||
-rw-r--r-- | src/bsnmp/snmppriv.h | 45 | ||||
-rw-r--r-- | src/common/hash.c | 380 | ||||
-rw-r--r-- | src/common/hash.h | 151 | ||||
-rw-r--r-- | src/common/server-mainloop.c | 424 | ||||
-rw-r--r-- | src/common/server-mainloop.h | 27 | ||||
-rw-r--r-- | src/common/sock-any.c | 385 | ||||
-rw-r--r-- | src/common/sock-any.h | 90 | ||||
-rw-r--r-- | src/common/stringx.c | 155 | ||||
-rw-r--r-- | src/common/stringx.h | 53 | ||||
-rw-r--r-- | src/config.c | 575 | ||||
-rw-r--r-- | src/rrd-update.c | 134 | ||||
-rw-r--r-- | src/rrdbotd.c | 239 | ||||
-rw-r--r-- | src/rrdbotd.h | 182 | ||||
-rw-r--r-- | src/snmp-engine.c | 636 | ||||
-rw-r--r-- | src/snmp-help.c | 111 | ||||
-rw-r--r-- | src/snmpclient.c | 967 | ||||
-rw-r--r-- | src/snmpclient.h | 187 | ||||
-rw-r--r-- | src/usuals.h | 77 |
24 files changed, 0 insertions, 7271 deletions
diff --git a/src/Makefile.am b/src/Makefile.am deleted file mode 100644 index fa8144a..0000000 --- a/src/Makefile.am +++ /dev/null @@ -1,11 +0,0 @@ -SUBDIRS = bsnmp - -sbin_PROGRAMS = rrdbotd - -rrdbotd_SOURCES = rrdbotd.c rrdbotd.h config.c \ - snmp-help.c snmp-engine.c rrd-update.c \ - common/server-mainloop.c common/server-mainloop.h \ - common/sock-any.h common/sock-any.c \ - usuals.h common/stringx.h common/stringx.c common/hash.h common/hash.c -rrdbotd_CFLAGS = -I${top_srcdir}/src/common/ -I${top_srcdir} -Ibsnmp -rrdbotd_LDADD = $(top_builddir)/src/bsnmp/libbsnmp-custom.a diff --git a/src/bsnmp/Makefile.am b/src/bsnmp/Makefile.am deleted file mode 100644 index 7a7a739..0000000 --- a/src/bsnmp/Makefile.am +++ /dev/null @@ -1,4 +0,0 @@ - -noinst_LIBRARIES = libbsnmp-custom.a -libbsnmp_custom_a_SOURCES = asn1.c asn1.h snmp.c snmp.h snmppriv.h - diff --git a/src/bsnmp/asn1.c b/src/bsnmp/asn1.c deleted file mode 100644 index 8a206ee..0000000 --- a/src/bsnmp/asn1.c +++ /dev/null @@ -1,1000 +0,0 @@ -/* - * Copyright (c) 2001-2003 - * Fraunhofer Institute for Open Communication Systems (FhG Fokus). - * All rights reserved. - * - * Author: Harti Brandt <harti@freebsd.org> - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * 1. Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * 2. Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * - * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND - * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE - * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE - * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL - * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS - * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) - * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT - * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY - * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - * - * $Begemot: bsnmp/lib/asn1.c,v 1.28 2004/08/06 08:46:49 brandt Exp $ - * - * ASN.1 for SNMP. - */ -#include <sys/types.h> -#include <stdio.h> -#include <stdlib.h> -#include <stdarg.h> -#include <string.h> -#include <stdint.h> -#include <assert.h> -#include "asn1.h" - -static void asn_error_func(const struct asn_buf *, const char *, ...); - -void (*asn_error)(const struct asn_buf *, const char *, ...) = asn_error_func; - -/* - * Read the next header. This reads the tag (note, that only single - * byte tags are supported for now) and the length field. The length field - * is restricted to a 32-bit value. - * All errors of this function stop the decoding. - */ -enum asn_err -asn_get_header(struct asn_buf *b, u_char *type, asn_len_t *len) -{ - u_int length; - - if (b->asn_len == 0) { - asn_error(b, "no identifier for header"); - return (ASN_ERR_EOBUF); - } - *type = *b->asn_cptr; - if ((*type & ASN_TYPE_MASK) > 0x30) { - asn_error(b, "types > 0x30 not supported (%u)", - *type & ASN_TYPE_MASK); - return (ASN_ERR_FAILED); - } - b->asn_cptr++; - b->asn_len--; - if (b->asn_len == 0) { - asn_error(b, "no length field"); - return (ASN_ERR_EOBUF); - } - if (*b->asn_cptr & 0x80) { - length = *b->asn_cptr++ & 0x7f; - b->asn_len--; - if (length == 0) { - asn_error(b, "indefinite length not supported"); - return (ASN_ERR_FAILED); - } - if (length > ASN_MAXLENLEN) { - asn_error(b, "long length too long (%u)", length); - return (ASN_ERR_FAILED); - } - if (length > b->asn_len) { - asn_error(b, "long length truncated"); - return (ASN_ERR_EOBUF); - } - *len = 0; - while (length--) { - *len = (*len << 8) | *b->asn_cptr++; - b->asn_len--; - } - } else { - *len = *b->asn_cptr++; - b->asn_len--; - } - return (ASN_ERR_OK); -} - -/* - * Write a length field (restricted to values < 2^32-1) and return the - * number of bytes this field takes. If ptr is NULL, the length is computed - * but nothing is written. If the length would be too large return 0. - */ -static u_int -asn_put_len(u_char *ptr, asn_len_t len) -{ - u_int lenlen, lenlen1; - asn_len_t tmp; - - if (len > ASN_MAXLEN) { - asn_error(NULL, "encoding length too long: (%u)", len); - return (0); - } - - if (len <= 127) { - if (ptr) - *ptr++ = (u_char)len; - return (1); - } else { - lenlen = 0; - /* compute number of bytes for value (is at least 1) */ - for (tmp = len; tmp != 0; tmp >>= 8) - lenlen++; - if (ptr != NULL) { - *ptr++ = (u_char)lenlen | 0x80; - lenlen1 = lenlen; - while (lenlen1-- > 0) { - ptr[lenlen1] = len & 0xff; - len >>= 8; - } - } - return (lenlen + 1); - } -} - -/* - * Write a header (tag and length fields). - * Tags are restricted to one byte tags (value <= 0x30) and the - * lenght field to 16-bit. All errors stop the encoding. - */ -enum asn_err -asn_put_header(struct asn_buf *b, u_char type, asn_len_t len) -{ - u_int lenlen; - - /* tag field */ - if ((type & ASN_TYPE_MASK) > 0x30) { - asn_error(NULL, "types > 0x30 not supported (%u)", - type & ASN_TYPE_MASK); - return (ASN_ERR_FAILED); - } - if (b->asn_len == 0) - return (ASN_ERR_EOBUF); - - *b->asn_ptr++ = type; - b->asn_len--; - - /* length field */ - if ((lenlen = asn_put_len(NULL, len)) == 0) - return (ASN_ERR_FAILED); - if (b->asn_len < lenlen) - return (ASN_ERR_EOBUF); - - (void)asn_put_len(b->asn_ptr, len); - b->asn_ptr += lenlen; - b->asn_len -= lenlen; - return (ASN_ERR_OK); -} - - -/* - * This constructs a temporary sequence header with space for the maximum - * length field (three byte). Set the pointer that ptr points to to the - * start of the encoded header. This is used for a later call to - * asn_commit_header which will fix-up the length field and move the - * value if needed. All errors should stop the encoding. - */ -#define TEMP_LEN (1 + ASN_MAXLENLEN + 1) -enum asn_err -asn_put_temp_header(struct asn_buf *b, u_char type, u_char **ptr) -{ - int ret; - - if (b->asn_len < TEMP_LEN) - return (ASN_ERR_EOBUF); - *ptr = b->asn_ptr; - if ((ret = asn_put_header(b, type, ASN_MAXLEN)) == ASN_ERR_OK) - assert(b->asn_ptr == *ptr + TEMP_LEN); - return (ret); -} -enum asn_err -asn_commit_header(struct asn_buf *b, u_char *ptr) -{ - asn_len_t len; - u_int lenlen, shift; - - /* compute length of encoded value without header */ - len = b->asn_ptr - (ptr + TEMP_LEN); - - /* insert length. may not fail. */ - lenlen = asn_put_len(ptr + 1, len); - if (lenlen > TEMP_LEN - 1) - return (ASN_ERR_FAILED); - - if (lenlen < TEMP_LEN - 1) { - /* shift value down */ - shift = (TEMP_LEN - 1) - lenlen; - memmove(ptr + 1 + lenlen, ptr + TEMP_LEN, len); - b->asn_ptr -= shift; - b->asn_len += shift; - } - return (ASN_ERR_OK); -} -#undef TEMP_LEN - -/* - * BER integer. This may be used to get a signed 64 bit integer at maximum. - * The maximum length should be checked by the caller. This cannot overflow - * if the caller ensures that len is at maximum 8. - * - * <bytes> - */ -static enum asn_err -asn_get_real_integer(struct asn_buf *b, asn_len_t len, int64_t *vp) -{ - uint64_t val; - int neg = 0; - enum asn_err err; - - if (b->asn_len < len) { - asn_error(b, "truncated integer"); - return (ASN_ERR_EOBUF); - } - if (len == 0) { - asn_error(b, "zero-length integer"); - *vp = 0; - return (ASN_ERR_BADLEN); - } - err = ASN_ERR_OK; - if (len > 8) - err = ASN_ERR_RANGE; - else if (len > 1 && - ((*b->asn_cptr == 0x00 && (b->asn_cptr[1] & 0x80) == 0) || - (*b->asn_cptr == 0xff && (b->asn_cptr[1] & 0x80) == 0x80))) { - asn_error(b, "non-minimal integer"); - err = ASN_ERR_BADLEN; - } - - if (*b->asn_cptr & 0x80) - neg = 1; - val = 0; - while (len--) { - val <<= 8; - val |= neg ? (u_char)~*b->asn_cptr : *b->asn_cptr; - b->asn_len--; - b->asn_cptr++; - } - if (neg) { - *vp = -(int64_t)val - 1; - } else - *vp = (int64_t)val; - return (err); -} - -/* - * Write a signed integer with the given type. The caller has to ensure - * that the actual value is ok for this type. - */ -static enum asn_err -asn_put_real_integer(struct asn_buf *b, u_char type, int64_t ival) -{ - int i, neg = 0; -# define OCTETS 8 - u_char buf[OCTETS]; - uint64_t val; - enum asn_err ret; - - if (ival < 0) { - /* this may fail if |INT64_MIN| > |INT64_MAX| and - * the value is between * INT64_MIN <= ival < -(INT64_MAX+1) */ - val = (uint64_t)-(ival + 1); - neg = 1; - } else - val = (uint64_t)ival; - - /* split the value into octets */ - for (i = OCTETS - 1; i >= 0; i--) { - buf[i] = val & 0xff; - if (neg) - buf[i] = ~buf[i]; - val >>= 8; - } - /* no leading 9 zeroes or ones */ - for (i = 0; i < OCTETS - 1; i++) - if (!((buf[i] == 0xff && (buf[i + 1] & 0x80) != 0) || - (buf[i] == 0x00 && (buf[i + 1] & 0x80) == 0))) - break; - if ((ret = asn_put_header(b, type, OCTETS - i))) - return (ret); - if (OCTETS - (u_int)i > b->asn_len) - return (ASN_ERR_EOBUF); - - while (i < OCTETS) { - *b->asn_ptr++ = buf[i++]; - b->asn_len--; - } - return (ASN_ERR_OK); -# undef OCTETS -} - - -/* - * The same for unsigned 64-bitters. Here we have the problem, that overflow - * can happen, because the value maybe 9 bytes long. In this case the - * first byte must be 0. - */ -static enum asn_err -asn_get_real_unsigned(struct asn_buf *b, asn_len_t len, uint64_t *vp) -{ - enum asn_err err; - - if (b->asn_len < len) { - asn_error(b, "truncated integer"); - return (ASN_ERR_EOBUF); - } - if (len == 0) { - asn_error(b, "zero-length integer"); - *vp = 0; - return (ASN_ERR_BADLEN); - } - err = ASN_ERR_OK; - *vp = 0; - if ((*b->asn_cptr & 0x80) || (len == 9 && *b->asn_cptr != 0)) { - /* negative integer or too larger */ - *vp = 0xffffffffffffffffULL; - err = ASN_ERR_RANGE; - } else if (len > 1 && - *b->asn_cptr == 0x00 && (b->asn_cptr[1] & 0x80) == 0) { - asn_error(b, "non-minimal unsigned"); - err = ASN_ERR_BADLEN; - } - - while (len--) { - *vp = (*vp << 8) | *b->asn_cptr++; - b->asn_len--; - } - return (err); -} - - -/* - * Values with the msb on need 9 octets. - */ -static int -asn_put_real_unsigned(struct asn_buf *b, u_char type, uint64_t val) -{ - int i; -# define OCTETS 9 - u_char buf[OCTETS]; - enum asn_err ret; - - /* split the value into octets */ - for (i = OCTETS - 1; i >= 0; i--) { - buf[i] = val & 0xff; - val >>= 8; - } - /* no leading 9 zeroes */ - for (i = 0; i < OCTETS - 1; i++) - if (!(buf[i] == 0x00 && (buf[i + 1] & 0x80) == 0)) - break; - if ((ret = asn_put_header(b, type, OCTETS - i))) - return (ret); - if (OCTETS - (u_int)i > b->asn_len) - return (ASN_ERR_EOBUF); - - while (i < OCTETS) { - *b->asn_ptr++ = buf[i++]; - b->asn_len--; - } -#undef OCTETS - return (ASN_ERR_OK); -} - -/* - * The ASN.1 INTEGER type is restricted to 32-bit signed by the SMI. - */ -enum asn_err -asn_get_integer_raw(struct asn_buf *b, asn_len_t len, int32_t *vp) -{ - int64_t val; - enum asn_err ret; - - if ((ret = asn_get_real_integer(b, len, &val)) == ASN_ERR_OK) { - if (len > 4) - ret = ASN_ERR_BADLEN; - else if (val > INT32_MAX || val < INT32_MIN) - /* may not happen */ - ret = ASN_ERR_RANGE; - *vp = (int32_t)val; - } - return (ret); -} - -enum asn_err -asn_get_integer(struct asn_buf *b, int32_t *vp) -{ - asn_len_t len; - u_char type; - enum asn_err err; - - if ((err = asn_get_header(b, &type, &len)) != ASN_ERR_OK) - return (err); - if (type != ASN_TYPE_INTEGER) { - asn_error(b, "bad type for integer (%u)", type); - return (ASN_ERR_TAG); - } - - return (asn_get_integer_raw(b, len, vp)); -} - -enum asn_err -asn_put_integer(struct asn_buf *b, int32_t val) -{ - return (asn_put_real_integer(b, ASN_TYPE_INTEGER, val)); -} - -/* - * OCTETSTRING - * - * <0x04> <len> <data ...> - * - * Get an octetstring. noctets must point to the buffer size and on - * return will contain the size of the octetstring, regardless of the - * buffer size. - */ -enum asn_err -asn_get_octetstring_raw(struct asn_buf *b, asn_len_t len, u_char *octets, - u_int *noctets) -{ - enum asn_err err = ASN_ERR_OK; - - if (*noctets < len) { - asn_error(b, "octetstring truncated"); - err = ASN_ERR_RANGE; - } - if (b->asn_len < len) { - asn_error(b, "truncatet octetstring"); - return (ASN_ERR_EOBUF); - } - if (*noctets < len) - memcpy(octets, b->asn_cptr, *noctets); - else - memcpy(octets, b->asn_cptr, len); - *noctets = len; - b->asn_cptr += len; - b->asn_len -= len; - return (err); -} - -enum asn_err -asn_get_octetstring(struct asn_buf *b, u_char *octets, u_int *noctets) -{ - enum asn_err err; - u_char type; - asn_len_t len; - - if ((err = asn_get_header(b, &type, &len)) != ASN_ERR_OK) - return (err); - if (type != ASN_TYPE_OCTETSTRING) { - asn_error(b, "bad type for octetstring (%u)", type); - return (ASN_ERR_TAG); - } - return (asn_get_octetstring_raw(b, len, octets, noctets)); -} - -enum asn_err -asn_put_octetstring(struct asn_buf *b, const u_char *octets, u_int noctets) -{ - enum asn_err ret; - - if ((ret = asn_put_header(b, ASN_TYPE_OCTETSTRING, noctets)) != ASN_ERR_OK) - return (ret); - if (b->asn_len < noctets) - return (ASN_ERR_EOBUF); - - memcpy(b->asn_ptr, octets, noctets); - b->asn_ptr += noctets; - b->asn_len -= noctets; - return (ASN_ERR_OK); -} - -/* - * NULL - * - * <0x05> <0x00> - */ -enum asn_err -asn_get_null_raw(struct asn_buf *b, asn_len_t len) -{ - if (len != 0) { - if (b->asn_len < len) { - asn_error(b, "truncated NULL"); - return (ASN_ERR_EOBUF); - } - asn_error(b, "bad length for NULL (%u)", len); - b->asn_len -= len; - b->asn_ptr += len; - return (ASN_ERR_BADLEN); - } - return (ASN_ERR_OK); -} - -enum asn_err -asn_get_null(struct asn_buf *b) -{ - u_char type; - asn_len_t len; - enum asn_err err; - - if ((err = asn_get_header(b, &type, &len)) != ASN_ERR_OK) - return (err); - if (type != ASN_TYPE_NULL) { - asn_error(b, "bad type for NULL (%u)", type); - return (ASN_ERR_TAG); - } - return (asn_get_null_raw(b, len)); -} - -enum asn_err -asn_put_null(struct asn_buf *b) -{ - return (asn_put_header(b, ASN_TYPE_NULL, 0)); -} - -enum asn_err -asn_put_exception(struct asn_buf *b, u_int except) -{ - return (asn_put_header(b, ASN_CLASS_CONTEXT | except, 0)); -} - -/* - * OBJID - * - * <0x06> <len> <subid...> - */ -enum asn_err -asn_get_objid_raw(struct asn_buf *b, asn_len_t len, struct asn_oid *oid) -{ - asn_subid_t subid; - enum asn_err err; - - if (b->asn_len < len) { - asn_error(b, "truncated OBJID"); - return (ASN_ERR_EOBUF); - } - oid->len = 0; - if (len == 0) { - asn_error(b, "short OBJID"); - oid->subs[oid->len++] = 0; - oid->subs[oid->len++] = 0; - return (ASN_ERR_BADLEN); - } - err = ASN_ERR_OK; - while (len != 0) { - if (oid->len == ASN_MAXOIDLEN) { - asn_error(b, "OID too long (%u)", oid->len); - b->asn_cptr += len; - b->asn_len -= len; - return (ASN_ERR_BADLEN); - } - subid = 0; - do { - if (len == 0) { - asn_error(b, "unterminated subid"); - return (ASN_ERR_EOBUF); - } - if (subid > (ASN_MAXID >> 7)) { - asn_error(b, "OBID subid too larger"); - err = ASN_ERR_RANGE; - } - subid = (subid << 7) | (*b->asn_cptr & 0x7f); - len--; - b->asn_len--; - } while (*b->asn_cptr++ & 0x80); - if (oid->len == 0) { - if (subid < 80) { - oid->subs[oid->len++] = subid / 40; - oid->subs[oid->len++] = subid % 40; - } else { - oid->subs[oid->len++] = 2; - oid->subs[oid->len++] = subid - 80; - } - } else { - oid->subs[oid->len++] = subid; - } - } - return (err); - -} - -enum asn_err -asn_get_objid(struct asn_buf *b, struct asn_oid *oid) -{ - u_char type; - asn_len_t len; - enum asn_err err; - - if ((err = asn_get_header(b, &type, &len)) != ASN_ERR_OK) - return (err); - if (type != ASN_TYPE_OBJID) { - asn_error(b, "bad type for OBJID (%u)", type); - return (ASN_ERR_TAG); - } - return (asn_get_objid_raw(b, len, oid)); -} - -enum asn_err -asn_put_objid(struct asn_buf *b, const struct asn_oid *oid) -{ - asn_subid_t first, sub; - enum asn_err err, err1; - u_int i, oidlen; - asn_len_t len; - - err = ASN_ERR_OK; - if (oid->len == 0) { - /* illegal */ - asn_error(NULL, "short oid"); - err = ASN_ERR_RANGE; - first = 0; - oidlen = 2; - } else if (oid->len == 1) { - /* illegal */ - asn_error(b, "short oid"); - if (oid->subs[0] > 2) - asn_error(NULL, "oid[0] too large (%u)", oid->subs[0]); - err = ASN_ERR_RANGE; - first = oid->subs[0] * 40; - oidlen = 2; - } else { - if (oid->len > ASN_MAXOIDLEN) { - asn_error(NULL, "oid too long %u", oid->len); - err = ASN_ERR_RANGE; - } - if (oid->subs[0] > 2 || - (oid->subs[0] < 2 && oid->subs[0] >= 40)) { - asn_error(NULL, "oid out of range (%u,%u)", - oid->subs[0], oid->subs[1]); - err = ASN_ERR_RANGE; - } - first = 40 * oid->subs[0] + oid->subs[1]; - oidlen = oid->len; - } - len = 0; - for (i = 1; i < oidlen; i++) { - sub = (i == 1) ? first : oid->subs[i]; - if (sub > ASN_MAXID) { - asn_error(NULL, "oid subid too large"); - err = ASN_ERR_RANGE; - } - len += (sub <= 0x7f) ? 1 - : (sub <= 0x3fff) ? 2 - : (sub <= 0x1fffff) ? 3 - : (sub <= 0xfffffff) ? 4 - : 5; - } - if ((err1 = asn_put_header(b, ASN_TYPE_OBJID, len)) != ASN_ERR_OK) - return (err1); - if (b->asn_len < len) - return (ASN_ERR_EOBUF); - - for (i = 1; i < oidlen; i++) { - sub = (i == 1) ? first : oid->subs[i]; - if (sub <= 0x7f) { - *b->asn_ptr++ = sub; - b->asn_len--; - } else if (sub <= 0x3fff) { - *b->asn_ptr++ = (sub >> 7) | 0x80; - *b->asn_ptr++ = sub & 0x7f; - b->asn_len -= 2; - } else if (sub <= 0x1fffff) { - *b->asn_ptr++ = (sub >> 14) | 0x80; - *b->asn_ptr++ = ((sub >> 7) & 0x7f) | 0x80; - *b->asn_ptr++ = sub & 0x7f; - b->asn_len -= 3; - } else if (sub <= 0xfffffff) { - *b->asn_ptr++ = (sub >> 21) | 0x80; - *b->asn_ptr++ = ((sub >> 14) & 0x7f) | 0x80; - *b->asn_ptr++ = ((sub >> 7) & 0x7f) | 0x80; - *b->asn_ptr++ = sub & 0x7f; - b->asn_len -= 4; - } else { - *b->asn_ptr++ = (sub >> 28) | 0x80; - *b->asn_ptr++ = ((sub >> 21) & 0x7f) | 0x80; - *b->asn_ptr++ = ((sub >> 14) & 0x7f) | 0x80; - *b->asn_ptr++ = ((sub >> 7) & 0x7f) | 0x80; - *b->asn_ptr++ = sub & 0x7f; - b->asn_len -= 5; - } - } - return (err); -} -/* - * SEQUENCE header - * - * <0x10|0x20> <len> <data...> - */ -enum asn_err -asn_get_sequence(struct asn_buf *b, asn_len_t *len) -{ - u_char type; - enum asn_err err; - - if ((err = asn_get_header(b, &type, len)) != ASN_ERR_OK) - return (err); - if (type != (ASN_TYPE_SEQUENCE|ASN_TYPE_CONSTRUCTED)) { - asn_error(b, "bad sequence type %u", type); - return (ASN_ERR_TAG); - } - if (*len > b->asn_len) { - asn_error(b, "truncated sequence"); - return (ASN_ERR_EOBUF); - } - return (ASN_ERR_OK); -} - -/* - * Application types - * - * 0x40 4 MSB 2MSB 2LSB LSB - */ -enum asn_err -asn_get_ipaddress_raw(struct asn_buf *b, asn_len_t len, u_char *addr) -{ - u_int i; - - if (b->asn_len < len) { - asn_error(b, "truncated ip-address"); - return (ASN_ERR_EOBUF); - } - if (len < 4) { - asn_error(b, "short length for ip-Address %u", len); - for (i = 0; i < len; i++) - *addr++ = *b->asn_cptr++; - while (i++ < len) - *addr++ = 0; - b->asn_len -= len; - return (ASN_ERR_BADLEN); - } - for (i = 0; i < 4; i++) - *addr++ = *b->asn_cptr++; - b->asn_cptr += len - 4; - b->asn_len -= len; - return (ASN_ERR_OK); -} - -enum asn_err -asn_get_ipaddress(struct asn_buf *b, u_char *addr) -{ - u_char type; - asn_len_t len; - enum asn_err err; - - if ((err = asn_get_header(b, &type, &len)) != ASN_ERR_OK) - return (err); - if (type != (ASN_CLASS_APPLICATION|ASN_APP_IPADDRESS)) { - asn_error(b, "bad type for ip-address %u", type); - return (ASN_ERR_TAG); - } - return (asn_get_ipaddress_raw(b, len, addr)); -} - -enum asn_err -asn_put_ipaddress(struct asn_buf *b, const u_char *addr) -{ - enum asn_err err; - - if ((err = asn_put_header(b, ASN_CLASS_APPLICATION|ASN_APP_IPADDRESS, - 4)) != ASN_ERR_OK) - return (err); - if (b->asn_len < 4) - return (ASN_ERR_EOBUF); - - memcpy(b->asn_ptr, addr, 4); - b->asn_ptr += 4; - b->asn_len -= 4; - return (ASN_ERR_OK); -} - - -/* - * UNSIGNED32 - * - * 0x42|0x41 <len> ... - */ -enum asn_err -asn_get_uint32_raw(struct asn_buf *b, asn_len_t len, uint32_t *vp) -{ - uint64_t v; - enum asn_err err; - - if ((err = asn_get_real_unsigned(b, len, &v)) == ASN_ERR_OK) { - if (len > 5) { - asn_error(b, "uint32 too long %u", len); - err = ASN_ERR_BADLEN; - } else if (v > UINT32_MAX) { - asn_error(b, "uint32 too large %llu", v); - err = ASN_ERR_RANGE; - } - *vp = (uint32_t)v; - } - return (err); -} - -enum asn_err -asn_put_uint32(struct asn_buf *b, u_char type, uint32_t val) -{ - uint64_t v = val; - - return (asn_put_real_unsigned(b, ASN_CLASS_APPLICATION|type, v)); -} - -/* - * COUNTER64 - * 0x46 <len> ... - */ -enum asn_err -asn_get_counter64_raw(struct asn_buf *b, asn_len_t len, uint64_t *vp) -{ - return (asn_get_real_unsigned(b, len, vp)); -} - -enum asn_err -asn_put_counter64(struct asn_buf *b, uint64_t val) -{ - return (asn_put_real_unsigned(b, - ASN_CLASS_APPLICATION | ASN_APP_COUNTER64, val)); -} - -/* - * TimeTicks - * 0x43 <len> ... - */ -enum asn_err -asn_get_timeticks(struct asn_buf *b, uint32_t *vp) -{ - asn_len_t len; - u_char type; - enum asn_err err; - - if ((err = asn_get_header(b, &type, &len)) != ASN_ERR_OK) - return (err); - if (type != (ASN_CLASS_APPLICATION|ASN_APP_TIMETICKS)) { - asn_error(b, "bad type for timeticks %u", type); - return (ASN_ERR_TAG); - } - return (asn_get_uint32_raw(b, len, vp)); -} - -enum asn_err -asn_put_timeticks(struct asn_buf *b, uint32_t val) -{ - uint64_t v = val; - - return (asn_put_real_unsigned(b, - ASN_CLASS_APPLICATION | ASN_APP_TIMETICKS, v)); -} - -/* - * Construct a new OID by taking a range of sub ids of the original oid. - */ -void -asn_slice_oid(struct asn_oid *dest, const struct asn_oid *src, - u_int from, u_int to) -{ - if (from >= to) { - dest->len = 0; - return; - } - dest->len = to - from; - memcpy(dest->subs, &src->subs[from], dest->len * sizeof(dest->subs[0])); -} - -/* - * Append from to to - */ -void -asn_append_oid(struct asn_oid *to, const struct asn_oid *from) -{ - memcpy(&to->subs[to->len], &from->subs[0], - from->len * sizeof(from->subs[0])); - to->len += from->len; -} - -/* - * Skip a value - */ -enum asn_err -asn_skip(struct asn_buf *b, asn_len_t len) -{ - if (b->asn_len < len) - return (ASN_ERR_EOBUF); - b->asn_cptr += len; - b->asn_len -= len; - return (ASN_ERR_OK); -} - -/* - * Compare two OIDs. - * - * o1 < o2 : -1 - * o1 > o2 : +1 - * o1 = o2 : 0 - */ -int -asn_compare_oid(const struct asn_oid *o1, const struct asn_oid *o2) -{ - u_long i; - - for (i = 0; i < o1->len && i < o2->len; i++) { - if (o1->subs[i] < o2->subs[i]) - return (-1); - if (o1->subs[i] > o2->subs[i]) - return (+1); - } - if (o1->len < o2->len) - return (-1); - if (o1->len > o2->len) - return (+1); - return (0); -} - -/* - * Check whether an OID is a sub-string of another OID. - */ -int -asn_is_suboid(const struct asn_oid *o1, const struct asn_oid *o2) -{ - u_long i; - - for (i = 0; i < o1->len; i++) - if (i >= o2->len || o1->subs[i] != o2->subs[i]) - return (0); - return (1); -} - -/* - * Put a string representation of an oid into a user buffer. This buffer - * is assumed to be at least ASN_OIDSTRLEN characters long. - * - * sprintf is assumed not to fail here. - */ -char * -asn_oid2str_r(const struct asn_oid *oid, char *buf) -{ - u_int len, i; - char *ptr; - - if ((len = oid->len) > ASN_MAXOIDLEN) - len = ASN_MAXOIDLEN; - buf[0] = '\0'; - for (i = 0, ptr = buf; i < len; i++) { - if (i > 0) - *ptr++ = '.'; - ptr += sprintf(ptr, "%u", oid->subs[i]); - } - return (buf); -} - -/* - * Make a string from an OID in a private buffer. - */ -char * -asn_oid2str(const struct asn_oid *oid) -{ - static char str[ASN_OIDSTRLEN]; - - return (asn_oid2str_r(oid, str)); -} - - -static void -asn_error_func(const struct asn_buf *b, const char *err, ...) -{ - va_list ap; - u_long i; - - fprintf(stderr, "ASN.1: "); - va_start(ap, err); - vfprintf(stderr, err, ap); - va_end(ap); - - if (b != NULL) { - fprintf(stderr, " at"); - for (i = 0; b->asn_len > i; i++) - fprintf(stderr, " %02x", b->asn_cptr[i]); - } - fprintf(stderr, "\n"); -} diff --git a/src/bsnmp/asn1.h b/src/bsnmp/asn1.h deleted file mode 100644 index 23b7814..0000000 --- a/src/bsnmp/asn1.h +++ /dev/null @@ -1,183 +0,0 @@ -/* - * Copyright (c) 2001-2003 - * Fraunhofer Institute for Open Communication Systems (FhG Fokus). - * All rights reserved. - * - * Author: Harti Brandt <harti@freebsd.org> - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * 1. Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * 2. Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * - * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND - * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE - * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE - * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL - * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS - * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) - * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT - * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY - * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - * - * $Begemot: bsnmp/lib/asn1.h,v 1.18 2004/08/06 08:46:50 brandt Exp $ - * - * ASN.1 for SNMP - */ -#ifndef asn1_h_ -#define asn1_h_ - -#include <sys/types.h> -#include <stdint.h> - -struct asn_buf { - union { - u_char *ptr; - const u_char *cptr; - } asn_u; - size_t asn_len; -}; -#define asn_cptr asn_u.cptr -#define asn_ptr asn_u.ptr - -/* these restrictions are in the SMI */ -#define ASN_MAXID 0xffffffff -#define ASN_MAXOIDLEN 128 - -/* the string needed for this (with trailing zero) */ -#define ASN_OIDSTRLEN (ASN_MAXOIDLEN * (10 + 1) - 1 + 1) - -/* type of subidentifiers */ -typedef uint32_t asn_subid_t; - -struct asn_oid { - u_int len; - asn_subid_t subs[ASN_MAXOIDLEN]; -}; - -enum asn_err { - /* conversion was ok */ - ASN_ERR_OK = 0, - /* conversion failed and stopped */ - ASN_ERR_FAILED = 1 | 0x1000, - /* length field bad, value skipped */ - ASN_ERR_BADLEN = 2, - /* out of buffer, stopped */ - ASN_ERR_EOBUF = 3 | 0x1000, - /* length ok, but value is out of range */ - ASN_ERR_RANGE = 4, - /* not the expected tag, stopped */ - ASN_ERR_TAG = 5 | 0x1000, -}; -#define ASN_ERR_STOPPED(E) (((E) & 0x1000) != 0) - -/* type for the length field of encoded values. The length is restricted - * to 65535, but using uint16_t would give conversion warnings on gcc */ -typedef uint32_t asn_len_t; /* could be also uint16_t */ - -/* maximal length of a long length field without the length of the length */ -#define ASN_MAXLEN 65535 -#define ASN_MAXLENLEN 2 /* number of bytes in a length */ - -/* maximum size of an octet string as per SMIv2 */ -#define ASN_MAXOCTETSTRING 65535 - -extern void (*asn_error)(const struct asn_buf *, const char *, ...); - -enum asn_err asn_get_header(struct asn_buf *, u_char *, asn_len_t *); -enum asn_err asn_put_header(struct asn_buf *, u_char, asn_len_t); - -enum asn_err asn_put_temp_header(struct asn_buf *, u_char, u_char **); -enum asn_err asn_commit_header(struct asn_buf *, u_char *); - -enum asn_err asn_get_integer_raw(struct asn_buf *, asn_len_t, int32_t *); -enum asn_err asn_get_integer(struct asn_buf *, int32_t *); -enum asn_err asn_put_integer(struct asn_buf *, int32_t); - -enum asn_err asn_get_octetstring_raw(struct asn_buf *, asn_len_t, u_char *, u_int *); -enum asn_err asn_get_octetstring(struct asn_buf *, u_char *, u_int *); -enum asn_err asn_put_octetstring(struct asn_buf *, const u_char *, u_int); - -enum asn_err asn_get_null_raw(struct asn_buf *b, asn_len_t); -enum asn_err asn_get_null(struct asn_buf *); -enum asn_err asn_put_null(struct asn_buf *); - -enum asn_err asn_put_exception(struct asn_buf *, u_int); - -enum asn_err asn_get_objid_raw(struct asn_buf *, asn_len_t, struct asn_oid *); -enum asn_err asn_get_objid(struct asn_buf *, struct asn_oid *); -enum asn_err asn_put_objid(struct asn_buf *, const struct asn_oid *); - -enum asn_err asn_get_sequence(struct asn_buf *, asn_len_t *); - -enum asn_err asn_get_ipaddress_raw(struct asn_buf *, asn_len_t, u_char *); -enum asn_err asn_get_ipaddress(struct asn_buf *, u_char *); -enum asn_err asn_put_ipaddress(struct asn_buf *, const u_char *); - -enum asn_err asn_get_uint32_raw(struct asn_buf *, asn_len_t, uint32_t *); -enum asn_err asn_put_uint32(struct asn_buf *, u_char, uint32_t); - -enum asn_err asn_get_counter64_raw(struct asn_buf *, asn_len_t, uint64_t *); -enum asn_err asn_put_counter64(struct asn_buf *, uint64_t); - -enum asn_err asn_get_timeticks(struct asn_buf *, uint32_t *); -enum asn_err asn_put_timeticks(struct asn_buf *, uint32_t); - -enum asn_err asn_skip(struct asn_buf *, asn_len_t); - -/* - * Utility functions for OIDs - */ -/* get a sub-OID from the middle of another OID */ -void asn_slice_oid(struct asn_oid *, const struct asn_oid *, u_int, u_int); - -/* append an OID to another one */ -void asn_append_oid(struct asn_oid *, const struct asn_oid *); - -/* compare two OIDs */ -int asn_compare_oid(const struct asn_oid *, const struct asn_oid *); - -/* check whether the first is a suboid of the second one */ -int asn_is_suboid(const struct asn_oid *, const struct asn_oid *); - -/* format an OID into a user buffer of size ASN_OIDSTRLEN */ -char *asn_oid2str_r(const struct asn_oid *, char *); - -/* format an OID into a private static buffer */ -char *asn_oid2str(const struct asn_oid *); - -enum { - ASN_TYPE_BOOLEAN = 0x01, - ASN_TYPE_INTEGER = 0x02, - ASN_TYPE_BITSTRING = 0x03, - ASN_TYPE_OCTETSTRING = 0x04, - ASN_TYPE_NULL = 0x05, - ASN_TYPE_OBJID = 0x06, - ASN_TYPE_SEQUENCE = 0x10, - - ASN_TYPE_CONSTRUCTED = 0x20, - ASN_CLASS_UNIVERSAL = 0x00, - ASN_CLASS_APPLICATION = 0x40, - ASN_CLASS_CONTEXT = 0x80, - ASN_CLASS_PRIVATE = 0xc0, - ASN_TYPE_MASK = 0x1f, - - ASN_APP_IPADDRESS = 0x00, - ASN_APP_COUNTER = 0x01, - ASN_APP_GAUGE = 0x02, - ASN_APP_TIMETICKS = 0x03, - ASN_APP_OPAQUE = 0x04, /* not implemented */ - ASN_APP_COUNTER64 = 0x06, - - ASN_EXCEPT_NOSUCHOBJECT = 0x00, - ASN_EXCEPT_NOSUCHINSTANCE = 0x01, - ASN_EXCEPT_ENDOFMIBVIEW = 0x02, -}; - -#endif diff --git a/src/bsnmp/snmp.c b/src/bsnmp/snmp.c deleted file mode 100644 index f758262..0000000 --- a/src/bsnmp/snmp.c +++ /dev/null @@ -1,1081 +0,0 @@ -/* - * Copyright (c) 2001-2003 - * Fraunhofer Institute for Open Communication Systems (FhG Fokus). - * All rights reserved. - * - * Author: Harti Brandt <harti@freebsd.org> - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * 1. Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * 2. Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * - * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND - * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE - * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE - * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL - * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS - * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) - * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT - * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY - * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - * - * $Begemot: bsnmp/lib/snmp.c,v 1.38 2004/08/06 08:46:53 brandt Exp $ - * - * SNMP - */ -#include <sys/types.h> -#include <sys/socket.h> -#include <stdio.h> -#include <stdlib.h> -#include <stddef.h> -#include <stdarg.h> -#include <stdint.h> -#include <string.h> -#include <ctype.h> -#include <netdb.h> -#include <errno.h> - -#include "asn1.h" -#include "snmp.h" -#include "snmppriv.h" - -static void snmp_error_func(const char *, ...); -static void snmp_printf_func(const char *, ...); - -void (*snmp_error)(const char *, ...) = snmp_error_func; -void (*snmp_printf)(const char *, ...) = snmp_printf_func; - -/* - * Get the next variable binding from the list. - * ASN errors on the sequence or the OID are always fatal. - */ -static enum asn_err -get_var_binding(struct asn_buf *b, struct snmp_value *binding) -{ - u_char type; - asn_len_t len, trailer; - enum asn_err err; - - if (asn_get_sequence(b, &len) != ASN_ERR_OK) { - snmp_error("cannot parse varbind header"); - return (ASN_ERR_FAILED); - } - - /* temporary truncate the length so that the parser does not - * eat up bytes behind the sequence in the case the encoding is - * wrong of inner elements. */ - trailer = b->asn_len - len; - b->asn_len = len; - - if (asn_get_objid(b, &binding->var) != ASN_ERR_OK) { - snmp_error("cannot parse binding objid"); - return (ASN_ERR_FAILED); - } - if (asn_get_header(b, &type, &len) != ASN_ERR_OK) { - snmp_error("cannot parse binding value header"); - return (ASN_ERR_FAILED); - } - - switch (type) { - - case ASN_TYPE_NULL: - binding->syntax = SNMP_SYNTAX_NULL; - err = asn_get_null_raw(b, len); - break; - - case ASN_TYPE_INTEGER: - binding->syntax = SNMP_SYNTAX_INTEGER; - err = asn_get_integer_raw(b, len, &binding->v.integer); - break; - - case ASN_TYPE_OCTETSTRING: - binding->syntax = SNMP_SYNTAX_OCTETSTRING; - binding->v.octetstring.octets = malloc(len); - if (binding->v.octetstring.octets == NULL) { - snmp_error("%s", strerror(errno)); - return (ASN_ERR_FAILED); - } - binding->v.octetstring.len = len; - err = asn_get_octetstring_raw(b, len, - binding->v.octetstring.octets, - &binding->v.octetstring.len); - if (ASN_ERR_STOPPED(err)) { - free(binding->v.octetstring.octets); - binding->v.octetstring.octets = NULL; - } - break; - - case ASN_TYPE_OBJID: - binding->syntax = SNMP_SYNTAX_OID; - err = asn_get_objid_raw(b, len, &binding->v.oid); - break; - - case ASN_CLASS_APPLICATION|ASN_APP_IPADDRESS: - binding->syntax = SNMP_SYNTAX_IPADDRESS; - err = asn_get_ipaddress_raw(b, len, binding->v.ipaddress); - break; - - case ASN_CLASS_APPLICATION|ASN_APP_TIMETICKS: - binding->syntax = SNMP_SYNTAX_TIMETICKS; - err = asn_get_uint32_raw(b, len, &binding->v.uint32); - break; - - case ASN_CLASS_APPLICATION|ASN_APP_COUNTER: - binding->syntax = SNMP_SYNTAX_COUNTER; - err = asn_get_uint32_raw(b, len, &binding->v.uint32); - break; - - case ASN_CLASS_APPLICATION|ASN_APP_GAUGE: - binding->syntax = SNMP_SYNTAX_GAUGE; - err = asn_get_uint32_raw(b, len, &binding->v.uint32); - break; - - case ASN_CLASS_APPLICATION|ASN_APP_COUNTER64: - binding->syntax = SNMP_SYNTAX_COUNTER64; - err = asn_get_counter64_raw(b, len, &binding->v.counter64); - break; - - case ASN_CLASS_CONTEXT | ASN_EXCEPT_NOSUCHOBJECT: - binding->syntax = SNMP_SYNTAX_NOSUCHOBJECT; - err = asn_get_null_raw(b, len); - break; - - case ASN_CLASS_CONTEXT | ASN_EXCEPT_NOSUCHINSTANCE: - binding->syntax = SNMP_SYNTAX_NOSUCHINSTANCE; - err = asn_get_null_raw(b, len); - break; - - case ASN_CLASS_CONTEXT | ASN_EXCEPT_ENDOFMIBVIEW: - binding->syntax = SNMP_SYNTAX_ENDOFMIBVIEW; - err = asn_get_null_raw(b, len); - break; - - default: - if ((err = asn_skip(b, len)) == ASN_ERR_OK) - err = ASN_ERR_TAG; - snmp_error("bad binding value type 0x%x", type); - break; - } - - if (ASN_ERR_STOPPED(err)) { - snmp_error("cannot parse binding value"); - return (err); - } - - if (b->asn_len != 0) - snmp_error("ignoring junk at end of binding"); - - b->asn_len = trailer; - - return (err); -} - -/* - * Parse the different PDUs contents. Any ASN error in the outer components - * are fatal. Only errors in variable values may be tolerated. If all - * components can be parsed it returns either ASN_ERR_OK or the first - * error that was found. - */ -enum asn_err -snmp_parse_pdus_hdr(struct asn_buf *b, struct snmp_pdu *pdu, asn_len_t *lenp) -{ - if (pdu->type == SNMP_PDU_TRAP) { - if (asn_get_objid(b, &pdu->enterprise) != ASN_ERR_OK) { - snmp_error("cannot parse trap enterprise"); - return (ASN_ERR_FAILED); - } - if (asn_get_ipaddress(b, pdu->agent_addr) != ASN_ERR_OK) { - snmp_error("cannot parse trap agent address"); - return (ASN_ERR_FAILED); - } - if (asn_get_integer(b, &pdu->generic_trap) != ASN_ERR_OK) { - snmp_error("cannot parse 'generic-trap'"); - return (ASN_ERR_FAILED); - } - if (asn_get_integer(b, &pdu->specific_trap) != ASN_ERR_OK) { - snmp_error("cannot parse 'specific-trap'"); - return (ASN_ERR_FAILED); - } - if (asn_get_timeticks(b, &pdu->time_stamp) != ASN_ERR_OK) { - snmp_error("cannot parse trap 'time-stamp'"); - return (ASN_ERR_FAILED); - } - } else { - if (asn_get_integer(b, &pdu->request_id) != ASN_ERR_OK) { - snmp_error("cannot parse 'request-id'"); - return (ASN_ERR_FAILED); - } - if (asn_get_integer(b, &pdu->error_status) != ASN_ERR_OK) { - snmp_error("cannot parse 'error_status'"); - return (ASN_ERR_FAILED); - } - if (asn_get_integer(b, &pdu->error_index) != ASN_ERR_OK) { - snmp_error("cannot parse 'error_index'"); - return (ASN_ERR_FAILED); - } - } - - if (asn_get_sequence(b, lenp) != ASN_ERR_OK) { - snmp_error("cannot get varlist header"); - return (ASN_ERR_FAILED); - } - - return (ASN_ERR_OK); -} - -static enum asn_err -parse_pdus(struct asn_buf *b, struct snmp_pdu *pdu, int32_t *ip) -{ - asn_len_t len, trailer; - struct snmp_value *v; - enum asn_err err, err1; - - err = snmp_parse_pdus_hdr(b, pdu, &len); - if (ASN_ERR_STOPPED(err)) - return (err); - - trailer = b->asn_len - len; - - v = pdu->bindings; - err = ASN_ERR_OK; - while (b->asn_len != 0) { - if (pdu->nbindings == SNMP_MAX_BINDINGS) { - snmp_error("too many bindings (> %u) in PDU", - SNMP_MAX_BINDINGS); - return (ASN_ERR_FAILED); - } - err1 = get_var_binding(b, v); - if (ASN_ERR_STOPPED(err1)) - return (ASN_ERR_FAILED); - if (err1 != ASN_ERR_OK && err == ASN_ERR_OK) { - err = err1; - *ip = pdu->nbindings + 1; - } - pdu->nbindings++; - v++; - } - - b->asn_len = trailer; - - return (err); -} - -/* - * Parse the outer SEQUENCE value. ASN_ERR_TAG means 'bad version'. - */ -enum asn_err -snmp_parse_message_hdr(struct asn_buf *b, struct snmp_pdu *pdu, asn_len_t *lenp) -{ - int32_t version; - u_char type; - u_int comm_len; - - if (asn_get_integer(b, &version) != ASN_ERR_OK) { - snmp_error("cannot decode version"); - return (ASN_ERR_FAILED); - } - - if (version == 0) { - pdu->version = SNMP_V1; - } else if (version == 1) { - pdu->version = SNMP_V2c; - } else { - pdu->version = SNMP_Verr; - snmp_error("unsupported SNMP version"); - return (ASN_ERR_TAG); - } - - comm_len = SNMP_COMMUNITY_MAXLEN; - if (asn_get_octetstring(b, (u_char *)pdu->community, - &comm_len) != ASN_ERR_OK) { - snmp_error("cannot decode community"); - return (ASN_ERR_FAILED); - } - pdu->community[comm_len] = '\0'; - - if (asn_get_header(b, &type, lenp) != ASN_ERR_OK) { - snmp_error("cannot get pdu header"); - return (ASN_ERR_FAILED); - } - if ((type & ~ASN_TYPE_MASK) != - (ASN_TYPE_CONSTRUCTED | ASN_CLASS_CONTEXT)) { - snmp_error("bad pdu header tag"); - return (ASN_ERR_FAILED); - } - pdu->type = type & ASN_TYPE_MASK; - - switch (pdu->type) { - - case SNMP_PDU_GET: - case SNMP_PDU_GETNEXT: - case SNMP_PDU_RESPONSE: - case SNMP_PDU_SET: - break; - - case SNMP_PDU_TRAP: - if (pdu->version != SNMP_V1) { - snmp_error("bad pdu type %u", pdu->type); - return (ASN_ERR_FAILED); - } - break; - - case SNMP_PDU_GETBULK: - case SNMP_PDU_INFORM: - case SNMP_PDU_TRAP2: - case SNMP_PDU_REPORT: - if (pdu->version == SNMP_V1) { - snmp_error("bad pdu type %u", pdu->type); - return (ASN_ERR_FAILED); - } - break; - - default: - snmp_error("bad pdu type %u", pdu->type); - return (ASN_ERR_FAILED); - } - - - if (*lenp > b->asn_len) { - snmp_error("pdu length too long"); - return (ASN_ERR_FAILED); - } - - return (ASN_ERR_OK); -} - -static enum asn_err -parse_message(struct asn_buf *b, struct snmp_pdu *pdu, int32_t *ip) -{ - enum asn_err err; - asn_len_t len, trailer; - - err = snmp_parse_message_hdr(b, pdu, &len); - if (ASN_ERR_STOPPED(err)) - return (err); - - trailer = b->asn_len - len; - b->asn_len = len; - - err = parse_pdus(b, pdu, ip); - if (ASN_ERR_STOPPED(err)) - return (ASN_ERR_FAILED); - - if (b->asn_len != 0) - snmp_error("ignoring trailing junk after pdu"); - - b->asn_len = trailer; - - return (err); -} - -/* - * Decode the PDU except for the variable bindings itself. - * If decoding fails because of a bad binding, but the rest can be - * decoded, ip points to the index of the failed variable (errors - * OORANGE, BADLEN or BADVERS). - */ -enum snmp_code -snmp_pdu_decode(struct asn_buf *b, struct snmp_pdu *pdu, int32_t *ip) -{ - asn_len_t len; - - memset(pdu, 0, sizeof(*pdu)); - - if (asn_get_sequence(b, &len) != ASN_ERR_OK) { - snmp_error("cannot decode pdu header"); - return (SNMP_CODE_FAILED); - } - if (b->asn_len < len) { - snmp_error("outer sequence value too short"); - return (SNMP_CODE_FAILED); - } - if (b->asn_len != len) { - snmp_error("ignoring trailing junk in message"); - b->asn_len = len; - } - - switch (parse_message(b, pdu, ip)) { - - case ASN_ERR_OK: - return (SNMP_CODE_OK); - - case ASN_ERR_FAILED: - case ASN_ERR_EOBUF: - snmp_pdu_free(pdu); - return (SNMP_CODE_FAILED); - - case ASN_ERR_BADLEN: - return (SNMP_CODE_BADLEN); - - case ASN_ERR_RANGE: - return (SNMP_CODE_OORANGE); - - case ASN_ERR_TAG: - if (pdu->version == SNMP_Verr) - return (SNMP_CODE_BADVERS); - else - return (SNMP_CODE_BADENC); - } - - return (SNMP_CODE_OK); -} - -/* - * Check whether what we have is the complete PDU by snooping at the - * enclosing structure header. This returns: - * -1 if there are ASN.1 errors - * 0 if we need more data - * > 0 the length of this PDU - */ -int -snmp_pdu_snoop(const struct asn_buf *b0) -{ - u_int length; - asn_len_t len; - struct asn_buf b = *b0; - - /* <0x10|0x20> <len> <data...> */ - - if (b.asn_len == 0) - return (0); - if (b.asn_cptr[0] != (ASN_TYPE_SEQUENCE | ASN_TYPE_CONSTRUCTED)) { - asn_error(&b, "bad sequence type %u", b.asn_cptr[0]); - return (-1); - } - b.asn_len--; - b.asn_cptr++; - - if (b.asn_len == 0) - return (0); - - if (*b.asn_cptr & 0x80) { - /* long length */ - length = *b.asn_cptr++ & 0x7f; - b.asn_len--; - if (length == 0) { - asn_error(&b, "indefinite length not supported"); - return (-1); - } - if (length > ASN_MAXLENLEN) { - asn_error(&b, "long length too long (%u)", length); - return (-1); - } - if (length > b.asn_len) - return (0); - len = 0; - while (length--) { - len = (len << 8) | *b.asn_cptr++; - b.asn_len--; - } - } else { - len = *b.asn_cptr++; - b.asn_len--; - } - - if (len > b.asn_len) - return (0); - - return (len + b.asn_cptr - b0->asn_cptr); -} - -/* - * Encode the SNMP PDU without the variable bindings field. - * We do this the rather uneffective way by - * moving things around and assuming that the length field will never - * use more than 2 bytes. - * We need a number of pointers to apply the fixes afterwards. - */ -enum snmp_code -snmp_pdu_encode_header(struct asn_buf *b, struct snmp_pdu *pdu) -{ - enum asn_err err; - - if (asn_put_temp_header(b, (ASN_TYPE_SEQUENCE|ASN_TYPE_CONSTRUCTED), - &pdu->outer_ptr) != ASN_ERR_OK) - return (SNMP_CODE_FAILED); - - if (pdu->version == SNMP_V1) - err = asn_put_integer(b, 0); - else if (pdu->version == SNMP_V2c) - err = asn_put_integer(b, 1); - else - return (SNMP_CODE_BADVERS); - if (err != ASN_ERR_OK) - return (SNMP_CODE_FAILED); - - if (asn_put_octetstring(b, (u_char *)pdu->community, - strlen(pdu->community)) != ASN_ERR_OK) - return (SNMP_CODE_FAILED); - - if (asn_put_temp_header(b, (ASN_TYPE_CONSTRUCTED | ASN_CLASS_CONTEXT | - pdu->type), &pdu->pdu_ptr) != ASN_ERR_OK) - return (SNMP_CODE_FAILED); - - if (pdu->type == SNMP_PDU_TRAP) { - if (pdu->version != SNMP_V1 || - asn_put_objid(b, &pdu->enterprise) != ASN_ERR_OK || - asn_put_ipaddress(b, pdu->agent_addr) != ASN_ERR_OK || - asn_put_integer(b, pdu->generic_trap) != ASN_ERR_OK || - asn_put_integer(b, pdu->specific_trap) != ASN_ERR_OK || - asn_put_timeticks(b, pdu->time_stamp) != ASN_ERR_OK) - return (SNMP_CODE_FAILED); - } else { - if (pdu->version == SNMP_V1 && (pdu->type == SNMP_PDU_GETBULK || - pdu->type == SNMP_PDU_INFORM || - pdu->type == SNMP_PDU_TRAP2 || - pdu->type == SNMP_PDU_REPORT)) - return (SNMP_CODE_FAILED); - - if (asn_put_integer(b, pdu->request_id) != ASN_ERR_OK || - asn_put_integer(b, pdu->error_status) != ASN_ERR_OK || - asn_put_integer(b, pdu->error_index) != ASN_ERR_OK) - return (SNMP_CODE_FAILED); - } - - if (asn_put_temp_header(b, (ASN_TYPE_SEQUENCE|ASN_TYPE_CONSTRUCTED), - &pdu->vars_ptr) != ASN_ERR_OK) - return (SNMP_CODE_FAILED); - - return (SNMP_CODE_OK); -} - -enum snmp_code -snmp_fix_encoding(struct asn_buf *b, const struct snmp_pdu *pdu) -{ - if (asn_commit_header(b, pdu->vars_ptr) != ASN_ERR_OK || - asn_commit_header(b, pdu->pdu_ptr) != ASN_ERR_OK || - asn_commit_header(b, pdu->outer_ptr) != ASN_ERR_OK) - return (SNMP_CODE_FAILED); - return (SNMP_CODE_OK); -} - -/* - * Encode a binding. Caller must ensure, that the syntax is ok for that version. - * Be sure not to cobber b, when something fails. - */ -enum asn_err -snmp_binding_encode(struct asn_buf *b, const struct snmp_value *binding) -{ - u_char *ptr; - enum asn_err err; - struct asn_buf save = *b; - - if ((err = asn_put_temp_header(b, (ASN_TYPE_SEQUENCE | - ASN_TYPE_CONSTRUCTED), &ptr)) != ASN_ERR_OK) { - *b = save; - return (err); - } - - if ((err = asn_put_objid(b, &binding->var)) != ASN_ERR_OK) { - *b = save; - return (err); - } - - switch (binding->syntax) { - - case SNMP_SYNTAX_NULL: - err = asn_put_null(b); - break; - - case SNMP_SYNTAX_INTEGER: - err = asn_put_integer(b, binding->v.integer); - break; - - case SNMP_SYNTAX_OCTETSTRING: - err = asn_put_octetstring(b, binding->v.octetstring.octets, - binding->v.octetstring.len); - break; - - case SNMP_SYNTAX_OID: - err = asn_put_objid(b, &binding->v.oid); - break; - - case SNMP_SYNTAX_IPADDRESS: - err = asn_put_ipaddress(b, binding->v.ipaddress); - break; - - case SNMP_SYNTAX_TIMETICKS: - err = asn_put_uint32(b, ASN_APP_TIMETICKS, binding->v.uint32); - break; - - case SNMP_SYNTAX_COUNTER: - err = asn_put_uint32(b, ASN_APP_COUNTER, binding->v.uint32); - break; - - case SNMP_SYNTAX_GAUGE: - err = asn_put_uint32(b, ASN_APP_GAUGE, binding->v.uint32); - break; - - case SNMP_SYNTAX_COUNTER64: - err = asn_put_counter64(b, binding->v.counter64); - break; - - case SNMP_SYNTAX_NOSUCHOBJECT: - err = asn_put_exception(b, ASN_EXCEPT_NOSUCHOBJECT); - break; - - case SNMP_SYNTAX_NOSUCHINSTANCE: - err = asn_put_exception(b, ASN_EXCEPT_NOSUCHINSTANCE); - break; - - case SNMP_SYNTAX_ENDOFMIBVIEW: - err = asn_put_exception(b, ASN_EXCEPT_ENDOFMIBVIEW); - break; - } - - if (err != ASN_ERR_OK) { - *b = save; - return (err); - } - - err = asn_commit_header(b, ptr); - if (err != ASN_ERR_OK) { - *b = save; - return (err); - } - - return (ASN_ERR_OK); -} - -/* - * Encode an PDU. - */ -enum snmp_code -snmp_pdu_encode(struct snmp_pdu *pdu, struct asn_buf *resp_b) -{ - u_int idx; - enum snmp_code err; - - if ((err = snmp_pdu_encode_header(resp_b, pdu)) != SNMP_CODE_OK) - return (err); - for (idx = 0; idx < pdu->nbindings; idx++) - if ((err = snmp_binding_encode(resp_b, &pdu->bindings[idx])) - != ASN_ERR_OK) - return (SNMP_CODE_FAILED); - - return (snmp_fix_encoding(resp_b, pdu)); -} - -static void -dump_binding(const struct snmp_value *b) -{ - u_int i; - char buf[ASN_OIDSTRLEN]; - - snmp_printf("%s=", asn_oid2str_r(&b->var, buf)); - switch (b->syntax) { - - case SNMP_SYNTAX_NULL: - snmp_printf("NULL"); - break; - - case SNMP_SYNTAX_INTEGER: - snmp_printf("INTEGER %d", b->v.integer); - break; - - case SNMP_SYNTAX_OCTETSTRING: - snmp_printf("OCTET STRING %lu:", b->v.octetstring.len); - for (i = 0; i < b->v.octetstring.len; i++) - snmp_printf(" %02x", b->v.octetstring.octets[i]); - break; - - case SNMP_SYNTAX_OID: - snmp_printf("OID %s", asn_oid2str_r(&b->v.oid, buf)); - break; - - case SNMP_SYNTAX_IPADDRESS: - snmp_printf("IPADDRESS %u.%u.%u.%u", b->v.ipaddress[0], - b->v.ipaddress[1], b->v.ipaddress[2], b->v.ipaddress[3]); - break; - - case SNMP_SYNTAX_COUNTER: - snmp_printf("COUNTER %u", b->v.uint32); - break; - - case SNMP_SYNTAX_GAUGE: - snmp_printf("GAUGE %u", b->v.uint32); - break; - - case SNMP_SYNTAX_TIMETICKS: - snmp_printf("TIMETICKS %u", b->v.uint32); - break; - - case SNMP_SYNTAX_COUNTER64: - snmp_printf("COUNTER64 %lld", b->v.counter64); - break; - - case SNMP_SYNTAX_NOSUCHOBJECT: - snmp_printf("NoSuchObject"); - break; - - case SNMP_SYNTAX_NOSUCHINSTANCE: - snmp_printf("NoSuchInstance"); - break; - - case SNMP_SYNTAX_ENDOFMIBVIEW: - snmp_printf("EndOfMibView"); - break; - - default: - snmp_printf("UNKNOWN SYNTAX %u", b->syntax); - break; - } -} - -static __inline void -dump_bindings(const struct snmp_pdu *pdu) -{ - u_int i; - - for (i = 0; i < pdu->nbindings; i++) { - snmp_printf(" [%u]: ", i); - dump_binding(&pdu->bindings[i]); - snmp_printf("\n"); - } -} - -static __inline void -dump_notrap(const struct snmp_pdu *pdu) -{ - snmp_printf(" request_id=%d", pdu->request_id); - snmp_printf(" error_status=%d", pdu->error_status); - snmp_printf(" error_index=%d\n", pdu->error_index); - dump_bindings(pdu); -} - -void -snmp_pdu_dump(const struct snmp_pdu *pdu) -{ - char buf[ASN_OIDSTRLEN]; - const char *vers; - static const char *types[] = { - [SNMP_PDU_GET] = "GET", - [SNMP_PDU_GETNEXT] = "GETNEXT", - [SNMP_PDU_RESPONSE] = "RESPONSE", - [SNMP_PDU_SET] = "SET", - [SNMP_PDU_TRAP] = "TRAPv1", - [SNMP_PDU_GETBULK] = "GETBULK", - [SNMP_PDU_INFORM] = "INFORM", - [SNMP_PDU_TRAP2] = "TRAPv2", - [SNMP_PDU_REPORT] = "REPORT", - }; - - if (pdu->version == SNMP_V1) - vers = "SNMPv1"; - else if (pdu->version == SNMP_V2c) - vers = "SNMPv2c"; - else - vers = "v?"; - - switch (pdu->type) { - case SNMP_PDU_TRAP: - snmp_printf("%s %s '%s'", types[pdu->type], vers, pdu->community); - snmp_printf(" enterprise=%s", asn_oid2str_r(&pdu->enterprise, buf)); - snmp_printf(" agent_addr=%u.%u.%u.%u", pdu->agent_addr[0], - pdu->agent_addr[1], pdu->agent_addr[2], pdu->agent_addr[3]); - snmp_printf(" generic_trap=%d", pdu->generic_trap); - snmp_printf(" specific_trap=%d", pdu->specific_trap); - snmp_printf(" time-stamp=%u\n", pdu->time_stamp); - dump_bindings(pdu); - break; - - case SNMP_PDU_GET: - case SNMP_PDU_GETNEXT: - case SNMP_PDU_RESPONSE: - case SNMP_PDU_SET: - case SNMP_PDU_GETBULK: - case SNMP_PDU_INFORM: - case SNMP_PDU_TRAP2: - case SNMP_PDU_REPORT: - snmp_printf("%s %s '%s'", types[pdu->type], vers, pdu->community); - dump_notrap(pdu); - break; - - default: - snmp_printf("bad pdu type %u\n", pdu->type); - break; - } -} - -void -snmp_value_free(struct snmp_value *value) -{ - if (value->syntax == SNMP_SYNTAX_OCTETSTRING) - free(value->v.octetstring.octets); - value->syntax = SNMP_SYNTAX_NULL; -} - -int -snmp_value_copy(struct snmp_value *to, const struct snmp_value *from) -{ - to->var = from->var; - to->syntax = from->syntax; - - if (from->syntax == SNMP_SYNTAX_OCTETSTRING) { - if ((to->v.octetstring.len = from->v.octetstring.len) == 0) - to->v.octetstring.octets = NULL; - else { - to->v.octetstring.octets = malloc(to->v.octetstring.len); - if (to->v.octetstring.octets == NULL) - return (-1); - (void)memcpy(to->v.octetstring.octets, - from->v.octetstring.octets, to->v.octetstring.len); - } - } else - to->v = from->v; - return (0); -} - -void -snmp_pdu_free(struct snmp_pdu *pdu) -{ - u_int i; - - for (i = 0; i < pdu->nbindings; i++) - snmp_value_free(&pdu->bindings[i]); -} - -/* - * Parse an ASCII SNMP value into the binary form - */ -int -snmp_value_parse(const char *str, enum snmp_syntax syntax, union snmp_values *v) -{ - char *end; - - switch (syntax) { - - case SNMP_SYNTAX_NULL: - case SNMP_SYNTAX_NOSUCHOBJECT: - case SNMP_SYNTAX_NOSUCHINSTANCE: - case SNMP_SYNTAX_ENDOFMIBVIEW: - if (*str != '\0') - return (-1); - return (0); - - case SNMP_SYNTAX_INTEGER: - v->integer = strtoll(str, &end, 0); - if (*end != '\0') - return (-1); - return (0); - - case SNMP_SYNTAX_OCTETSTRING: - { - u_long len; /* actual length of string */ - u_long alloc; /* allocate length of string */ - u_char *octs; /* actual octets */ - u_long oct; /* actual octet */ - u_char *nocts; /* to avoid memory leak */ - u_char c; /* actual character */ - -# define STUFFC(C) \ - if (alloc == len) { \ - alloc += 100; \ - if ((nocts = realloc(octs, alloc)) == NULL) { \ - free(octs); \ - return (-1); \ - } \ - octs = nocts; \ - } \ - octs[len++] = (C); - - len = alloc = 0; - octs = NULL; - - if (*str == '"') { - str++; - while((c = *str++) != '\0') { - if (c == '"') { - if (*str != '\0') { - free(octs); - return (-1); - } - break; - } - if (c == '\\') { - switch (c = *str++) { - - case '\\': - break; - case 'a': - c = '\a'; - break; - case 'b': - c = '\b'; - break; - case 'f': - c = '\f'; - break; - case 'n': - c = '\n'; - break; - case 'r': - c = '\r'; - break; - case 't': - c = '\t'; - break; - case 'v': - c = '\v'; - break; - case 'x': - c = 0; - if (!isxdigit(*str)) - break; - if (isdigit(*str)) - c = *str++ - '0'; - else if (isupper(*str)) - c = *str++ - 'A' + 10; - else - c = *str++ - 'a' + 10; - if (!isxdigit(*str)) - break; - if (isdigit(*str)) - c += *str++ - '0'; - else if (isupper(*str)) - c += *str++ - 'A' + 10; - else - c += *str++ - 'a' + 10; - break; - case '0': case '1': case '2': - case '3': case '4': case '5': - case '6': case '7': - c = *str++ - '0'; - if (*str < '0' || *str > '7') - break; - c = *str++ - '0'; - if (*str < '0' || *str > '7') - break; - c = *str++ - '0'; - break; - default: - break; - } - } - STUFFC(c); - } - } else { - while (*str != '\0') { - oct = strtoul(str, &end, 16); - str = end; - if (oct > 0xff) { - free(octs); - return (-1); - } - STUFFC(oct); - if (*str == ':') - str++; - else if(*str != '\0') { - free(octs); - return (-1); - } - } - } - v->octetstring.octets = octs; - v->octetstring.len = len; - return (0); -# undef STUFFC - } - - case SNMP_SYNTAX_OID: - { - u_long subid; - - v->oid.len = 0; - - for (;;) { - if (v->oid.len == ASN_MAXOIDLEN) - return (-1); - subid = strtoul(str, &end, 10); - str = end; - if (subid > ASN_MAXID) - return (-1); - v->oid.subs[v->oid.len++] = (asn_subid_t)subid; - if (*str == '\0') - break; - if (*str != '.') - return (-1); - str++; - } - return (0); - } - - case SNMP_SYNTAX_IPADDRESS: - { - struct hostent *he; - u_long ip[4]; - int n; - - if (sscanf(str, "%lu.%lu.%lu.%lu%n", &ip[0], &ip[1], &ip[2], - &ip[3], &n) == 4 && (size_t)n == strlen(str) && - ip[0] <= 0xff && ip[1] <= 0xff && - ip[2] <= 0xff && ip[3] <= 0xff) { - v->ipaddress[0] = (u_char)ip[0]; - v->ipaddress[1] = (u_char)ip[1]; - v->ipaddress[2] = (u_char)ip[2]; - v->ipaddress[3] = (u_char)ip[3]; - return (0); - } - - if ((he = gethostbyname(str)) == NULL) - return (-1); - if (he->h_addrtype != AF_INET) - return (-1); - - v->ipaddress[0] = he->h_addr[0]; - v->ipaddress[1] = he->h_addr[1]; - v->ipaddress[2] = he->h_addr[2]; - v->ipaddress[3] = he->h_addr[3]; - return (0); - } - - case SNMP_SYNTAX_COUNTER: - case SNMP_SYNTAX_GAUGE: - case SNMP_SYNTAX_TIMETICKS: - { - uint64_t sub; - - sub = strtoull(str, &end, 0); - if (*end != '\0' || sub > 0xffffffff) - return (-1); - v->uint32 = (uint32_t)sub; - return (0); - } - - case SNMP_SYNTAX_COUNTER64: - v->counter64 = strtoull(str, &end, 0); - if (*end != '\0') - return (-1); - return (0); - } - abort(); -} - -static void -snmp_error_func(const char *fmt, ...) -{ - va_list ap; - - va_start(ap, fmt); - fprintf(stderr, "SNMP: "); - vfprintf(stderr, fmt, ap); - fprintf(stderr, "\n"); - va_end(ap); -} - -static void -snmp_printf_func(const char *fmt, ...) -{ - va_list ap; - - va_start(ap, fmt); - vfprintf(stderr, fmt, ap); - va_end(ap); -} diff --git a/src/bsnmp/snmp.h b/src/bsnmp/snmp.h deleted file mode 100644 index 1ae0775..0000000 --- a/src/bsnmp/snmp.h +++ /dev/null @@ -1,174 +0,0 @@ -/* - * Copyright (c) 2001-2003 - * Fraunhofer Institute for Open Communication Systems (FhG Fokus). - * All rights reserved. - * - * Author: Harti Brandt <harti@freebsd.org> - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * 1. Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * 2. Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * - * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND - * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE - * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE - * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL - * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS - * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) - * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT - * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY - * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - * - * $Begemot: bsnmp/lib/snmp.h,v 1.30 2004/08/06 08:46:54 brandt Exp $ - * - * Header file for SNMP functions. - */ -#ifndef snmp_h_ -#define snmp_h_ - -#include <sys/types.h> - -#define SNMP_COMMUNITY_MAXLEN 128 -#define SNMP_MAX_BINDINGS 100 - -enum snmp_syntax { - SNMP_SYNTAX_NULL = 0, - SNMP_SYNTAX_INTEGER, /* == INTEGER32 */ - SNMP_SYNTAX_OCTETSTRING, - SNMP_SYNTAX_OID, - SNMP_SYNTAX_IPADDRESS, - SNMP_SYNTAX_COUNTER, - SNMP_SYNTAX_GAUGE, /* == UNSIGNED32 */ - SNMP_SYNTAX_TIMETICKS, - - /* v2 additions */ - SNMP_SYNTAX_COUNTER64, - SNMP_SYNTAX_NOSUCHOBJECT, /* exception */ - SNMP_SYNTAX_NOSUCHINSTANCE, /* exception */ - SNMP_SYNTAX_ENDOFMIBVIEW, /* exception */ -}; - -struct snmp_value { - struct asn_oid var; - enum snmp_syntax syntax; - union snmp_values { - int32_t integer; /* also integer32 */ - struct { - u_int len; - u_char *octets; - } octetstring; - struct asn_oid oid; - u_char ipaddress[4]; - uint32_t uint32; /* also gauge32, counter32, - unsigned32, timeticks */ - uint64_t counter64; - } v; -}; - -enum snmp_version { - SNMP_Verr = 0, - SNMP_V1 = 1, - SNMP_V2c, -}; - -struct snmp_pdu { - char community[SNMP_COMMUNITY_MAXLEN + 1]; - enum snmp_version version; - u_int type; - - /* trap only */ - struct asn_oid enterprise; - u_char agent_addr[4]; - int32_t generic_trap; - int32_t specific_trap; - uint32_t time_stamp; - - /* others */ - int32_t request_id; - int32_t error_status; - int32_t error_index; - - /* fixes for encoding */ - u_char *outer_ptr; - u_char *pdu_ptr; - u_char *vars_ptr; - - struct snmp_value bindings[SNMP_MAX_BINDINGS]; - u_int nbindings; -}; -#define snmp_v1_pdu snmp_pdu - -#define SNMP_PDU_GET 0 -#define SNMP_PDU_GETNEXT 1 -#define SNMP_PDU_RESPONSE 2 -#define SNMP_PDU_SET 3 -#define SNMP_PDU_TRAP 4 /* v1 */ -#define SNMP_PDU_GETBULK 5 /* v2 */ -#define SNMP_PDU_INFORM 6 /* v2 */ -#define SNMP_PDU_TRAP2 7 /* v2 */ -#define SNMP_PDU_REPORT 8 /* v2 */ - -#define SNMP_ERR_NOERROR 0 -#define SNMP_ERR_TOOBIG 1 -#define SNMP_ERR_NOSUCHNAME 2 /* v1 */ -#define SNMP_ERR_BADVALUE 3 /* v1 */ -#define SNMP_ERR_READONLY 4 /* v1 */ -#define SNMP_ERR_GENERR 5 -#define SNMP_ERR_NO_ACCESS 6 /* v2 */ -#define SNMP_ERR_WRONG_TYPE 7 /* v2 */ -#define SNMP_ERR_WRONG_LENGTH 8 /* v2 */ -#define SNMP_ERR_WRONG_ENCODING 9 /* v2 */ -#define SNMP_ERR_WRONG_VALUE 10 /* v2 */ -#define SNMP_ERR_NO_CREATION 11 /* v2 */ -#define SNMP_ERR_INCONS_VALUE 12 /* v2 */ -#define SNMP_ERR_RES_UNAVAIL 13 /* v2 */ -#define SNMP_ERR_COMMIT_FAILED 14 /* v2 */ -#define SNMP_ERR_UNDO_FAILED 15 /* v2 */ -#define SNMP_ERR_AUTH_ERR 16 /* v2 */ -#define SNMP_ERR_NOT_WRITEABLE 17 /* v2 */ -#define SNMP_ERR_INCONS_NAME 18 /* v2 */ - -#define SNMP_TRAP_COLDSTART 0 -#define SNMP_TRAP_WARMSTART 1 -#define SNMP_TRAP_LINKDOWN 2 -#define SNMP_TRAP_LINKUP 3 -#define SNMP_TRAP_AUTHENTICATION_FAILURE 4 -#define SNMP_TRAP_EGP_NEIGHBOR_LOSS 5 -#define SNMP_TRAP_ENTERPRISE 6 - -enum snmp_code { - SNMP_CODE_OK = 0, - SNMP_CODE_FAILED, - SNMP_CODE_BADVERS, - SNMP_CODE_BADLEN, - SNMP_CODE_BADENC, - SNMP_CODE_OORANGE, -}; - -void snmp_value_free(struct snmp_value *); -int snmp_value_parse(const char *, enum snmp_syntax, union snmp_values *); -int snmp_value_copy(struct snmp_value *, const struct snmp_value *); - -void snmp_pdu_free(struct snmp_pdu *); -enum snmp_code snmp_pdu_decode(struct asn_buf *b, struct snmp_pdu *pdu, int32_t *); -enum snmp_code snmp_pdu_encode(struct snmp_pdu *pdu, struct asn_buf *resp_b); - -int snmp_pdu_snoop(const struct asn_buf *); - -void snmp_pdu_dump(const struct snmp_pdu *pdu); - -extern void (*snmp_error)(const char *, ...); -extern void (*snmp_printf)(const char *, ...); - -#define TRUTH_MK(F) ((F) ? 1 : 2) -#define TRUTH_GET(T) (((T) == 1) ? 1 : 0) -#define TRUTH_OK(T) ((T) == 1 || (T) == 2) - -#endif diff --git a/src/bsnmp/snmppriv.h b/src/bsnmp/snmppriv.h deleted file mode 100644 index 87ef60e..0000000 --- a/src/bsnmp/snmppriv.h +++ /dev/null @@ -1,45 +0,0 @@ -/* - * Copyright (c) 2001-2003 - * Fraunhofer Institute for Open Communication Systems (FhG Fokus). - * All rights reserved. - * - * Author: Harti Brandt <harti@freebsd.org> - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * 1. Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * 2. Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * - * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND - * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE - * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE - * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL - * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS - * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) - * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT - * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY - * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - * - * $Begemot: bsnmp/lib/snmppriv.h,v 1.9 2004/08/06 08:46:58 brandt Exp $ - * - * Private functions. - */ -#include <sys/cdefs.h> - -enum asn_err snmp_binding_encode(struct asn_buf *, const struct snmp_value *); -enum snmp_code snmp_pdu_encode_header(struct asn_buf *, struct snmp_pdu *); -enum snmp_code snmp_fix_encoding(struct asn_buf *, const struct snmp_pdu *); -enum asn_err snmp_parse_message_hdr(struct asn_buf *b, struct snmp_pdu *pdu, - asn_len_t *lenp); -enum asn_err snmp_parse_pdus_hdr(struct asn_buf *b, struct snmp_pdu *pdu, - asn_len_t *lenp); - -#define DEFAULT_HOST "localhost" -#define DEFAULT_PORT "snmp" -#define DEFAULT_LOCAL "/var/run/snmp.sock" diff --git a/src/common/hash.c b/src/common/hash.c deleted file mode 100644 index 789e85c..0000000 --- a/src/common/hash.c +++ /dev/null @@ -1,380 +0,0 @@ -/* - * Copyright (c) 2004, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - */ - -/* - * Originally from apache 2.0 - * Modifications for general use by <nielsen@memberwebs.com> - */ - -/* Copyright 2000-2004 The Apache Software Foundation - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -#include <sys/types.h> -#include <stdlib.h> -#include "hash.h" - -#define KEY_DATA(he) ((he)->key) - -/* - * The internal form of a hash table. - * - * The table is an array indexed by the hash of the key; collisions - * are resolved by hanging a linked list of hash entries off each - * element of the array. Although this is a really simple design it - * isn't too bad given that pools have a low allocation overhead. - */ - -typedef struct hsh_entry_t hsh_entry_t; - -struct hsh_entry_t -{ - hsh_entry_t* next; - unsigned int hash; - const void* key; - size_t klen; - const void* val; -}; - -/* - * Data structure for iterating through a hash table. - * - * We keep a pointer to the next hash entry here to allow the current - * hash entry to be freed or otherwise mangled between calls to - * hsh_next(). - */ -struct hsh_index_t -{ - hsh_t* ht; - hsh_entry_t* ths; - hsh_entry_t* next; - unsigned int index; -}; - -/* - * The size of the array is always a power of two. We use the maximum - * index rather than the size so that we can use bitwise-AND for - * modular arithmetic. - * The count of hash entries may be greater depending on the chosen - * collision rate. - */ -struct hsh_t -{ - hsh_entry_t** array; - hsh_index_t iterator; /* For hsh_first(...) */ - unsigned int count; - unsigned int max; -}; - - -#define INITIAL_MAX 15 /* tunable == 2^n - 1 */ -#define int_malloc malloc -#define int_calloc calloc -#define int_free free - -/* - * Hash creation functions. - */ - -static hsh_entry_t** alloc_array(hsh_t* ht, unsigned int max) -{ - return (hsh_entry_t**)int_calloc(sizeof(*(ht->array)), (max + 1)); -} - -hsh_t* hsh_create() -{ - hsh_t* ht = int_malloc(sizeof(hsh_t)); - if(ht) - { - ht->count = 0; - ht->max = INITIAL_MAX; - ht->array = alloc_array(ht, ht->max); - if(!ht->array) - { - int_free(ht); - return NULL; - } - } - return ht; -} - -void hsh_free(hsh_t* ht) -{ - hsh_index_t* hi; - - for(hi = hsh_first(ht); hi; hi = hsh_next(hi)) - int_free(hi->ths); - - if(ht->array) - int_free(ht->array); - - int_free(ht); -} - -/* - * Hash iteration functions. - */ - -hsh_index_t* hsh_next(hsh_index_t* hi) -{ - hi->ths = hi->next; - while(!hi->ths) - { - if(hi->index > hi->ht->max) - return NULL; - - hi->ths = hi->ht->array[hi->index++]; - } - hi->next = hi->ths->next; - return hi; -} - -hsh_index_t* hsh_first(hsh_t* ht) -{ - hsh_index_t* hi = &ht->iterator; - - hi->ht = ht; - hi->index = 0; - hi->ths = NULL; - hi->next = NULL; - return hsh_next(hi); -} - -void* hsh_this(hsh_index_t* hi, const void** key, size_t* klen) -{ - if(key) - *key = KEY_DATA(hi->ths); - if(klen) - *klen = hi->ths->klen; - return (void*)hi->ths->val; -} - - -/* - * Expanding a hash table - */ - -static int expand_array(hsh_t* ht) -{ - hsh_index_t* hi; - hsh_entry_t** new_array; - unsigned int new_max; - - new_max = ht->max * 2 + 1; - new_array = alloc_array(ht, new_max); - - if(!new_array) - return 0; - - for(hi = hsh_first(ht); hi; hi = hsh_next(hi)) - { - unsigned int i = hi->ths->hash & new_max; - hi->ths->next = new_array[i]; - new_array[i] = hi->ths; - } - - if(ht->array) - free(ht->array); - - ht->array = new_array; - ht->max = new_max; - return 1; -} - -/* - * This is where we keep the details of the hash function and control - * the maximum collision rate. - * - * If val is non-NULL it creates and initializes a new hash entry if - * there isn't already one there; it returns an updatable pointer so - * that hash entries can be removed. - */ - -static hsh_entry_t** find_entry(hsh_t* ht, const void* key, size_t klen, const void* val) -{ - hsh_entry_t** hep; - hsh_entry_t* he; - const unsigned char* p; - unsigned int hash; - size_t i; - - /* - * This is the popular `times 33' hash algorithm which is used by - * perl and also appears in Berkeley DB. This is one of the best - * known hash functions for strings because it is both computed - * very fast and distributes very well. - * - * The originator may be Dan Bernstein but the code in Berkeley DB - * cites Chris Torek as the source. The best citation I have found - * is "Chris Torek, Hash function for text in C, Usenet message - * <27038@mimsy.umd.edu> in comp.lang.c , October, 1990." in Rich - * Salz's USENIX 1992 paper about INN which can be found at - * <http://citeseer.nj.nec.com/salz92internetnews.html>. - * - * The magic of number 33, i.e. why it works better than many other - * constants, prime or not, has never been adequately explained by - * anyone. So I try an explanation: if one experimentally tests all - * multipliers between 1 and 256 (as I did while writing a low-level - * data structure library some time ago) one detects that even - * numbers are not useable at all. The remaining 128 odd numbers - * (except for the number 1) work more or less all equally well. - * They all distribute in an acceptable way and this way fill a hash - * table with an average percent of approx. 86%. - * - * If one compares the chi^2 values of the variants (see - * Bob Jenkins ``Hashing Frequently Asked Questions'' at - * http://burtleburtle.net/bob/hash/hashfaq.html for a description - * of chi^2), the number 33 not even has the best value. But the - * number 33 and a few other equally good numbers like 17, 31, 63, - * 127 and 129 have nevertheless a great advantage to the remaining - * numbers in the large set of possible multipliers: their multiply - * operation can be replaced by a faster operation based on just one - * shift plus either a single addition or subtraction operation. And - * because a hash function has to both distribute good _and_ has to - * be very fast to compute, those few numbers should be preferred. - * - * -- Ralf S. Engelschall <rse@engelschall.com> - */ - hash = 0; - - if(klen == HSH_KEY_STRING) - { - for(p = key; *p; p++) - hash = hash * 33 + *p; - - klen = p - (const unsigned char *)key; - } - else - { - for(p = key, i = klen; i; i--, p++) - hash = hash * 33 + *p; - } - - /* scan linked list */ - for(hep = &ht->array[hash & ht->max], he = *hep; - he; hep = &he->next, he = *hep) - { - if(he->hash == hash && - he->klen == klen && - memcmp(KEY_DATA(he), key, klen) == 0) - break; - } - - if(he || !val) - return hep; - - /* add a new entry for non-NULL val */ - he = int_malloc(sizeof(*he)); - - if(he) - { - /* Key points to external data */ - he->key = key; - he->klen = klen; - - he->next = NULL; - he->hash = hash; - he->val = val; - - *hep = he; - ht->count++; - } - - return hep; -} - -void* hsh_get(hsh_t* ht, const void *key, size_t klen) -{ - hsh_entry_t** he = find_entry(ht, key, klen, NULL); - - if(he && *he) - return (void*)((*he)->val); - else - return NULL; -} - -int hsh_set(hsh_t* ht, const void* key, size_t klen, void* val) -{ - hsh_entry_t** hep = find_entry(ht, key, klen, val); - - if(hep && *hep) - { - /* replace entry */ - (*hep)->val = val; - - /* check that the collision rate isn't too high */ - if(ht->count > ht->max) - { - if(!expand_array(ht)) - return 0; - } - - return 1; - } - - return 0; -} - -void* hsh_rem(hsh_t* ht, const void* key, size_t klen) -{ - hsh_entry_t** hep = find_entry(ht, key, klen, NULL); - void* val = NULL; - - if(hep && *hep) - { - hsh_entry_t* old = *hep; - *hep = (*hep)->next; - --ht->count; - val = (void*)old->val; - free(old); - } - - return val; -} - -unsigned int hsh_count(hsh_t* ht) -{ - return ht->count; -} - diff --git a/src/common/hash.h b/src/common/hash.h deleted file mode 100644 index df34a7a..0000000 --- a/src/common/hash.h +++ /dev/null @@ -1,151 +0,0 @@ -/* - * Copyright (c) 2004, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - */ - -/* - * Originally from apache 2.0 - * Modifications for general use by <nielsen@memberwebs.com> - */ - -/* Copyright 2000-2004 The Apache Software Foundation - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -#ifndef __HSH_H__ -#define __HSH_H__ - -/* - * OPTIONAL FEATURES - * - * Features to define. You need to build both this file and - * the corresponding hash.c file with whatever options you set here. - * These affect the method signatures, so see the sections below - * for the actual options - */ - -/* - * ARGUMENT DOCUMENTATION - * - * ht: The hashtable - * key: Pointer to the key value - * klen: The length of the key - * val: Pointer to the value - * hi: A hashtable iterator - * stamp: A unix timestamp - */ - - -/* ---------------------------------------------------------------------------------- - * TYPES - */ - -/* Abstract type for hash tables. */ -typedef struct hsh_t hsh_t; - -/* Abstract type for scanning hash tables. */ -typedef struct hsh_index_t hsh_index_t; - -/* ----------------------------------------------------------------------------- - * MAIN - */ - -/* - * hsh_create : Create a hash table - * - returns an allocated hashtable - */ -hsh_t* hsh_create(); - -/* - * hsh_free : Free a hash table - */ -void hsh_free(hsh_t* ht); - -/* - * hsh_count: Number of values in hash table - * - returns the number of entries in hash table - */ -unsigned int hsh_count(hsh_t* ht); - -/* - * hsh_get: Retrieves a value from the hash table - * - returns the value of the entry - */ -void* hsh_get(hsh_t* ht, const void* key, size_t klen); - -/* - * hsh_set: Set a value in the hash table - * - returns 1 if the entry was added properly - */ -int hsh_set(hsh_t* ht, const void* key, size_t klen, void* val); - -/* - * hsh_rem: Remove a value from the hash table - * - returns the value of the removed entry - */ -void* hsh_rem(hsh_t* ht, const void* key, size_t klen); - -/* - * hsh_first: Start enumerating through the hash table - * - returns a hash iterator - */ -hsh_index_t* hsh_first(hsh_t* ht); - -/* - * hsh_next: Enumerate through hash table - * - returns the hash iterator or null when no more entries - */ -hsh_index_t* hsh_next(hsh_index_t* hi); - -/* - * hsh_this: While enumerating get current value - * - returns the value that the iterator currently points to - */ -void* hsh_this(hsh_index_t* hi, const void** key, size_t* klen); - -/* - * This can be passed as 'klen' in any of the above functions to indicate - * a string-valued key, and have hash compute the length automatically. - */ -#define HSH_KEY_STRING (-1) - -#endif /* __HSH_H__ */ diff --git a/src/common/server-mainloop.c b/src/common/server-mainloop.c deleted file mode 100644 index 07f02b2..0000000 --- a/src/common/server-mainloop.c +++ /dev/null @@ -1,424 +0,0 @@ - -#include "usuals.h" -#include <errno.h> -#include <sys/time.h> - -#include "server-mainloop.h" - -typedef struct _socket_callback -{ - int fd; - server_socket_callback callback; - void* arg; - - struct _socket_callback* next; -} -socket_callback; - -typedef struct _timer_callback -{ - struct timeval at; - struct timeval interval; - server_timer_callback callback; - void* arg; - - struct _timer_callback* next; -} -timer_callback; - -typedef struct _server_context -{ - int stopped; - fd_set read_fds; - fd_set write_fds; - int max_fd; - socket_callback* callbacks; - timer_callback* timers; -} -server_context; - -/* Global context */ -static server_context ctx; - -static void -timeval_add(struct timeval* t1, struct timeval* t2) -{ - ASSERT(t1->tv_usec < 1000000); - ASSERT(t2->tv_usec < 1000000); - - t1->tv_sec += t2->tv_sec; - t1->tv_usec += t2->tv_usec; - if(t1->tv_usec >= 1000000) - { - t1->tv_usec -= 1000000; - t1->tv_sec += 1; - } -} - -static void -timeval_subtract(struct timeval* t1, struct timeval* t2) -{ - ASSERT(t1->tv_usec < 1000000); - ASSERT(t2->tv_usec < 1000000); - - t1->tv_sec -= t2->tv_sec; - if(t1->tv_usec < t2->tv_usec) - { - t1->tv_usec += 1000000; - t1->tv_sec -= 1; - } - t1->tv_usec -= t2->tv_usec; -} - -static int -timeval_compare(struct timeval* t1, struct timeval* t2) -{ - ASSERT(t1->tv_usec < 1000000); - ASSERT(t2->tv_usec < 1000000); - - if(t1->tv_sec > t2->tv_sec) - return 1; - else if(t1->tv_sec < t2->tv_sec) - return -1; - else - { - if(t1->tv_usec > t2->tv_usec) - return 1; - else if(t1->tv_usec < t2->tv_usec) - return -1; - else - return 0; - } -} - -#define timeval_empty(tv) \ - ((tv)->tv_sec == 0 && (tv)->tv_usec == 0) - -#define timeval_to_ms(tv) \ - ((((uint64_t)(tv).tv_sec) * 1000L) + (((uint64_t)(tv).tv_usec) / 1000L)) - -static int -timeval_dump(struct timeval* tv) -{ - fprintf(stderr, "{ %d:%d }", tv->tv_sec, tv->tv_usec / 1000); -} - -static int -add_timer(int ms, int oneshot, server_timer_callback callback, void* arg) -{ - struct timeval interval; - timer_callback* cb; - int i; - - ASSERT(ms > 0); - ASSERT(callback != NULL); - - interval.tv_sec = ms / 1000; - interval.tv_usec = (ms % 1000) * 1000; /* into micro seconds */ - - cb = (timer_callback*)calloc(1, sizeof(*cb)); - if(!cb) - { - errno = ENOMEM; - return -1; - } - - if(gettimeofday(&(cb->at), NULL) == -1) - { - free(cb); - return -1; - } - - timeval_add(&(cb->at), &interval); - - if (oneshot) - memset(&(cb->interval), 0, sizeof(cb->interval)); - else - memcpy(&(cb->interval), &interval, sizeof(cb->interval)); - - cb->callback = callback; - cb->arg = arg; - - cb->next = ctx.timers; - ctx.timers = cb; - - return 0; -} - -static timer_callback* -remove_timer(timer_callback* timcb) -{ - timer_callback* cb; - timer_callback* next; - int i; - - if(!ctx.timers) - return; - - /* First in list */; - if(ctx.timers == timcb) - { - cb = ctx.timers; - ctx.timers = ctx.timers->next; - free(cb); - return ctx.timers; - } - - /* One ahead processing of rest */ - for(cb = ctx.timers; cb->next; cb = cb->next) - { - if(cb->next == timcb) - { - next = cb->next->next; - free(cb->next); - cb->next = next; - return cb->next; - } - } -} - -void -server_init() -{ - memset(&ctx, 0, sizeof (ctx)); - FD_ZERO(&ctx.read_fds); - FD_ZERO(&ctx.write_fds); - - ctx.max_fd = -1; - ctx.stopped = 1; - ctx.callbacks = NULL; - ctx.timers = NULL; -} - -void -server_uninit() -{ - timer_callback* timcb; - timer_callback* timn; - socket_callback* sockcb; - socket_callback* sockn; - - for(timcb = ctx.timers; timcb; timcb = timn) - { - timn = timcb->next; - free(timcb); - } - - ctx.timers = NULL; - - for(sockcb = ctx.callbacks; sockcb; sockcb = sockn) - { - sockn = sockcb->next; - free(sockcb); - } - - ctx.timers = NULL; -} - -uint64_t -server_get_time() -{ - struct timeval tv; - if(gettimeofday(&tv, NULL) == -1) - return 0L; - return timeval_to_ms(tv); -} - -int -server_run() -{ - struct timeval* timeout; - struct timeval tv, current; - timer_callback* timcb; - socket_callback* sockcb; - fd_set rfds, wfds; - int r, i; - - /* No watches have been set */ - ASSERT(ctx.max_fd > -1); - - ctx.stopped = 0; - - while(!ctx.stopped) - { - /* Watch for the various fds */ - memcpy(&rfds, &ctx.read_fds, sizeof(rfds)); - memcpy(&wfds, &ctx.write_fds, sizeof(wfds)); - - /* Prepare for timers */ - timeout = NULL; - if(gettimeofday(¤t, NULL) == -1) - return -1; - - /* Cycle through timers */ - for(timcb = ctx.timers; timcb; ) - { - ASSERT(timcb->callback); - - /* Call any timers that have already passed */ - if(timeval_compare(¤t, &timcb->at) >= 0) - { - /* Convert to milliseconds, and make the call */ - r = (timcb->callback)(timeval_to_ms(current), timcb->arg); - - /* Reset timer if so desired */ - if (r == 1 && !timeval_empty(&timcb->interval)) - { - timeval_add(&timcb->at, &timcb->interval); - - /* If the time has already passed, just use current time */ - if(timeval_compare(&(timcb->at), ¤t) <= 0) - memcpy(&(timcb->at), ¤t, sizeof(timcb->at)); - } - - /* Otherwise remove it. Either one shot, or returned 0 */ - else - { - timcb = remove_timer(timcb); - continue; - } - } - - /* Get soonest timer */ - if (!timeout || timeval_compare(timeout, &timcb->at) < 0) - timeout = &timcb->at; - - timcb = timcb->next; - } - - /* Convert to an offset */ - if(timeout) - { - memcpy(&tv, timeout, sizeof(tv)); - timeout = &tv; - timeval_subtract(timeout, ¤t); - } - - /* fprintf(stderr, "selecting with timeout: "); - timeval_dump(timeout); - fprintf(stderr, "\n"); */ - - r = select(ctx.max_fd, &rfds, &wfds, NULL, timeout); - if (r < 0) - { - /* Interrupted so try again, and possibly exit */ - if (errno == EINTR) - continue; - - /* Programmer errors */ - ASSERT (errno != EBADF); - ASSERT (errno != EINVAL); - return r; - } - - /* Timeout, just jump to timeout processing */ - if(r == 0) - continue; - - for(sockcb = ctx.callbacks; sockcb; sockcb = sockcb->next) - { - ASSERT(sockcb->fd != -1); - - /* Call any that are set */ - if (FD_ISSET(sockcb->fd, &rfds)) - (sockcb->callback)(sockcb->fd, SERVER_READ, sockcb->arg); - if (FD_ISSET(sockcb->fd, &wfds)) - (sockcb->callback)(sockcb->fd, SERVER_WRITE, sockcb->arg); - } - } - - return 0; -} - -void -server_stop() -{ - ctx.stopped = 1; -} - -int -server_stopped() -{ - return ctx.stopped; -} - -int -server_watch(int fd, int type, server_socket_callback callback, void* arg) -{ - socket_callback* cb; - int i; - ASSERT(type != 0); - ASSERT(fd != -1); - ASSERT(callback != NULL); - - cb = (socket_callback*)calloc(sizeof(*cb), 1); - if(!cb) - { - errno = ENOMEM; - return -1; - } - - cb->fd = fd; - cb->callback = callback; - cb->arg = arg; - - cb->next = ctx.callbacks; - ctx.callbacks = cb; - - if (type & SERVER_READ) - FD_SET(fd, &ctx.read_fds); - if (type & SERVER_WRITE) - FD_SET(fd, &ctx.write_fds); - - if(fd >= ctx.max_fd) - ctx.max_fd = fd + 1; - - return 0; -} - -void -server_unwatch(int fd) -{ - socket_callback* cb; - int i; - - ASSERT(fd != -1); - - FD_CLR(fd, &ctx.read_fds); - FD_CLR(fd, &ctx.write_fds); - - if(!ctx.callbacks) - return; - - /* First in list */; - if(ctx.callbacks->fd == fd) - { - cb = ctx.callbacks; - ctx.callbacks = cb->next; - free(cb); - return; - } - - /* One ahead processing of rest */ - for(cb = ctx.callbacks; cb->next; cb = cb->next) - { - if(cb->next->fd == fd) - { - cb->next = cb->next->next; - free(cb->next); - return; - } - } -} - -int -server_timer(int ms, server_timer_callback callback, void* arg) -{ - return add_timer(ms, 0, callback, arg); -} - -int -server_oneshot(int ms, server_timer_callback callback, void* arg) -{ - return add_timer(ms, 1, callback, arg); -} diff --git a/src/common/server-mainloop.h b/src/common/server-mainloop.h deleted file mode 100644 index ceff28d..0000000 --- a/src/common/server-mainloop.h +++ /dev/null @@ -1,27 +0,0 @@ - -#ifndef __SERVER_MAINLOOP_H__ -#define __SERVER_MAINLOOP_H__ - -#include <stdint.h> - -/* TODO: Prefix functions with svr */ - -#define SERVER_READ 0x01 -#define SERVER_WRITE 0x02 - -typedef void (*server_socket_callback)(int fd, int type, void* arg); -/* TODO: We should declare our own time type: 'mstime' */ -typedef int (*server_timer_callback)(uint64_t when, void* arg); - -void server_init(); -void server_uninit(); -int server_run(); -void server_stop(); -int server_stopped(); -int server_watch(int fd, int type, server_socket_callback callback, void* arg); -void server_unwatch(int fd); -int server_timer(int length, server_timer_callback callback, void* arg); -int server_oneshot(int length, server_timer_callback callback, void* arg); -uint64_t server_get_time(); - -#endif /* __SERVER_MAINLOOP_H__ */ diff --git a/src/common/sock-any.c b/src/common/sock-any.c deleted file mode 100644 index 1938d5d..0000000 --- a/src/common/sock-any.c +++ /dev/null @@ -1,385 +0,0 @@ -/* - * Copyright (c) 2004, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - * - * CONTRIBUTORS - * Nate Nielsen <nielsen@memberwebs.com> - * - */ - -#include <sys/types.h> -#include <sys/socket.h> - -#include <ctype.h> -#include <stdlib.h> -#include <errno.h> -#include <netdb.h> -#include <string.h> -#include <stdio.h> - -#include "sock-any.h" - -#include <arpa/inet.h> - -#define LOCALHOST_ADDR 0x7F000001 - -int sock_any_pton(const char* addr, struct sockaddr_any* any, int opts) -{ - size_t l; - char buf[256]; - char* t; - char* t2; - int defport = (opts & 0xFFFF); - - memset(any, 0, sizeof(*any)); - - /* Just a port? */ - do - { - #define PORT_CHARS "0123456789" - #define PORT_MIN 1 - #define PORT_MAX 5 - - int port = 0; - - l = strspn(addr, PORT_CHARS); - if(l < PORT_MIN || l > PORT_MAX || addr[l] != 0) - break; - - port = strtol(addr, &t2, 10); - if(*t2 || port <= 0 || port >= 65536) - break; - - any->s.in.sin_port = htons(port); - - /* Fill in the type based on defaults */ -#ifdef HAVE_INET6 - if(opts & SANY_OPT_DEFINET6) - any->s.in.sin_family = AF_INET6; - else -#endif - any->s.in.sin_family = AF_INET; - - /* Fill in the address based on defaults */ - if(opts & SANY_OPT_DEFLOCAL) - { -#ifdef HAVE_INET6 - if(opts & SANY_OPT_DEFINET6) - memcpy(&(any->s.in.sin6_addr), &in6addr_loopback, sizeof(struct in6_addr)); - else -#endif - any->s.in.sin_addr.s_addr = htonl(INADDR_LOOPBACK); - } - - /* - * Note the 'any' option is the default since we zero out - * the entire structure above. - */ - - any->namelen = sizeof(any->s.in); - return AF_INET; - } - while(0); - - /* Look and see if we can parse an ipv4 address */ - do - { - #define IPV4_PORT_CHARS - #define IPV4_CHARS "0123456789." - #define IPV4_MIN 3 - #define IPV4_MAX 21 - - int port = 0; - t = NULL; - - l = strlen(addr); - if(l < IPV4_MIN || l > IPV4_MAX) - break; - - strcpy(buf, addr); - - /* Find the last set that contains just numbers */ - l = strspn(buf, IPV4_CHARS); - if(l < IPV4_MIN) - break; - - /* Either end of string or port */ - if(buf[l] != 0 && buf[l] != ':') - break; - - /* Get the port out */ - if(buf[l] != 0) - { - t = buf + l + 1; - buf[l] = 0; - } - - if(t) - { - port = strtol(t, &t2, 10); - if(*t2 || port <= 0 || port >= 65536) - break; - } - - any->s.in.sin_family = AF_INET; - any->s.in.sin_port = htons((unsigned short)(port <= 0 ? defport : port)); - - if(inet_pton(AF_INET, buf, &(any->s.in.sin_addr)) <= 0) - break; - - any->namelen = sizeof(any->s.in); - return AF_INET; - } - while(0); - -#ifdef HAVE_INET6 - do - { - #define IPV6_CHARS "0123456789:" - #define IPV6_MIN 3 - #define IPV6_MAX 51 - - int port = -1; - t = NULL; - - l = strlen(addr); - if(l < IPV6_MIN || l > IPV6_MAX) - break; - - /* If it starts with a '[' then we can get port */ - if(buf[0] == '[') - { - port = 0; - addr++; - } - - strcpy(buf, addr); - - /* Find the last set that contains just numbers */ - l = strspn(buf, IPV6_CHARS); - if(l < IPV6_MIN) - break; - - /* Either end of string or port */ - if(buf[l] != 0) - { - /* If had bracket, then needs to end with a bracket */ - if(port != 0 || buf[l] != ']') - break; - - /* Get the port out */ - t = buf + l + 1; - - if(*t = ':') - t++; - } - - if(t) - { - port = strtol(t, &t, 10); - if(*t || port <= 0 || port >= 65536) - break; - } - - any->s.in6.sin6_family = AF_INET6; - any->s.in6.sin6_port = htons((unsigned short)port <= 0 : defport : port); - - if(inet_pton(AF_INET6, buf, &(any->s.in6.sin6_addr)) >= 0) - break; - - any->namelen = sizeof(any->s.in6); - return AF_INET6; - } - while(0); -#endif - - /* A unix socket path */ - do - { - /* No colon and must have a path component */ - if(strchr(addr, ':') || !strchr(addr, '/')) - break; - - l = strlen(addr); - if(l >= sizeof(any->s.un.sun_path)) - break; - - any->s.un.sun_family = AF_UNIX; - strcpy(any->s.un.sun_path, addr); - - any->namelen = sizeof(any->s.un) - (sizeof(any->s.un.sun_path) - l); - return AF_UNIX; - } - while(0); - - /* A DNS name and a port? */ - do - { - struct addrinfo* res; - int port = 0; - t = NULL; - - l = strlen(addr); - if(l >= 255 || !isalpha(addr[0])) - break; - - /* Some basic illegal character checks */ - if(strcspn(addr, " /\\") != l) - break; - - strcpy(buf, addr); - - /* Find the last set that contains just numbers */ - t = strchr(buf, ':'); - if(t) - { - *t = 0; - t++; - } - - if(t) - { - port = strtol(t, &t2, 10); - if(*t2 || port <= 0 || port >= 65536) - break; - } - - /* Try and resolve the domain name */ - if(getaddrinfo(buf, NULL, NULL, &res) != 0 || !res) - break; - - memcpy(&(any->s.a), res->ai_addr, sizeof(struct sockaddr)); - any->namelen = res->ai_addrlen; - freeaddrinfo(res); - - port = htons((unsigned short)(port <= 0 ? defport : port)); - - switch(any->s.a.sa_family) - { - case PF_INET: - any->s.in.sin_port = port; - break; -#ifdef HAVE_INET6 - case PF_INET6: - any->s.in6.sin6_port = port; - break; -#endif - }; - - return any->s.a.sa_family; - } - while(0); - - return -1; -} - -int sock_any_ntop(const struct sockaddr_any* any, char* addr, size_t addrlen, int opts) -{ - int len = 0; - int port = 0; - - switch(any->s.a.sa_family) - { - case AF_UNIX: - len = strlen(any->s.un.sun_path); - if(addrlen < len + 1) - { - errno = ENOSPC; - return -1; - } - - strcpy(addr, any->s.un.sun_path); - break; - - case AF_INET: - if(inet_ntop(any->s.a.sa_family, &(any->s.in.sin_addr), addr, addrlen) == NULL) - return -1; - port = ntohs(any->s.in.sin_port); - break; - -#ifdef HAVE_INET6 - case AF_INET6: - if(inet_ntop(any->s.a.sa_family, &(any->s.in6.sin6_addr), addr, addrlen) == NULL) - return -1; - port = ntohs(any->s.in6.sin6_port); - break; -#endif - - default: - errno = EAFNOSUPPORT; - return -1; - } - - if(!(opts & SANY_OPT_NOPORT) && port != 0) - { - strncat(addr, ":", addrlen); - addr[addrlen - 1] = 0; - - len = strlen(addr); - addr += len; - addrlen -= len; - - snprintf(addr, addrlen, "%d", port); - } - - return 0; -} - -int sock_any_cmp(const struct sockaddr_any* a1, const struct sockaddr_any* a2, int opts) -{ - if(a1->s.a.sa_family != a2->s.a.sa_family) - return -1; - - switch(a1->s.a.sa_family) - { - case AF_UNIX: - return strcmp(a1->s.un.sun_path, a2->s.un.sun_path); - - case AF_INET: - if(memcmp(&(a1->s.in.sin_addr), &(a2->s.in.sin_addr), sizeof(a2->s.in.sin_addr)) != 0) - return -1; - if(!(opts && SANY_OPT_NOPORT) && a1->s.in.sin_port != a2->s.in.sin_port) - return -1; - return 0; -#ifdef HAVE_INET6 - case AF_INET6: - if(memcmp(&(a1->s.in6.sin6_addr), &(a2->s.in6.sin6_addr), sizeof(a2->s.in6.sin6_addr)) != 0) - return -1; - if(!(opts && SANY_OPT_NOPORT) && a1->s.in6.sin6_port != a2->s.in6.sin6_port) - return -1; - return 0; -#endif - default: - errno = EAFNOSUPPORT; - return -1; - } -} diff --git a/src/common/sock-any.h b/src/common/sock-any.h deleted file mode 100644 index 31cb13b..0000000 --- a/src/common/sock-any.h +++ /dev/null @@ -1,90 +0,0 @@ -/* - * Copyright (c) 2004, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - * - * CONTRIBUTORS - * Nate Nielsen <nielsen@memberwebs.com> - * - */ - -#ifndef __SOCK_ANY_H__ -#define __SOCK_ANY_H__ - -#include <sys/socket.h> -#include <sys/un.h> -#include <netinet/in.h> - -struct sockaddr_any -{ - union _sockaddr_any - { - /* The header */ - struct sockaddr a; - - /* The different types */ - struct sockaddr_un un; - struct sockaddr_in in; -#ifdef HAVE_INET6 - struct sockaddr_in6 in6; -#endif - } s; - size_t namelen; -}; - -#define SANY_ADDR(any) ((any).s.a) -#define SANY_LEN(any) ((any).namelen) -#define SANY_TYPE(any) ((any).s.a.sa_family) - -int sock_any_pton(const char* addr, struct sockaddr_any* any, int opts); - -/* The default port to fill in when no IP/IPv6 port specified */ -#define SANY_OPT_DEFPORT(p) (int)((p) & 0xFFFF) - -/* When only port specified default to IPANY */ -#define SANY_OPT_DEFANY 0x00000000 - -/* When only port specified default to LOCALHOST */ -#define SANY_OPT_DEFLOCAL 0x00100000 - -/* When only port specified default to IPv6 */ -#ifdef HAVE_INET6 -#define SANY_OPT_DEFINET6 0x00200000 -#endif - -int sock_any_ntop(const struct sockaddr_any* any, char* addr, size_t addrlen, int opts); - -/* Don't print or compare the port */ -#define SANY_OPT_NOPORT 0x01000000 - -int sock_any_cmp(const struct sockaddr_any* a1, const struct sockaddr_any* a2, int opts); - -#endif /* __SOCK_ANY_H__ */ diff --git a/src/common/stringx.c b/src/common/stringx.c deleted file mode 100644 index 37a3df9..0000000 --- a/src/common/stringx.c +++ /dev/null @@ -1,155 +0,0 @@ -/* - * Copyright (c) 2004, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - * - * CONTRIBUTORS - * Nate Nielsen <nielsen@memberwebs.com> - * - */ - -#include <sys/types.h> - -#include <ctype.h> -#include <syslog.h> -#include <stdlib.h> -#include <stdio.h> -#include <strings.h> - -#include "usuals.h" -#include "stringx.h" - -void -remove_cr(char* data) -{ - char* p; - for(p = data; *data; data++, p++) - { - while(*data == '\r') - data++; - *p = *data; - } - - /* Renull terminate */ - *p = 0; -} - -char* -trim_start(const char* data) -{ - while(*data && isspace(*data)) - ++data; - return (char*)data; -} - -void -trim_end(char* data) -{ - char* t = data + strlen(data); - while(t > data && isspace(*(t - 1))) - { - t--; - *t = 0; - } -} - -char* -trim_space(char* data) -{ - data = (char*)trim_start(data); - trim_end(data); - return data; -} - -/* String to bool helper function */ -int strtob(const char* str) -{ - if(strcasecmp(str, "0") == 0 || - strcasecmp(str, "no") == 0 || - strcasecmp(str, "false") == 0 || - strcasecmp(str, "f") == 0 || - strcasecmp(str, "off") == 0) - return 0; - - if(strcasecmp(str, "1") == 0 || - strcasecmp(str, "yes") == 0 || - strcasecmp(str, "true") == 0 || - strcasecmp(str, "t") == 0 || - strcasecmp(str, "on") == 0) - return 1; - - return -1; -} - -size_t -strlcpy(char *dst, const char *src, size_t len) -{ - size_t ret = strlen(dst); - - while (len > 1) { - *dst++ = *src++; - len--; - } - if (len > 0) - *dst = '\0'; - return (ret); -} - -size_t strlcat(char* dst, const char* src, size_t siz) -{ - char* d = dst; - const char* s = src; - size_t n = siz; - size_t dlen; - - /* Find the end of dst and adjust bytes left but don't go past end */ - while(n-- != 0 && *d != '\0') - d++; - dlen = d - dst; - n = siz - dlen; - - if(n == 0) - return dlen + strlen(s); - while(*s != '\0') - { - if(n != 1) - { - *d++ = *s; - n--; - } - - s++; - } - - *d = '\0'; - - return dlen + (s - src); /* count does not include NUL */ -} diff --git a/src/common/stringx.h b/src/common/stringx.h deleted file mode 100644 index 042cf2d..0000000 --- a/src/common/stringx.h +++ /dev/null @@ -1,53 +0,0 @@ -/* - * Copyright (c) 2004, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - * - * CONTRIBUTORS - * Nate Nielsen <nielsen@memberwebs.com> - * - */ - -#ifndef __STRINGX_H__ -#define __STRINGX_H__ - -void remove_cr(char* data); - -char* trim_start(const char* data); -void trim_end(char* data); -char* trim_space(char* data); - -int strtob(const char* str); - -size_t -strlcpy(char *dst, const char *src, size_t len); - -#endif /* __STRINGX_H__ */ diff --git a/src/config.c b/src/config.c deleted file mode 100644 index ca64514..0000000 --- a/src/config.c +++ /dev/null @@ -1,575 +0,0 @@ -/* - * Copyright (c) 2005, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - * - * CONTRIBUTORS - * Nate Nielsen <nielsen@memberwebs.com> - * - */ - -#include "usuals.h" -#include <errno.h> -#include <unistd.h> -#include <syslog.h> -#include <dirent.h> - -#include "stringx.h" -#include "rrdbotd.h" - -/* TODO: Put file names in all the parsing error messages */ - -/* - * These routines parse the configuration files and setup the in memory - * data structures. They're mostly run before becoming a daemon, and just - * exit on error. - */ - -typedef struct _config_ctx -{ - const char* confname; - char* configmem; - uint interval; - uint timeout; - rb_item* items; -} -config_ctx; - -/* ----------------------------------------------------------------------------- - * STRINGS - */ - -#define CONFIG_POLL "poll" -#define CONFIG_INTERVAL "interval" -#define CONFIG_FIELD "field." -#define CONFIG_SNMP "snmp" - -#define FIELD_VALID "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ_-0123456789." - -/* ----------------------------------------------------------------------------- - * CONFIG LOADING - */ - -static rb_item* -sort_items_by_host(rb_item *item) -{ - rb_item *sort = NULL; - rb_item *cur; - rb_item *it; - - while(item) - { - cur = item; - item = item->next; - cur->next = NULL; - - /* First item */ - if(!sort) - { - sort = cur; - continue; - } - - /* Before first item */ - else if(cur->host <= sort->host) - { - cur->next = sort; - sort = cur; - continue; - } - - for(it = sort; it->next; it = it->next) - { - if(cur->host <= sort->next->host) - break; - } - - ASSERT(it); - cur->next = it->next; - it->next = cur; - } - - return sort; -} - -static void -config_done(config_ctx* ctx) -{ - char key[MAXPATHLEN]; - rb_item* it; - rb_poller* poll; - char *t; - - /* No polling specified */ - if(!ctx->items) - return; - - /* Make sure we found an interval */ - if(ctx->interval == 0) - errx(2, "no interval specified in config file: %s", ctx->confname); - - if(ctx->timeout == 0) - ctx->timeout = g_state.timeout; - - /* And a nice key for lookups */ - snprintf(key, sizeof(key), "%d-%d:%s/%s.rrd", ctx->timeout, - ctx->interval, g_state.rrddir, ctx->confname); - key[sizeof(key) - 1] = 0; - - /* See if we have one of these pollers already */ - poll = (rb_poller*)hsh_get(g_state.poll_by_key, key, -1); - if(!poll) - { - poll = (rb_poller*)calloc(1, sizeof(*poll)); - - if(!poll || !hsh_set(g_state.poll_by_key, key, -1, poll)) - errx(1, "out of memory"); - - strcpy(poll->key, key); - t = strchr(poll->key, ':'); - ASSERT(t); - poll->rrdname = t + 1; - - poll->interval = ctx->interval * 1000; - poll->timeout = ctx->timeout * 1000; - - /* Add it to the main lists */ - poll->next = g_state.polls; - g_state.polls = poll; - } - - /* Get the last item and add to the list */ - for(it = ctx->items; it->next; it = it->next) - it->poller = poll; - - ASSERT(it); - it->poller = poll; - - /* Add the items to this poller */ - it->next = poll->items; - poll->items = sort_items_by_host(ctx->items); - - /* - * This remains allocated for the life of the program as - * All the configuration strings are in this memory. - * This allows all the users of these strings not to worry - * about reallocating or freeing them - */ - rb_atexit(free, ctx->configmem); - ctx->configmem = NULL; - - /* Clear current config and get ready for next */ - ctx->items = NULL; - ctx->interval = 0; - - -} - -static void -parse_uri(char *uri, char** scheme, char** host, - char** user, char** path, config_ctx* ctx) -{ - /* Copy only for error messages as we mess with original */ - char* copy = strdup(uri); - char* t; - - *host = NULL; - *path = NULL; - *user = NULL; - - *scheme = strsep(&uri, ":"); - if(uri == NULL) - errx(2, "invalid poll uri (scheme invalid): %s", copy); - - if((uri[0] != '/' && uri[1] != '/')) - errx(2, "invalid poll uri (scheme invalid): %s", copy); - - uri += 2; - *host = strsep(&uri, "/"); - if(*host[0]) - { - /* Parse the user name out from the host */ - t = strchr(*host, '@'); - if(t) - { - *t = 0; - *user = *host; - *host = t + 1; - } - } - - if(!*host[0]) - errx(2, "invalid poll uri (no hostname found): %s", copy); - - if(!uri || !uri[0] || !uri[1]) - errx(2, "invalid poll uri (no pathname found): %s", copy); - - *path = uri; - - /* This copy only for error messages */ - free(copy); -} - -static rb_item* -parse_item(const char* field, char* uri, config_ctx *ctx) -{ - rb_item *ritem; - rb_host *rhost; - - char* host; - char* user; - char* scheme; - char* path; - - /* Parse the SNMP URI */ - parse_uri(uri, &scheme, &host, &user, &path, ctx); - ASSERT(scheme && host && path); - - /* TODO: SNMP version support */ - - /* Currently we only support SNMP pollers */ - if(strcmp(scheme, CONFIG_SNMP) != 0) - errx(2, "invalid poll scheme: %s", scheme); - - /* TODO: THis code assumes all hosts have the same community - the lookups below won't work wehn host/community is different */ - - /* See if we can find an associated host */ - rhost = (rb_host*)hsh_get(g_state.host_by_name, host, -1); - if(!rhost) - { - /* Make a new one if necessary */ - rhost = (rb_host*)calloc(1, sizeof(*rhost)); - - if(!rhost || !hsh_set(g_state.host_by_name, host, -1, rhost)) - errx(1, "out of memory"); - - /* TODO: Version support */ - rhost->version = 1; - rhost->name = host; - rhost->community = user ? user : "public"; - - /* TODO: Eventually resolving should be in a separate thread, - and done regularly */ - if(sock_any_pton(host, &(rhost->address), - SANY_OPT_DEFPORT(161) | SANY_OPT_DEFLOCAL) == -1) - { - rb_message(LOG_WARNING, "couldn't resolve host address (ignoring): %s", host); - free(rhost); - return; - } - - /* And add it to the list */ - rhost->next = g_state.hosts; - g_state.hosts = rhost; - } - - /* Make a new item */ - ritem = calloc(1, sizeof(*ritem)); - if(!ritem) - errx(1, "out of memory"); - - ritem->rrdfield = field; - ritem->host = rhost; - ritem->poller = NULL; /* Set later in config_done */ - ritem->req = NULL; - ritem->value = RB_UNKNOWN; - - /* And parse the OID */ - if(rb_parse_mib(path, &(ritem->snmpfield)) == -1) - errx(2, "invalid OID: %s", path + 1); - - /* And add it to the list */ - ritem->next = ctx->items; - ctx->items = ritem; -} - -static void -config_value(const char* header, const char* name, char* value, - config_ctx* ctx) -{ - if(strcmp(header, "poll") != 0) - return; - - if(strcmp(name, CONFIG_INTERVAL) == 0) - { - char* t; - int i; - - if(ctx->interval > 0) - errx(2, CONFIG_INTERVAL " specified twice: %s", value); - - i = strtol(value, &t, 10); - if(i < 1) - err(2, CONFIG_INTERVAL " must be a positive number: %s", value); - - ctx->interval = (uint32_t)i; - } - - /* If it starts with "field." */ - if(strncmp(name, CONFIG_FIELD, KL(CONFIG_FIELD)) == 0) - { - rb_poller* poll; - const char* field; - const char* t; - - /* Check the name */ - field = name + KL(CONFIG_FIELD); - t = field + strspn(field, FIELD_VALID); - if(*t) - err(2, "the '%s' field name must only contain characters, digits, underscore and dash", field); - - /* Parse out the field */ - parse_item(field, value, ctx); - } -} - -/* ----------------------------------------------------------------------------- - * CONFIG FILES - */ - -static char* -read_config_file(const char* configfile) -{ - char* config = NULL; - FILE* f = NULL; - long len; - - ASSERT(configfile); - - f = fopen(configfile, "r"); - if(f == NULL) - err(1, "couldn't open config file: %s", configfile); - - /* Figure out size */ - if(fseek(f, 0, SEEK_END) == -1 || (len = ftell(f)) == -1 || fseek(f, 0, SEEK_SET) == -1) - err(1, "couldn't seek config file: %s", configfile); - - if((config = (char*)malloc(len + 2)) == NULL) - errx(1, "out of memory"); - - /* And read in one block */ - if(fread(config, 1, len, f) != len) - err(1, "couldn't read config file: %s", configfile); - - fclose(f); - - /* Null terminate the data */ - config[len] = '\n'; - config[len + 1] = 0; - - /* Remove nasty dos line endings */ - remove_cr(config); - - rb_messagex(LOG_DEBUG, "read config file: %s", configfile); - return config; -} - -static void -parse_config_file(const char* configfile, config_ctx *ctx) -{ - char* name = NULL; - char* value = NULL; - char* config; - char* next; - char* header; - char* p; - char* t; - int pos; - - config = read_config_file(configfile); - ctx->configmem = config; - next = config; - - /* Go through lines and process them */ - while((t = strchr(next, '\n')) != NULL) - { - *t = 0; - p = next; /* Do this before cleaning below */ - next = t + 1; - - t = trim_start(p); - - /* Continuation line (had spaces at start) */ - if(p < t && *t) - { - if(!value) - errx(2, "invalid continuation in config: %s", p); - - /* Calculate the end of the current value */ - t = value + strlen(value); - ASSERT(t < p); - - /* Continuations are separated by spaces */ - *t = ' '; - t++; - - continue; - } - - // No continuation hand off value if necessary - if(name && value) - { - rb_messagex(LOG_DEBUG, "parsed configuration value: [%s] %s = %s", header, name, value); - config_value(header, name, value, ctx); - } - - name = NULL; - value = NULL; - - /* Empty lines / comments at start / comments without continuation */ - if(!*t || *p == '#') - continue; - - /* A header */ - if(*p == '[') - { - t = p + strcspn(p, "]"); - if(!*t || t == p + 1) - errx(2, "invalid config header: %s", p); - - *t = 0; - header = trim_space(p + 1); - continue; - } - - /* Look for the break between name = value on the same line */ - t = p + strcspn(p, ":="); - if(!*t) - errx(2, "invalid config line: %s", p); - - /* Null terminate and split value part */ - *t = 0; - t++; - - name = trim_space(p); - value = trim_space(t); - } - - if(name && value) - { - rb_messagex(LOG_DEBUG, "parsed configuration value: [%s] %s = %s", header, name, value); - config_value(header, name, value, ctx); - } - - config_done(ctx); - - /* If nobody claimed this memory then we don't need to keep it around */ - if(ctx->configmem) - free(ctx->configmem); - ctx->configmem = NULL; -} - -void -rb_config_parse() -{ - char configfile[MAXPATHLEN]; - struct dirent* dire; - config_ctx ctx; - DIR* dir; - - /* Setup the hash tables properly */ - g_state.poll_by_key = hsh_create(); - g_state.host_by_name = hsh_create(); - - dir = opendir(g_state.confdir); - if(!dir) - err("couldn't read config directory: %s", g_state.confdir); - - while((dire = readdir(dir)) != NULL) - { - if(dire->d_type != DT_REG && dire->d_type != DT_LNK) - continue; - - /* Build a happy path name */ - snprintf(configfile, MAXPATHLEN, "%s/%s", g_state.confdir, dire->d_name); - configfile[MAXPATHLEN - 1] = 0; - - memset(&ctx, 0, sizeof(ctx)); - ctx.confname = dire->d_name; - - parse_config_file(configfile, &ctx); - } - - closedir(dir); -} - -/* ----------------------------------------------------------------------------- - * FREEING MEMORY - */ - -static void -free_items(rb_item* item) -{ - rb_item* next; - for(; item; item = next) - { - next = item->next; - free(item); - } -} - -static void -free_hosts(rb_host* host) -{ - rb_host* next; - for(; host; host = next) - { - next = host->next; - free(host); - } -} - -static void -free_pollers(rb_poller* poll) -{ - rb_poller* next; - for(; poll; poll = next) - { - free_items(poll->items); - - next = poll->next; - free(poll); - } - -} - -void -rb_config_free() -{ - hsh_free(g_state.poll_by_key); - hsh_free(g_state.host_by_name); - - free_hosts(g_state.hosts); - - /* Note that rb_item's are owned by pollers */ - free_pollers(g_state.polls); -} diff --git a/src/rrd-update.c b/src/rrd-update.c deleted file mode 100644 index b4be997..0000000 --- a/src/rrd-update.c +++ /dev/null @@ -1,134 +0,0 @@ -/* - * Copyright (c) 2005, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - * - * CONTRIBUTORS - * Nate Nielsen <nielsen@memberwebs.com> - * - */ - -#define _GNU_SOURCE - -#include "usuals.h" -#include <errno.h> -#include <syslog.h> -#include <unistd.h> - -#include "stringx.h" -#include "rrdbotd.h" - -#define MAX_NUMLEN 40 - -void rb_rrd_update(rb_poller *poll) -{ - char buf[MAX_NUMLEN]; - const char* argv[5]; - char* template; - char* items; - int r, tlen, ilen; - rb_item *it; - - if(!poll->items) - return; - - tlen = 0; - ilen = 3; - - for(it = poll->items; it; it = it->next) - { - tlen += strlen(it->rrdfield) + 1; - ilen += 40; - } - - template = (char*)calloc(tlen, sizeof(char)); - items = (char*)calloc(ilen, sizeof(char)); - - if(!template || !items) - { - if(items) - free(items); - if(template) - free(template); - rb_messagex(LOG_CRIT, "out of memory"); - return; - } - - /* Put in the right time */ - snprintf(items, ilen, "%lld:", (poll->last_polled / 1000L)); - - /* Build the rest of the arguments */ - for(it = poll->items; it; it = it->next) - { - if(it != poll->items) - { - strlcat(template, ":", tlen); - strlcat(items, ":", ilen); - } - - strlcat(template, it->rrdfield, tlen); - - if(it->value == RB_UNKNOWN) - strlcat(items, "U", ilen); - else - { - snprintf(buf, MAX_NUMLEN, "%.4lf", it->value); - buf[MAX_NUMLEN - 1] = 0; - strlcat(items, buf, ilen); - } - } - - /* Always have to clear before calling rrdtool. klunky :( */ - optind = 0; - opterr = 0; - - /* TODO: We need support for @ to specify when these values occurred */ - - argv[0] = "rrdupdate"; - argv[1] = poll->rrdname; - argv[2] = "-t"; - argv[3] = template; - argv[4] = items; - - rb_messagex(LOG_DEBUG, "updating RRD file: %s", poll->rrdname); - rb_messagex(LOG_DEBUG, "> template: %s", template); - rb_messagex(LOG_DEBUG, "> values: %s", items); - - rrd_clear_error(); - r = rrd_update(5, argv); - - if(r != 0) - rb_messagex(LOG_ERR, "couldn't update rrd file: %s: %s", - poll->rrdname, rrd_get_error()); - - free(template); - free(items); -} diff --git a/src/rrdbotd.c b/src/rrdbotd.c deleted file mode 100644 index 95ab092..0000000 --- a/src/rrdbotd.c +++ /dev/null @@ -1,239 +0,0 @@ -/* - * Copyright (c) 2005, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - * - * CONTRIBUTORS - * Nate Nielsen <nielsen@memberwebs.com> - * - */ - -#include "usuals.h" -#include <errno.h> -#include <unistd.h> -#include <stdarg.h> -#include <syslog.h> - -/* TODO: Abstract these headers away nicely */ -#include <bsnmp/asn1.h> -#include <bsnmp/snmp.h> - -#include "stringx.h" -#include "rrdbotd.h" - -/* TODO: Temporary */ -#include "snmpclient.h" - -/* ----------------------------------------------------------------------------- - * GLOBALS - */ - -/* The one main state object */ -rb_state g_state; - -/* TODO: These should be set from the command line */ -static int daemonized = 0; -static int debug_level = 7; - -/* ----------------------------------------------------------------------------- - * CLEANUP - */ - -typedef struct _exit_stack -{ - voidfunc func; - void* data; - - /* We have a list of these beauties */ - struct _exit_stack* next; -} -exit_stack; - -/* Our exit stack */ -static exit_stack* atexits = NULL; -static int atexit_registered = 0; - -static void -atexit_do_stack(void) -{ - exit_stack* next; - for(; atexits; atexits = next) - { - next = atexits->next; - (atexits->func)(atexits->data); - free(atexits); - } -} - -void -rb_atexit(voidfunc func, void* data) -{ - exit_stack* ae; - - ASSERT(func); - - ae = (exit_stack*)calloc(1, sizeof(exit_stack)); - if(ae) - { - ae->func = func; - ae->data = data; - ae->next = atexits; - atexits = ae; - - if(!atexit_registered) - atexit(atexit_do_stack); - } -} - -/* ----------------------------------------------------------------------------- - * LOGGING - */ - -static void -vmessage (int level, int err, const char* msg, va_list ap) -{ - #define MAX_MSGLEN 1024 - char buf[MAX_MSGLEN]; - int e = errno; - - if(daemonized) { - if (level >= LOG_DEBUG) - return; - } else { - if(debug_level < level) - return; - } - - ASSERT (msg); - snprintf(buf, MAX_MSGLEN, "%s%s", msg, err ? ": " : ""); - - if(err) - strncat(buf, strerror(e), MAX_MSGLEN); - - /* As a precaution */ - buf[MAX_MSGLEN - 1] = 0; - - /* Either to syslog or stderr */ - if (daemonized) - vsyslog (level, buf, ap); - else - vwarnx (buf, ap); -} - -void -rb_messagex (int level, const char* msg, ...) -{ - va_list ap; - va_start(ap, msg); - vmessage(level, 0, msg, ap); - va_end(ap); -} - -void -rb_message (int level, const char* msg, ...) -{ - va_list ap; - va_start(ap, msg); - vmessage(level, 1, msg, ap); - va_end(ap); -} - -/* ----------------------------------------------------------------------------- - * STARTUP - */ - -static void -usage() -{ - fprintf(stderr, "usage: rrdcollectd\n"); - fprintf(stderr, " rrdcollectd -v\n"); - exit(2); -} - -#include <values.h> - -int -main(int argc, char* argv[]) -{ - int daemonize; - char ch; - - /* Initialize the state stuff */ - memset(&g_state, 0, sizeof(g_state)); - - /* TODO: These should come from configure, and from arguments */ - g_state.rrddir = "/data/projects/rrdui/work"; - g_state.confdir = "/data/projects/rrdui/conf"; - g_state.retries = 3; - g_state.timeout = 5; - - /* Parse the arguments nicely */ - while((ch = getopt(argc, argv, "v")) != -1) - { - switch(ch) - { - - /* Print version number */ - case 'v': - printf("rrdcollectd (version %s)\n", VERSION); - exit(0); - break; - - /* Usage information */ - case '?': - default: - usage(); - break; - } - } - - argc -= optind; - argv += optind; - - /* The mainloop server */ - server_init(); - - /* Parse config and setup SNMP system */ - rb_config_parse(); - rb_snmp_engine_init(); - - /* Now let it go */ - if(server_run() == -1) - err(1, "critical failure running SNMP engine"); - - /* Cleanups */ - rb_snmp_engine_uninit(); - rb_config_free(); - server_uninit(); - - return 0; -} - diff --git a/src/rrdbotd.h b/src/rrdbotd.h deleted file mode 100644 index 74f8fd5..0000000 --- a/src/rrdbotd.h +++ /dev/null @@ -1,182 +0,0 @@ -/* - * Copyright (c) 2004, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - * - * CONTRIBUTORS - * Nate Nielsen <nielsen@memberwebs.com> - * - */ - -#ifndef __RRDBOTD_H__ -#define __RRDBOTD_H__ - -#include <values.h> -#include <stdint.h> - -#include "asn1.h" -#include "snmp.h" -#include "sock-any.h" -#include "hash.h" - -/* ----------------------------------------------------------------------------- - * DATA - */ - -typedef uint64_t mstime; -#define RB_UNKNOWN -DBL_MAX - -struct _rb_item; -struct _rb_poller; -struct _rb_host; -struct _rb_request; - -/* - * Note that all the members are either in the config memory - * or inline. This helps us keep memory management simple. - */ - -typedef struct _rb_item -{ - struct _rb_request* req; - - /* Specific to this item */ - const char* rrdfield; - struct snmp_value snmpfield; - - /* The last value / current request */ - double value; - - /* Pointers to related */ - const struct _rb_poller* poller; - const struct _rb_host* host; - - /* Next in list of items */ - struct _rb_item* next; -} -rb_item; - -typedef struct _rb_host -{ - const char* name; - const char* community; - int version; - - /* Host resolving and book keeping */ - struct sockaddr_any address; - mstime interval; - mstime last_resolved; - - /* Next in list of hosts */ - struct _rb_host* next; -} -rb_host; - -typedef struct _rb_poller -{ - /* The hash key is interval-timeout:rrdname */ - char key[MAXPATHLEN]; - - /* This points into the memory above */ - const char* rrdname; - - mstime interval; - mstime timeout; - - /* The things to poll. rb_poller owns this list */ - rb_item* items; - - /* Book keeping */ - mstime last_polled; - - /* Next in list of pollers */ - struct _rb_poller* next; -} -rb_poller; - -typedef struct _rb_state -{ - /* Settings from command line */ - const char* confdir; - const char* rrddir; - uint retries; - uint timeout; - - /* All the pollers/hosts */ - rb_poller* polls; - rb_host* hosts; - - /* Quick lookups for responses */ - hsh_t* poll_by_key; - hsh_t* host_by_name; -} -rb_state; - -/* One global rb_state with all the main settings */ -extern rb_state g_state; - -/* ----------------------------------------------------------------------------- - * UTILITIES (rrdbotd.c) - */ - -void rb_messagex (int level, const char* msg, ...); -void rb_message (int level, const char* msg, ...); - -typedef void (*voidfunc)(void*); -void rb_atexit (voidfunc func, void* data); - -/* ----------------------------------------------------------------------------- - * CONFIG (config.c) - */ - -void rb_config_parse(); -void rb_config_free(); - -/* ----------------------------------------------------------------------------- - * SNMP HELPERS (snmp-help.c) - */ - -int rb_snmp_parse_mib(const char* oid, struct snmp_value* value); - -/* ----------------------------------------------------------------------------- - * SNMP ENGINE (snmp-engine.c) - */ - -void rb_snmp_engine_init(); -void rb_snmp_engine_uninit(); - -/* ----------------------------------------------------------------------------- - * RRD UPDATE CODE (rrd-update.c) - */ - -void rb_rrd_update(rb_poller *poll); - -#endif /* __RRDBOTD_H__ */ diff --git a/src/snmp-engine.c b/src/snmp-engine.c deleted file mode 100644 index e1bff30..0000000 --- a/src/snmp-engine.c +++ /dev/null @@ -1,636 +0,0 @@ -/* - * Copyright (c) 2005, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - * - * CONTRIBUTORS - * Nate Nielsen <nielsen@memberwebs.com> - * - */ - -#include "usuals.h" -#include <errno.h> -#include <syslog.h> - -#include <bsnmp/asn1.h> -#include <bsnmp/snmp.h> - -#include "stringx.h" -#include "rrdbotd.h" -#include "server-mainloop.h" - -/* The socket to use */ -int snmp_socket = -1; - -/* The last request: start at a strange number */ -uint32_t snmp_request = 0x0A0A0A0A; - -/* Since we only deal with one packet at a time, global buffer */ -unsigned char snmp_buffer[0x1000]; - -/* ----------------------------------------------------------------------------- - * REQUESTS - */ - -/* rb_request waaaaayyyyy too big */ -typedef struct _rb_request -{ - /* The SNMP request identifier */ - uint32_t id; - - mstime next_retry; /* Time of the next retry */ - mstime interval; /* How long between retries */ - mstime timeout; /* When this request times out */ - uint sent; /* How many times we've sent */ - - /* The poller and host associated with this request */ - rb_poller* poll; - const rb_host* host; - - /* The actual request data */ - struct snmp_pdu pdu; -} -rb_request; - -/* a scrolling window on a loop */ -static rb_request* requests = NULL; -static int reqhigh = -1; -static int reqlow = -1; -static uint nrequests = 0; - -static rb_request* -new_req() -{ - rb_request* req = NULL; - uint num; - int i, first, overlap = 0; - - if(nrequests) - { - /* We allocate in a loop starting after the last allocation. */ - for(i = (reqhigh + 1) % nrequests; i != reqhigh; - i = (i + 1) % nrequests) - { - /* - * We can overlap past reqlow, but in that case no - * updating reqhigh. This can happen after reallocating. - */ - if(i == reqlow) - overlap = 1; - - if(requests[i].id == 0) - { - req = &(requests[i]); - - if(!overlap) - reqhigh = i; - break; - } - } - } - - if(!req) - { - /* - * A note about the scrolling window and extending allocations... - * The only reason this works is because whenever we reallocate - * reqhigh and reqlow are the same. - */ - ASSERT(reqlow == reqhigh); - - /* Reallocate the request block */ - /* TODO: Once we use less memory this can be higher */ - num = nrequests ? nrequests * 2 : 32; - requests = (rb_request*)realloc(requests, sizeof(rb_request) * num); - if(!requests) - { - /* Note we leave old requests allocated */ - errno = ENOMEM; - return NULL; - } - - /* Clear out the new ones */ - memset(requests + nrequests, 0, sizeof(rb_request) * (num - nrequests)); - - /* We return the next one */ - req = requests + nrequests; - - nrequests = num; - - if(reqhigh == -1) - reqhigh = 0; - if(reqlow == -1) - reqlow = nrequests - 1; - } - - /* A incrementing counter for each request */ - req->id = snmp_request++; - return req; -} - -static rb_request* -find_req(uint32_t id) -{ - int i, first; - - if(!nrequests) - return NULL; - - /* - * Search backwards from the in the scrolling window. This gives - * us as high performance for the high performing pollers and - * less performance for the low ones. - */ - for(i = reqhigh, first = 1; first || i != reqlow; - i = (i ? i : nrequests) - 1) - { - if(id == requests[i].id) - return &(requests[i]); - first = 0; - } - - return NULL; -} - -static void -free_req(rb_request* req) -{ - int i; - - memset(req, 0, sizeof(*req)); - - /* Update the bottom of the scrolling loop */ - for(i = reqlow; i != reqhigh; i = (i + 1) % nrequests) - { - /* If used then done */ - if(requests[i].id) - break; - - /* reqlow is not inclusive */ - reqlow = i; - } -} - -/* ----------------------------------------------------------------------------- - * PACKET HANDLING - */ - -static void -finish_poll(rb_poller* poll, mstime when) -{ -#ifdef _DEBUG - { - rb_item* it; - for(it = poll->items; it; it = it->next) - ASSERT(!it->req); - } -#endif - - /* Update the book-keeping */ - poll->last_polled = when; - - rb_messagex(LOG_DEBUG, "collected poll values. sending them to rrd"); - - /* And send off our collection of values */ - rb_rrd_update(poll); -} - -static void -send_req(rb_request* req, mstime when) -{ - struct asn_buf b; - ssize_t ret; - - b.asn_ptr = snmp_buffer; - b.asn_len = sizeof(snmp_buffer); - - if(snmp_pdu_encode(&(req->pdu), &b)) - rb_message(LOG_CRIT, "couldn't encode snmp buffer"); - else - { - ret = sendto(snmp_socket, snmp_buffer, b.asn_ptr - snmp_buffer, 0, - &SANY_ADDR(req->host->address), SANY_LEN(req->host->address)); - if(ret == -1) - rb_message(LOG_ERR, "couldn't send snmp packet to: %s", req->host->name); - else - rb_messagex(LOG_DEBUG, "sent request '%d' to: %s", req->id, req->host->name); - } - - /* And update our bookkeeping */ - req->sent++; - if(req->sent <= g_state.retries) - req->next_retry = when + req->interval; -} - -static void -timeout_req(rb_request* req, mstime when) -{ - rb_poller* poll = req->poll; - int incomplete = 0; - rb_item* it; - - ASSERT(poll); - - /* - * Marks of this requests items as unknown. Request is - * over, free. See if poller is done - */ - - for(it = poll->items; it; it = it->next) - { - if(it->req == req) - { - rb_messagex(LOG_DEBUG, "value for field '%s' timed out", it->rrdfield); - - it->value = RB_UNKNOWN; - it->req = NULL; - } - - else if(it->req) - incomplete = 1; - } - - free_req(req); - - if(!incomplete) - finish_poll(poll, when); -} - -static void -check_req(rb_request* req, mstime when) -{ - ASSERT(req->id); - - /* See if it's timed out */ - if(when >= req->timeout) - timeout_req(req, when); - - if(!req->next_retry) - return; - - /* Resend if necessary */ - if(when >= req->next_retry) - send_req(req, when); -} - -static void -respond_req(rb_request* req, struct snmp_pdu* pdu, mstime when) -{ - struct snmp_value* value; - rb_poller* poll = req->poll; - rb_item* item; - int incomplete = 0; - int i; - - ASSERT(req->id == pdu->request_id); - - for(i = 0; i < pdu->nbindings; i++) - { - value = &(pdu->bindings[i]); - - for(item = poll->items; item; item = item->next) - { - if(asn_compare_oid(&(value->var), &(item->snmpfield.var)) == 0) - { - const char *msg = NULL; - switch(value->syntax) - { - case SNMP_SYNTAX_NULL: - item->value = RB_UNKNOWN; - break; - case SNMP_SYNTAX_INTEGER: - item->value = value->v.integer; - break; - case SNMP_SYNTAX_COUNTER: - case SNMP_SYNTAX_GAUGE: - case SNMP_SYNTAX_TIMETICKS: - item->value = value->v.uint32; - break; - case SNMP_SYNTAX_COUNTER64: - item->value = value->v.counter64; - break; - case SNMP_SYNTAX_OCTETSTRING: - case SNMP_SYNTAX_OID: - case SNMP_SYNTAX_IPADDRESS: - msg = "snmp server returned non numeric value for field: %s"; - break; - case SNMP_SYNTAX_NOSUCHOBJECT: - case SNMP_SYNTAX_NOSUCHINSTANCE: - case SNMP_SYNTAX_ENDOFMIBVIEW: - msg = "value not available on snmp server for field: %s"; - break; - default: - msg = "snmp server returned invalid or unsupported value for field: %s"; - break; - } - - if(msg) - rb_messagex(LOG_WARNING, msg, item->rrdfield); - else - rb_messagex(LOG_DEBUG, "got value for field '%s': %0.2f", - item->rrdfield, item->value); - - /* Mark this value as done */ - item->req = NULL; - break; - } - } - } - - /* We're done with this request */ - free_req(req); - - /* Now see if the entire request is done */ - for(item = poll->items; item; item = item->next) - { - if(item->req) - return; - } - - - /* And if so then hand off */ - finish_poll(poll, when); -} - -static int -poller_timer(mstime when, void* arg) -{ - rb_poller* poll = (rb_poller*)arg; - const rb_host* last_host = NULL; - rb_request* req = NULL; - rb_item* it; - - /* - * If the previous poll has not completed, then we count it - * as a timeout. - */ - for(it = poll->items; it; it = it->next) - { - if(it->req) - { - ASSERT(it->req->poll == poll); - timeout_req(it->req, when); - } - - /* timeout_req above should have cleared this */ - ASSERT(!it->req); - } - - - for(it = poll->items; it; it = it->next) - { - /* - * We assume that the polled items are sorted by host. Done - * in config.c. This allows us to fire off the least amount - * of requests. Generate new requests when: - * - * - first or new host - * - too many items in the same request - */ - if(!req || it->host != last_host || - req->pdu.nbindings >= SNMP_MAX_BINDINGS) - { - /* Send off last request ... */ - if(req) - send_req(req, when); - - /* ... and make a new one */ - req = new_req(); - if(!req) - { - rb_message(LOG_CRIT, "couldn't allocate a new snmp request"); - return 1; - } - - req->poll = poll; - req->host = it->host; - - rb_messagex(LOG_DEBUG, "preparing request '%d' for: %s@%s", - req->id, req->host->community, req->host->name); - - /* Setup the packet */ - strlcpy(req->pdu.community, req->host->community, sizeof(req->pdu.community)); - req->pdu.request_id = req->id; - req->pdu.version = req->host->version; - req->pdu.type = SNMP_PDU_GET; - req->pdu.error_status = 0; - req->pdu.error_index = 0; - req->pdu.nbindings = 0; - - /* Send interval is 200 ms when poll interval is below 2 seconds */ - req->interval = (poll->interval <= 2000) ? 200L : 600L; - - /* Timeout is for the last packet sent, not first */ - req->timeout = when + (req->interval * ((mstime)g_state.retries)) + poll->timeout; - req->sent = 0; - - last_host = it->host; - } - - /* Add an item to this request */ - req->pdu.bindings[req->pdu.nbindings].var = it->snmpfield.var; - req->pdu.bindings[req->pdu.nbindings].syntax = it->snmpfield.syntax; - req->pdu.nbindings++; - - /* Mark item as active by this request */ - it->req = req; - it->value = RB_UNKNOWN; - } - - if(req) - send_req(req, when); - - return 1; -} - - -static void -receive_resp(int fd, int type, void* arg) -{ - char hostname[MAXPATHLEN]; - struct sockaddr_any from; - struct snmp_pdu pdu; - struct asn_buf b; - rb_request* req; - int len, ret; - int32_t ip; - - ASSERT(snmp_socket == fd); - - /* Read in the packet */ - - SANY_LEN(from) = sizeof(from); - len = recvfrom(snmp_socket, snmp_buffer, sizeof(snmp_buffer), 0, - &SANY_ADDR(from), &SANY_LEN(from)); - if(len < 0) - { - if(errno != EAGAIN && errno != EWOULDBLOCK) - rb_message(LOG_ERR, "error receiving snmp packet from network"); - return; - } - - - if(sock_any_ntop(&from, hostname, MAXPATHLEN, 0) == -1) - strcpy(hostname, "[UNKNOWN]"); - - rb_messagex(LOG_DEBUG, "received packet from: %s", hostname); - - /* Now parse the packet */ - - b.asn_ptr = snmp_buffer; - b.asn_len = len; - - ret = snmp_pdu_decode(&b, &pdu, &ip); - if(ret != SNMP_CODE_OK) - { - rb_message(LOG_WARNING, "invalid snmp packet received from: %s", hostname); - return; - } - - /* Lookup and check the request properly */ - - req = find_req(pdu.request_id); - if(!req) - return; - - if(pdu.error_status != SNMP_ERR_NOERROR) - { - /* TODO: Textual errors */ - rb_message(LOG_ERR, "snmp error '%d' from host: %s", pdu.error_status, hostname); - return; - } - - if(pdu.version != req->pdu.version) - rb_message(LOG_WARNING, "wrong version snmp packet from: %s", hostname); - - rb_messagex(LOG_DEBUG, "response to request '%d' from: %s", req->id, hostname); - respond_req(req, &pdu, server_get_time()); -} - -static int -resend_timer(mstime when, void* arg) -{ - int i; - - /* Search forwards through the scrolling window */ - for(i = (reqlow + 1) % nrequests; i <= reqhigh; - i = (i + 1) % nrequests) - { - if(requests[i].id) - check_req(&(requests[i]), when); - } - - return 1; -} - -static int -prep_timer(mstime when, void* arg) -{ - /* - * We don't prepare all timers at exactly the same time - * but we sort of randomly start the various timers. We're - * going to be hitting these over and over again, so there's - * lots of benefits to spreading them out randomly over a - * few seconds. - */ - - rb_poller* poll; - int next; - - /* All done? */ - if(!arg) - return 0; - - poll = (rb_poller*)arg; - if(server_timer(poll->interval, poller_timer, poll) == -1) - rb_message(LOG_CRIT, "couldn't setup poller timer"); - - /* Setup the next poller anywhere between 0 and 750 ms */ - next = rand() % 750; - server_oneshot(next, prep_timer, poll->next); - return 0; -} - - -void -rb_snmp_engine_init() -{ - struct sockaddr_in addr; - rb_request* req; - - ASSERT(snmp_socket == -1); - snmp_socket = socket(PF_INET, SOCK_DGRAM, 0); - if(snmp_socket < 0) - err(1, "couldn't open snmp socket"); - - /* Get a random IPv4 UDP socket for client use */ - memset(&addr, 0, sizeof(addr)); - addr.sin_family = AF_INET; - - if(bind(snmp_socket, (struct sockaddr*)&addr, sizeof(addr)) < 0) - err(1, "couldn't listen on port"); - - if (server_watch(snmp_socket, SERVER_READ, receive_resp, NULL) == -1) - err(1, "couldn't listen on socket"); - - /* Allocate some requests to make sure we have memory */ - req = new_req(); - if(!req) - err(1, "out of memory"); - free_req(req); - - /* Start the preparation timers for setting up randomly */ - if(server_oneshot(100, prep_timer, g_state.polls) == -1) - err(1, "couldn't setup timer"); - - /* We fire off the resend timer every 1/5 second */ - if(server_timer(200, resend_timer, NULL) == -1) - err(1, "couldn't setup timer"); -} - -void -rb_snmp_engine_uninit() -{ - if(snmp_socket != -1) - { - server_unwatch(snmp_socket); - close(snmp_socket); - snmp_socket = -1; - } - - if(requests) - { - free(requests); - nrequests = 0; - reqhigh = 0; - reqlow = 0; - } -} diff --git a/src/snmp-help.c b/src/snmp-help.c deleted file mode 100644 index cf13d40..0000000 --- a/src/snmp-help.c +++ /dev/null @@ -1,111 +0,0 @@ -/* - * Copyright (c) 2005, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - * - * CONTRIBUTORS - * Nate Nielsen <nielsen@memberwebs.com> - * - */ - -#include "usuals.h" -#include <errno.h> -#include <syslog.h> - -#include <bsnmp/asn1.h> -#include <bsnmp/snmp.h> - -#include "stringx.h" -#include "rrdbotd.h" - -static int -parse_numeric_mib(const char* mib, struct asn_oid* oid) -{ - int ret = 0; - unsigned int sub; - char* next; - char* t; - char* copy; - char* src; - - memset(oid, 0, sizeof(*oid)); - - copy = src = strdup(mib); - if(!src) - return -1; - - while(src && *src) - { - next = strchr(src, '.'); - if(next) - *next = 0; - - sub = strtoul(src, &t, 10); - - /* Too many parts */ - if(oid->len > ASN_MAXOIDLEN) - ret = -1; - - /* An invalid number */ - if(*t) - ret = -1; - - /* Make sure this is a valid part */ - if((oid->len == 0 && sub < 1) || sub < 0 || sub >= ASN_MAXID) - ret -1; - - if(ret < 0) - break; - - oid->subs[oid->len] = sub; - oid->len++; - - src = next ? next + 1 : NULL; - } - - free(copy); - return ret; -} - -int -rb_parse_mib(const char* mib, struct snmp_value* value) -{ - /* - * TODO: Eventually we'll have code here to parse symbolic - * names, and initialize snmp_value to the right type and - * all that jazz. For now we just have numeric OID support. - */ - - value->syntax = SNMP_SYNTAX_NULL; - memset(&(value->v), 0, sizeof(value->v)); - - return parse_numeric_mib(mib, &(value->var)); -} diff --git a/src/snmpclient.c b/src/snmpclient.c deleted file mode 100644 index 5d316ad..0000000 --- a/src/snmpclient.c +++ /dev/null @@ -1,967 +0,0 @@ -/* - * Copyright (c) 2004-2005 - * Hartmut Brandt. - * All rights reserved. - * Copyright (c) 2001-2003 - * Fraunhofer Institute for Open Communication Systems (FhG Fokus). - * All rights reserved. - * - * Author: Harti Brandt <harti@freebsd.org> - * Kendy Kutzner - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * 1. Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * 2. Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * - * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND - * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE - * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE - * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL - * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS - * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) - * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT - * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY - * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - * - * $Begemot: bsnmp/lib/snmpclient.c,v 1.31 2005/05/23 11:10:13 brandt_h Exp $ - * - * Support functions for SNMP clients. - */ -#include <sys/types.h> -#include <sys/time.h> -#include <sys/queue.h> -#include <sys/socket.h> -#include <sys/un.h> -#include <stdio.h> -#include <stdlib.h> -#include <stddef.h> -#include <stdarg.h> -#include <string.h> -#include <errno.h> -#include <unistd.h> -#include <fcntl.h> -#include <netdb.h> -#include <stdint.h> -#include <limits.h> -#ifdef HAVE_ERR_H -#include <err.h> -#endif - -#include "asn1.h" -#include "snmp.h" -#include "snmpclient.h" -#include "snmppriv.h" - -/* ---------------------------------------------------------------------------- */ - -#define LIST_EMPTY(head) ((head)->lh_first == NULL) -#define LIST_FIRST(head) ((head)->lh_first) -#define LIST_NEXT(elm, field) ((elm)->field.le_next) -#define LIST_FOREACH(var, head, field) \ - for ((var) = LIST_FIRST((head)); \ - (var); \ - (var) = LIST_NEXT((var), field)) - - -/* ---------------------------------------------------------------------------- */ - -/* global context */ -struct snmp_client snmp_client; - -/* List of all outstanding requests */ -struct sent_pdu { - int reqid; - struct snmp_pdu *pdu; - struct timeval time; - u_int retrycount; - snmp_send_cb_f callback; - void *arg; - void *timeout_id; - LIST_ENTRY(sent_pdu) entries; -}; -LIST_HEAD(sent_pdu_list, sent_pdu); - -static struct sent_pdu_list sent_pdus; - -/* - * Set the error string - */ -static void -seterr(struct snmp_client *sc, const char *fmt, ...) -{ - va_list ap; - - va_start(ap, fmt); - vsnprintf(sc->error, sizeof(sc->error), fmt, ap); - va_end(ap); -} - -/* - * Initialize a client structure - */ -void -snmp_client_init(struct snmp_client *c) -{ - memset(c, 0, sizeof(*c)); - - c->version = SNMP_V2c; - c->trans = SNMP_TRANS_UDP; - c->chost = NULL; - c->cport = NULL; - - strcpy(c->read_community, "public"); - strcpy(c->write_community, "private"); - - c->timeout.tv_sec = 3; - c->timeout.tv_usec = 0; - c->retries = 3; - c->dump_pdus = 0; - c->txbuflen = c->rxbuflen = 10000; - - c->fd = -1; - - c->max_reqid = INT32_MAX; - c->min_reqid = 0; - c->next_reqid = 0; -} - - -/* - * Open UDP client socket - */ -static int -open_client_udp(const char *host, const char *port) -{ - int error; - char *ptr; - struct addrinfo hints, *res0, *res; - - /* copy host- and portname */ - if (snmp_client.chost == NULL) { - if ((snmp_client.chost = malloc(1 + sizeof(DEFAULT_HOST))) - == NULL) { - seterr(&snmp_client, "%s", strerror(errno)); - return (-1); - } - strcpy(snmp_client.chost, DEFAULT_HOST); - } - if (host != NULL) { - if ((ptr = malloc(1 + strlen(host))) == NULL) { - seterr(&snmp_client, "%s", strerror(errno)); - return (-1); - } - free(snmp_client.chost); - snmp_client.chost = ptr; - strcpy(snmp_client.chost, host); - } - if (snmp_client.cport == NULL) { - if ((snmp_client.cport = malloc(1 + sizeof(DEFAULT_PORT))) - == NULL) { - seterr(&snmp_client, "%s", strerror(errno)); - return (-1); - } - strcpy(snmp_client.cport, DEFAULT_PORT); - } - if (port != NULL) { - if ((ptr = malloc(1 + strlen(port))) == NULL) { - seterr(&snmp_client, "%s", strerror(errno)); - return (-1); - } - free(snmp_client.cport); - snmp_client.cport = ptr; - strcpy(snmp_client.cport, port); - } - - /* open connection */ - memset(&hints, 0, sizeof(hints)); - hints.ai_flags = AI_CANONNAME; - hints.ai_family = AF_INET; - hints.ai_socktype = SOCK_DGRAM; - hints.ai_protocol = 0; - error = getaddrinfo(snmp_client.chost, snmp_client.cport, &hints, &res0); - if (error != 0) { - seterr(&snmp_client, "%s: %s", snmp_client.chost, - gai_strerror(error)); - return (-1); - } - res = res0; - for (;;) { - if ((snmp_client.fd = socket(res->ai_family, res->ai_socktype, - res->ai_protocol)) == -1) { - if ((res = res->ai_next) == NULL) { - seterr(&snmp_client, "%s", strerror(errno)); - freeaddrinfo(res0); - return (-1); - } - } else if (connect(snmp_client.fd, res->ai_addr, - res->ai_addrlen) == -1) { - if ((res = res->ai_next) == NULL) { - seterr(&snmp_client, "%s", strerror(errno)); - freeaddrinfo(res0); - return (-1); - } - } else - break; - } - freeaddrinfo(res0); - return (0); -} - -/* - * SNMP_OPEN - */ -int -snmp_open(const char *host, const char *port, const char *readcomm, - const char *writecomm) -{ - struct timeval tout; - - /* still open ? */ - if (snmp_client.fd != -1) { - errno = EBUSY; - seterr(&snmp_client, "%s", strerror(errno)); - return (-1); - } - - /* copy community strings */ - if (readcomm != NULL) { - strncpy(snmp_client.read_community, readcomm, - sizeof(snmp_client.read_community)); - snmp_client.read_community[sizeof(snmp_client.read_community) - 1] = 0; - } - if (writecomm != NULL) { - strncpy(snmp_client.write_community, writecomm, - sizeof(snmp_client.write_community)); - snmp_client.write_community[sizeof(snmp_client.write_community) - 1] = 0; - } - - switch (snmp_client.trans) { - - case SNMP_TRANS_UDP: - if (open_client_udp(host, port)) - return (-1); - break; - - default: - seterr(&snmp_client, "bad transport mapping"); - return (-1); - } - tout.tv_sec = 0; - tout.tv_usec = 0; - if (setsockopt(snmp_client.fd, SOL_SOCKET, SO_SNDTIMEO, - &tout, sizeof(struct timeval)) == -1) { - seterr(&snmp_client, "%s", strerror(errno)); - (void)close(snmp_client.fd); - snmp_client.fd = -1; - if (snmp_client.local_path[0] != '\0') - (void)remove(snmp_client.local_path); - return (-1); - } - - /* initialize list */ - LIST_INIT(&sent_pdus); - - return (0); -} - - -/* - * SNMP_CLOSE - * - * closes connection to snmp server - * - function cannot fail - * - clears connection - * - clears list of sent pdus - * - * input: - * void - * return: - * void - */ -void -snmp_close(void) -{ - struct sent_pdu *p1; - - if (snmp_client.fd != -1) { - (void)close(snmp_client.fd); - snmp_client.fd = -1; - if (snmp_client.local_path[0] != '\0') - (void)remove(snmp_client.local_path); - } - while(!LIST_EMPTY(&sent_pdus)){ - p1 = LIST_FIRST(&sent_pdus); - if (p1->timeout_id != NULL) - snmp_client.timeout_stop(p1->timeout_id); - LIST_REMOVE(p1, entries); - free(p1); - } - free(snmp_client.chost); - free(snmp_client.cport); -} - -/* - * initialize a snmp_pdu structure - */ -void -snmp_pdu_create(struct snmp_pdu *pdu, u_int op) -{ - memset(pdu,0,sizeof(struct snmp_pdu)); - if (op == SNMP_PDU_SET) - strncpy(pdu->community, snmp_client.write_community, - sizeof(pdu->community)); - else - strncpy(pdu->community, snmp_client.read_community, - sizeof(pdu->community)); - pdu->community[sizeof(pdu->community) - 1] = 0; - - - pdu->type = op; - pdu->version = snmp_client.version; - pdu->error_status = 0; - pdu->error_index = 0; - pdu->nbindings = 0; -} - -/* add pairs of (struct asn_oid, enum snmp_syntax) to an existing pdu */ -/* added 10/04/02 by kek: check for MAX_BINDINGS */ -int -snmp_add_binding(struct snmp_v1_pdu *pdu, ...) -{ - va_list ap; - const struct asn_oid *oid; - u_int ret; - - va_start(ap, pdu); - - ret = pdu->nbindings; - while ((oid = va_arg(ap, const struct asn_oid *)) != NULL) { - if (pdu->nbindings >= SNMP_MAX_BINDINGS){ - va_end(ap); - return (-1); - } - pdu->bindings[pdu->nbindings].var = *oid; - pdu->bindings[pdu->nbindings].syntax = - va_arg(ap, enum snmp_syntax); - pdu->nbindings++; - } - va_end(ap); - return (ret); -} - - -static int32_t -snmp_next_reqid(struct snmp_client * c) -{ - int32_t i; - - i = c->next_reqid; - if (c->next_reqid >= c->max_reqid) - c->next_reqid = c->min_reqid; - else - c->next_reqid++; - return (i); -} - -/* - * Send request and return request id. - */ -static int32_t -snmp_send_packet(struct snmp_pdu * pdu) -{ - u_char *buf; - struct asn_buf b; - ssize_t ret; - - if ((buf = malloc(snmp_client.txbuflen)) == NULL) { - seterr(&snmp_client, "%s", strerror(errno)); - return (-1); - } - - pdu->request_id = snmp_next_reqid(&snmp_client); - - b.asn_ptr = buf; - b.asn_len = snmp_client.txbuflen; - if (snmp_pdu_encode(pdu, &b)) { - seterr(&snmp_client, "%s", strerror(errno)); - free(buf); - return (-1); - } - - if (snmp_client.dump_pdus) - snmp_pdu_dump(pdu); - - if ((ret = send(snmp_client.fd, buf, b.asn_ptr - buf, 0)) == -1) { - seterr(&snmp_client, "%s", strerror(errno)); - free(buf); - return (-1); - } - free(buf); - - return pdu->request_id; -} - -/* - * to be called when a snmp request timed out - */ -static void -snmp_timeout(void * listentry_ptr) -{ - struct sent_pdu *listentry = listentry_ptr; - -#if 0 - warnx("snmp request %i timed out, attempt (%i/%i)", - listentry->reqid, listentry->retrycount, snmp_client.retries); -#endif - - listentry->retrycount++; - if (listentry->retrycount > snmp_client.retries) { - /* there is no answer at all */ - LIST_REMOVE(listentry, entries); - listentry->callback(listentry->pdu, NULL, listentry->arg); - free(listentry); - } else { - /* try again */ - /* new request with new request ID */ - listentry->reqid = snmp_send_packet(listentry->pdu); - listentry->timeout_id = - snmp_client.timeout_start(&snmp_client.timeout, - snmp_timeout, listentry); - } -} - -int32_t -snmp_pdu_send(struct snmp_pdu *pdu, snmp_send_cb_f func, void *arg) -{ - struct sent_pdu *listentry; - int32_t id; - - if ((listentry = malloc(sizeof(struct sent_pdu))) == NULL) { - seterr(&snmp_client, "%s", strerror(errno)); - return (-1); - } - - /* here we really send */ - if ((id = snmp_send_packet(pdu)) == -1) { - free(listentry); - return (-1); - } - - /* add entry to list of sent PDUs */ - listentry->pdu = pdu; - if (gettimeofday(&listentry->time, NULL) == -1) - warn("gettimeofday() failed"); - - listentry->reqid = pdu->request_id; - listentry->callback = func; - listentry->arg = arg; - listentry->retrycount=1; - listentry->timeout_id = - snmp_client.timeout_start(&snmp_client.timeout, snmp_timeout, - listentry); - - LIST_INSERT_HEAD(&sent_pdus, listentry, entries); - - return (id); -} - -/* - * Receive an SNMP packet. - * - * tv controls how we wait for a packet: if tv is a NULL pointer, - * the receive blocks forever, if tv points to a structure with all - * members 0 the socket is polled, in all other cases tv specifies the - * maximum time to wait for a packet. - * - * Return: - * -1 on errors - * 0 on timeout - * +1 if packet received - */ -static int -snmp_receive_packet(struct snmp_pdu *pdu, struct timeval *tv) -{ - int dopoll, setpoll; - int flags; - int saved_errno; - u_char *buf; - int ret; - struct asn_buf abuf; - int32_t ip; -#ifdef bsdi - int optlen; -#else - socklen_t optlen; -#endif - - if ((buf = malloc(snmp_client.rxbuflen)) == NULL) { - seterr(&snmp_client, "%s", strerror(errno)); - return (-1); - } - dopoll = setpoll = 0; - flags = 0; - if (tv != NULL) { - /* poll or timeout */ - if (tv->tv_sec != 0 || tv->tv_usec != 0) { - /* wait with timeout */ - if (setsockopt(snmp_client.fd, SOL_SOCKET, SO_RCVTIMEO, - tv, sizeof(*tv)) == -1) { - seterr(&snmp_client, "setsockopt: %s", - strerror(errno)); - free(buf); - return (-1); - } - optlen = sizeof(*tv); - if (getsockopt(snmp_client.fd, SOL_SOCKET, SO_RCVTIMEO, - tv, &optlen) == -1) { - seterr(&snmp_client, "getsockopt: %s", - strerror(errno)); - free(buf); - return (-1); - } - /* at this point tv_sec and tv_usec may appear - * as 0. This happens for timeouts lesser than - * the clock granularity. The kernel rounds these to - * 0 and this would result in a blocking receive. - * Instead of an else we check tv_sec and tv_usec - * again below and if this rounding happens, - * switch to a polling receive. */ - } - if (tv->tv_sec == 0 && tv->tv_usec == 0) { - /* poll */ - dopoll = 1; - if ((flags = fcntl(snmp_client.fd, F_GETFL, 0)) == -1) { - seterr(&snmp_client, "fcntl: %s", - strerror(errno)); - free(buf); - return (-1); - } - if (!(flags & O_NONBLOCK)) { - setpoll = 1; - flags |= O_NONBLOCK; - if (fcntl(snmp_client.fd, F_SETFL, flags) == -1) { - seterr(&snmp_client, "fcntl: %s", - strerror(errno)); - free(buf); - return (-1); - } - } - } - } - ret = recv(snmp_client.fd, buf, snmp_client.rxbuflen, 0); - saved_errno = errno; - if (tv != NULL) { - if (dopoll) { - if (setpoll) { - flags &= ~O_NONBLOCK; - (void)fcntl(snmp_client.fd, F_SETFL, flags); - } - } else { - tv->tv_sec = 0; - tv->tv_usec = 0; - (void)setsockopt(snmp_client.fd, SOL_SOCKET, SO_RCVTIMEO, - tv, sizeof(*tv)); - } - } - if (ret == -1) { - free(buf); - if (errno == EAGAIN || errno == EWOULDBLOCK) - return (0); - seterr(&snmp_client, "recv: %s", strerror(saved_errno)); - return (-1); - } - if (ret == 0) { - /* this happens when we have a streaming socket and the - * remote side has closed it */ - free(buf); - seterr(&snmp_client, "recv: socket closed by peer"); - errno = EPIPE; - return (-1); - } - - abuf.asn_ptr = buf; - abuf.asn_len = ret; - - if (SNMP_CODE_OK != (ret = snmp_pdu_decode(&abuf, pdu, &ip))) { - seterr(&snmp_client, "snmp_decode_pdu: failed %d", ret); - free(buf); - return (-1); - } - free(buf); - if (snmp_client.dump_pdus) - snmp_pdu_dump(pdu); - - return (+1); -} - -static int -snmp_deliver_packet(struct snmp_pdu * resp) -{ - struct sent_pdu *listentry; - - if (resp->type != SNMP_PDU_RESPONSE) { - warn("ignoring snmp pdu %u", resp->type); - return (-1); - } - - LIST_FOREACH(listentry, &sent_pdus, entries) - if (listentry->reqid == resp->request_id) - break; - if (listentry == NULL) - return (-1); - - LIST_REMOVE(listentry, entries); - listentry->callback(listentry->pdu, resp, listentry->arg); - - snmp_client.timeout_stop(listentry->timeout_id); - - free(listentry); - return (0); -} - -int -snmp_receive(int blocking) -{ - int ret; - - struct timeval tv; - struct snmp_pdu * resp; - - memset(&tv, 0, sizeof(tv)); - - resp = malloc(sizeof(struct snmp_pdu)); - if (resp == NULL) { - seterr(&snmp_client, "no memory for returning PDU"); - return (-1) ; - } - - if ((ret = snmp_receive_packet(resp, blocking ? NULL : &tv)) <= 0) { - free(resp); - return (ret); - } - ret = snmp_deliver_packet(resp); - snmp_pdu_free(resp); - free(resp); - return (ret); -} - - -/* - * Check a GETNEXT response. Here we have three possible outcomes: -1 an - * unexpected error happened. +1 response is ok and is within the table 0 - * response is ok, but is behind the table or error is NOSUCHNAME. The req - * should point to a template PDU which contains the base OIDs and the - * syntaxes. This is really only useful to sweep non-sparse tables. - */ -static int -ok_getnext(const struct snmp_pdu * req, const struct snmp_pdu * resp) -{ - u_int i; - - if (resp->version != req->version) { - warnx("SNMP GETNEXT: response has wrong version"); - return (-1); - } - - if (resp->error_status == SNMP_ERR_NOSUCHNAME) - return (0); - - if (resp->error_status != SNMP_ERR_NOERROR) { - warnx("SNMP GETNEXT: error %d", resp->error_status); - return (-1); - } - if (resp->nbindings != req->nbindings) { - warnx("SNMP GETNEXT: bad number of bindings in response"); - return (-1); - } - for (i = 0; i < req->nbindings; i++) { - if (!asn_is_suboid(&req->bindings[i].var, - &resp->bindings[i].var)) { - if (i != 0) - warnx("SNMP GETNEXT: inconsistent table " - "response"); - return (0); - } - if (resp->version != SNMP_V1 && - resp->bindings[i].syntax == SNMP_SYNTAX_ENDOFMIBVIEW) - return (0); - - if (resp->bindings[i].syntax != req->bindings[i].syntax) { - warnx("SNMP GETNEXT: bad syntax in response"); - return (0); - } - } - return (1); -} - -/* - * Check a GET response. Here we have three possible outcomes: -1 an - * unexpected error happened. +1 response is ok. 0 NOSUCHNAME The req should - * point to a template PDU which contains the OIDs and the syntaxes. This - * is only useful for SNMPv1 or single object GETS. - */ -static int -ok_get(const struct snmp_pdu * req, const struct snmp_pdu * resp) -{ - u_int i; - - if (resp->version != req->version) { - warnx("SNMP GET: response has wrong version"); - return (-1); - } - - if (resp->error_status == SNMP_ERR_NOSUCHNAME) - return (0); - - if (resp->error_status != SNMP_ERR_NOERROR) { - warnx("SNMP GET: error %d", resp->error_status); - return (-1); - } - - if (resp->nbindings != req->nbindings) { - warnx("SNMP GET: bad number of bindings in response"); - return (-1); - } - for (i = 0; i < req->nbindings; i++) { - if (asn_compare_oid(&req->bindings[i].var, - &resp->bindings[i].var) != 0) { - warnx("SNMP GET: bad OID in response"); - return (-1); - } - if (snmp_client.version != SNMP_V1 && - (resp->bindings[i].syntax == SNMP_SYNTAX_NOSUCHOBJECT || - resp->bindings[i].syntax == SNMP_SYNTAX_NOSUCHINSTANCE)) - return (0); - if (resp->bindings[i].syntax != req->bindings[i].syntax) { - warnx("SNMP GET: bad syntax in response"); - return (-1); - } - } - return (1); -} - -/* - * Check the reponse to a SET PDU. We check: - the error status must be 0 - - * the number of bindings must be equal in response and request - the - * syntaxes must be the same in response and request - the OIDs must be the - * same in response and request - */ -static int -ok_set(const struct snmp_pdu * req, const struct snmp_pdu * resp) -{ - u_int i; - - if (resp->version != req->version) { - warnx("SNMP SET: response has wrong version"); - return (-1); - } - - if (resp->error_status == SNMP_ERR_NOSUCHNAME) { - warnx("SNMP SET: error %d", resp->error_status); - return (0); - } - if (resp->error_status != SNMP_ERR_NOERROR) { - warnx("SNMP SET: error %d", resp->error_status); - return (-1); - } - - if (resp->nbindings != req->nbindings) { - warnx("SNMP SET: bad number of bindings in response"); - return (-1); - } - for (i = 0; i < req->nbindings; i++) { - if (asn_compare_oid(&req->bindings[i].var, - &resp->bindings[i].var) != 0) { - warnx("SNMP SET: wrong OID in response to SET"); - return (-1); - } - if (resp->bindings[i].syntax != req->bindings[i].syntax) { - warnx("SNMP SET: bad syntax in response"); - return (-1); - } - } - return (1); -} - -/* - * Simple checks for response PDUs against request PDUs. Return values: 1=ok, - * 0=nosuchname or similar, -1=failure, -2=no response at all - */ -int -snmp_pdu_check(const struct snmp_pdu *req, - const struct snmp_pdu *resp) -{ - if (resp == NULL) - return (-2); - - switch (req->type) { - - case SNMP_PDU_GET: - return (ok_get(req, resp)); - - case SNMP_PDU_SET: - return (ok_set(req, resp)); - - case SNMP_PDU_GETNEXT: - return (ok_getnext(req, resp)); - - } - errx(1, "%s: bad pdu type %i", __func__, req->type); -} - -int -snmp_dialog(struct snmp_v1_pdu *req, struct snmp_v1_pdu *resp) -{ - u_int i; - int32_t reqid; - int ret; - struct timeval tv = snmp_client.timeout; - struct timeval end; - struct snmp_pdu pdu; - - /* - * Make a copy of the request and replace the syntaxes by NULL - * if this is a GET,GETNEXT or GETBULK. - */ - pdu = *req; - if (pdu.type == SNMP_PDU_GET || pdu.type == SNMP_PDU_GETNEXT || - pdu.type == SNMP_PDU_GETBULK) { - for (i = 0; i < pdu.nbindings; i++) - pdu.bindings[i].syntax = SNMP_SYNTAX_NULL; - } - - for (i = 0; i <= snmp_client.retries; i++) { - (void)gettimeofday(&end, NULL); - timeradd(&end, &snmp_client.timeout, &end); - if ((reqid = snmp_send_packet(&pdu)) == -1) - return (-1); - for (;;) { - (void)gettimeofday(&tv, NULL); - if (timercmp(&end, &tv, <=)) - break; - timersub(&end, &tv, &tv); - if ((ret = snmp_receive_packet(resp, &tv)) == 0) - /* timeout */ - break; - - if (ret > 0) { - if (reqid == resp->request_id) - return (0); - /* not for us */ - (void)snmp_deliver_packet(resp); - } - if (ret < 0 && errno == EPIPE) - /* stream closed */ - return (-1); - } - } - errno = ETIMEDOUT; - seterr(&snmp_client, "retry count exceeded"); - return (-1); -} - -/* - * parse a server specification - * - * [trans::][community@][server][:port] - */ -int -snmp_parse_server(struct snmp_client *sc, const char *str) -{ - const char *p, *s = str; - - /* look for a double colon */ - for (p = s; *p != '\0'; p++) { - if (*p == '\\' && p[1] != '\0') { - p++; - continue; - } - if (*p == ':' && p[1] == ':') - break; - } - if (*p != '\0') { - if (p > s) { - if (p - s == 3 && strncmp(s, "udp", 3) == 0) - sc->trans = SNMP_TRANS_UDP; - else if (p - s == 6 && strncmp(s, "stream", 6) == 0) - sc->trans = SNMP_TRANS_LOC_STREAM; - else if (p - s == 5 && strncmp(s, "dgram", 5) == 0) - sc->trans = SNMP_TRANS_LOC_DGRAM; - else { - seterr(sc, "unknown SNMP transport '%.*s'", - (int)(p - s), s); - return (-1); - } - } - s = p + 2; - } - - /* look for a @ */ - for (p = s; *p != '\0'; p++) { - if (*p == '\\' && p[1] != '\0') { - p++; - continue; - } - if (*p == '@') - break; - } - - if (*p != '\0') { - if (p - s > SNMP_COMMUNITY_MAXLEN) { - seterr(sc, "community string too long"); - return (-1); - } - strncpy(sc->read_community, s, p - s); - sc->read_community[p - s] = '\0'; - strncpy(sc->write_community, s, p - s); - sc->write_community[p - s] = '\0'; - s = p + 1; - } - - /* look for a colon */ - for (p = s; *p != '\0'; p++) { - if (*p == '\\' && p[1] != '\0') { - p++; - continue; - } - if (*p == ':') - break; - } - - if (*p == ':') { - if (p > s) { - /* host:port */ - free(sc->chost); - if ((sc->chost = malloc(p - s + 1)) == NULL) { - seterr(sc, "%s", strerror(errno)); - return (-1); - } - strncpy(sc->chost, s, p - s); - sc->chost[p - s] = '\0'; - } - /* port */ - free(sc->cport); - if ((sc->cport = malloc(strlen(p + 1) + 1)) == NULL) { - seterr(sc, "%s", strerror(errno)); - return (-1); - } - strcpy(sc->cport, p + 1); - - } else if (p > s) { - /* host */ - free(sc->chost); - if ((sc->chost = malloc(strlen(s) + 1)) == NULL) { - seterr(sc, "%s", strerror(errno)); - return (-1); - } - strcpy(sc->chost, s); - } - return (0); -} diff --git a/src/snmpclient.h b/src/snmpclient.h deleted file mode 100644 index f47493f..0000000 --- a/src/snmpclient.h +++ /dev/null @@ -1,187 +0,0 @@ -/* - * Copyright (c) 2001-2003 - * Fraunhofer Institute for Open Communication Systems (FhG Fokus). - * All rights reserved. - * - * Author: Harti Brandt <harti@freebsd.org> - * Kendy Kutzner - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * 1. Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * 2. Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * - * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND - * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE - * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE - * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL - * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS - * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) - * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT - * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY - * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - * - * $Begemot: bsnmp/lib/snmpclient.h,v 1.19 2005/05/23 11:10:14 brandt_h Exp $ - */ -#ifndef _BSNMP_SNMPCLIENT_H -#define _BSNMP_SNMPCLIENT_H - -#include <sys/types.h> -#include <sys/socket.h> -#include <sys/time.h> -#include <netinet/in.h> -#include <stddef.h> - - -#define SNMP_STRERROR_LEN 200 - -#define SNMP_LOCAL_PATH "/tmp/snmpXXXXXXXXXXXXXX" - -/* - * transport methods - */ -#define SNMP_TRANS_UDP 0 -#define SNMP_TRANS_LOC_DGRAM 1 -#define SNMP_TRANS_LOC_STREAM 2 - -/* type of callback function for responses - * this callback function is responsible for free() any memory associated with - * any of the PDUs. Therefor it may call snmp_pdu_free() */ -typedef void (*snmp_send_cb_f)(struct snmp_pdu *, struct snmp_pdu *, void *); - -/* type of callback function for timeouts */ -typedef void (*snmp_timeout_cb_f)(void * ); - -/* timeout start function */ -typedef void *(*snmp_timeout_start_f)(struct timeval *timeout, - snmp_timeout_cb_f callback, void *); - -/* timeout stop function */ -typedef void (*snmp_timeout_stop_f)(void *timeout_id); - -/* - * Client context. - */ -struct snmp_client { - enum snmp_version version; - int trans; /* which transport to use */ - - /* these two are read-only for the application */ - char *cport; /* port number as string */ - char *chost; /* host name or IP address as string */ - - char read_community[SNMP_COMMUNITY_MAXLEN + 1]; - char write_community[SNMP_COMMUNITY_MAXLEN + 1]; - - struct timeval timeout; - u_int retries; - - int dump_pdus; - - size_t txbuflen; - size_t rxbuflen; - - int fd; - - int32_t next_reqid; - int32_t max_reqid; - int32_t min_reqid; - - char error[SNMP_STRERROR_LEN]; - - snmp_timeout_start_f timeout_start; - snmp_timeout_stop_f timeout_stop; - - char local_path[sizeof(SNMP_LOCAL_PATH)]; -}; - -/* the global context */ -extern struct snmp_client snmp_client; - -/* initizialies a snmp_client structure */ -void snmp_client_init(struct snmp_client *); - -/* initialize fields */ -int snmp_client_set_host(struct snmp_client *, const char *); -int snmp_client_set_port(struct snmp_client *, const char *); - -/* open connection to snmp server (hostname or portname can be NULL) */ -int snmp_open(const char *_hostname, const char *_portname, - const char *_read_community, const char *_write_community); - -/* close connection */ -void snmp_close(void); - -/* initialize a snmp_pdu structure */ -void snmp_pdu_create(struct snmp_pdu *, u_int _op); - -/* add pairs of (struct asn_oid *, enum snmp_syntax) to an existing pdu */ -int snmp_add_binding(struct snmp_pdu *, ...); - -/* check wheater the answer is valid or not */ -int snmp_pdu_check(const struct snmp_pdu *_req, const struct snmp_pdu *_resp); - -int32_t snmp_pdu_send(struct snmp_pdu *_pdu, snmp_send_cb_f _func, void *_arg); - -/* append an index to an oid */ -int snmp_oid_append(struct asn_oid *_oid, const char *_fmt, ...); - -/* receive a packet */ -int snmp_receive(int _blocking); - -/* - * This structure is used to describe an SNMP table that is to be fetched. - * The C-structure that is produced by the fetch function must start with - * a TAILQ_ENTRY and an u_int64_t. - */ -struct snmp_table { - /* base OID of the table */ - struct asn_oid table; - /* type OID of the LastChange variable for the table if any */ - struct asn_oid last_change; - /* maximum number of iterations if table has changed */ - u_int max_iter; - /* size of the C-structure */ - size_t entry_size; - /* number of index fields */ - u_int index_size; - /* bit mask of required fields */ - uint64_t req_mask; - - /* indexes and columns to fetch. Ended by a NULL syntax entry */ - struct snmp_table_entry { - /* the column sub-oid, ignored for index fields */ - asn_subid_t subid; - /* the syntax of the column or index */ - enum snmp_syntax syntax; - /* offset of the field into the C-structure. For octet strings - * this points to an u_char * followed by a size_t */ - off_t offset; -#if defined(__GNUC__) && __GNUC__ < 3 - } entries[0]; -#else - } entries[]; -#endif -}; - -/* callback type for table fetch */ -typedef void (*snmp_table_cb_f)(void *_list, void *_arg, int _res); - -/* fetch a table. The argument points to a TAILQ_HEAD */ -int snmp_table_fetch(const struct snmp_table *descr, void *); -int snmp_table_fetch_async(const struct snmp_table *, void *, - snmp_table_cb_f, void *); - -/* send a request and wait for the response */ -int snmp_dialog(struct snmp_pdu *_req, struct snmp_pdu *_resp); - -/* parse a server specification */ -int snmp_parse_server(struct snmp_client *, const char *); - -#endif /* _BSNMP_SNMPCLIENT_H */ diff --git a/src/usuals.h b/src/usuals.h deleted file mode 100644 index f83a117..0000000 --- a/src/usuals.h +++ /dev/null @@ -1,77 +0,0 @@ -/* - * Copyright (c) 2005, Nate Nielsen - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * - * * Redistributions of source code must retain the above - * copyright notice, this list of conditions and the - * following disclaimer. - * * Redistributions in binary form must reproduce the - * above copyright notice, this list of conditions and - * the following disclaimer in the documentation and/or - * other materials provided with the distribution. - * * The names of contributors to this software may not be - * used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF - * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - * - * CONTRIBUTORS - * Nate Nielsen <nielsen@memberwebs.com> - * - */ - -#ifndef __USUALS_H__ -#define __USUALS_H__ - -#include <sys/types.h> - -#include "config.h" - -#include <sys/types.h> -#include <sys/param.h> -#include <stdio.h> -#include <stdlib.h> -#include <errno.h> -#include <string.h> - -#ifndef NULL -#define NULL 0 -#endif - -#ifndef max -#define max(a,b) (((a) > (b)) ? (a) : (b)) -#endif - -#ifndef min -#define min(a,b) (((a) < (b)) ? (a) : (b)) -#endif - -#define countof(x) (sizeof(x) / sizeof(x[0])) - -#ifdef _DEBUG - #include "assert.h" - #define ASSERT(x) assert(x) -#else - #define ASSERT(x) -#endif - -#define KL(s) ((sizeof(s) - 1) / sizeof(char)) -#define RETURN(x) { ret = (x); goto finally; } - -#endif /* __USUALS_H__ */ |