diff options
Diffstat (limited to 'common')
| -rw-r--r-- | common/config-parser.c | 34 | ||||
| -rw-r--r-- | common/config-parser.h | 2 | ||||
| -rw-r--r-- | common/snmp-engine.c | 27 | ||||
| -rw-r--r-- | common/snmp-engine.h | 8 | 
4 files changed, 54 insertions, 17 deletions
| diff --git a/common/config-parser.c b/common/config-parser.c index fbb9c70..881838f 100644 --- a/common/config-parser.c +++ b/common/config-parser.c @@ -356,9 +356,24 @@ cfg_parse_dir(const char* dirname, void* data)      return ret;  } +static int +looks_like_port (const char *port) +{ +	if (!port || !*port) +		return 0; + +	while (*port) { +		if (!isdigit(*port)) +			return 0; +		++port; +	} + +	return 1; +} +  const char* -cfg_parse_uri (char *uri, char** scheme, char** host, char** user, -               char** path, char** query) +cfg_parse_uri (char *uri, char** scheme, char** host, char **port, +               char** user, char** path, char** query)  {      char* t; @@ -367,6 +382,7 @@ cfg_parse_uri (char *uri, char** scheme, char** host, char** user,      *user = NULL;      *path = NULL;      *query = NULL; +    *port = NULL;      *scheme = strsep(&uri, ":");      if(uri == NULL || (uri[0] != '/' && uri[1] != '/')) @@ -374,9 +390,10 @@ cfg_parse_uri (char *uri, char** scheme, char** host, char** user,      uri += 2;      *host = strsep(&uri, "/"); + +    /* Parse the community out from the host */      if(*host[0])      { -        /* Parse the community out from the host */          t = strchr(*host, '@');          if(t)          { @@ -386,6 +403,17 @@ cfg_parse_uri (char *uri, char** scheme, char** host, char** user,          }      } +    /* Parse out the port from the uri */ +    if(*host[0]) +    { +        t = strrchr(*host, ':'); +        if(t && looks_like_port(t + 1)) +        { +            *t = 0; +            *port = t + 1; +        } +    } +      if(!*host[0])          return "invalid uri: no host name found"; diff --git a/common/config-parser.h b/common/config-parser.h index c804874..1df5d78 100644 --- a/common/config-parser.h +++ b/common/config-parser.h @@ -51,7 +51,7 @@ int cfg_parse_dir(const char* dirname, void* data);  int cfg_parse_file(const char* filename, void* data, char** memory);  /* A helper for parsing URIs */ -const char* cfg_parse_uri (char *uri, char** scheme, char** host, char** user, char** path, char** query); +const char* cfg_parse_uri (char *uri, char** scheme, char** host, char **port, char** user, char** path, char** query);  /* Parsing snmp, snmp2 snmp2c etc... */  const char* cfg_parse_scheme (const char *input, enum snmp_version *scheme); diff --git a/common/snmp-engine.c b/common/snmp-engine.c index 96fb812..945418e 100644 --- a/common/snmp-engine.c +++ b/common/snmp-engine.c @@ -73,6 +73,7 @@ struct host {  	char key[128];  	char *hostname; +	char *portnum;  	char *community;  	int version; @@ -145,7 +146,7 @@ host_resolve (struct host *host, mstime when)  	log_debug ("resolving host: %s", host->hostname);  	host->last_resolve_try = when;  	host->is_resolving = 0; -	async_resolver_queue (host->hostname, "161", &hints, resolve_cb, host); +	async_resolver_queue (host->hostname, host->portnum, &hints, resolve_cb, host);  }  static int @@ -203,7 +204,8 @@ host_update_interval (struct host *host, mstime interval)  }  static struct host* -host_instance (const char *hostname, const char *community, int version, mstime interval) +host_instance (const char *hostname, const char *portnum, +               const char *community, int version, mstime interval)  {  	struct addrinfo hints, *ai;  	struct host *host; @@ -213,6 +215,9 @@ host_instance (const char *hostname, const char *community, int version, mstime  	ASSERT (hostname);  	initialize = 0; +	if (!portnum) +		portnum = "161"; +  	/*  	 * Build a lookup key. We can only combine requests for the same  	 * host when the version and community match. @@ -230,7 +235,7 @@ host_instance (const char *hostname, const char *community, int version, mstime  		hints.ai_socktype = SOCK_DGRAM;  		hints.ai_flags = AI_NUMERICSERV | AI_NUMERICHOST; -		r = getaddrinfo (hostname, "161", &hints, &ai); +		r = getaddrinfo (hostname, portnum, &hints, &ai);  		/* Ignore error and try to resolve again later */  		if (r == EAI_NONAME || r == EAI_AGAIN || r == EAI_MEMORY) { @@ -281,6 +286,7 @@ host_instance (const char *hostname, const char *community, int version, mstime  		host->version = version;  		host->hostname = strdup (hostname); +		host->portnum = strdup (portnum);  		host->community = strdup (community);  		host->resolve_interval = 0;  		host->last_resolved = 0; @@ -321,8 +327,10 @@ host_cleanup (void)  	for (host = host_list; host; host = next) {  		next = host->next; -	        if (host->hostname) +		if (host->hostname)  			free (host->hostname); +		if (host->portnum) +			free (host->portnum);  		if (host->community)  			free (host->community);  		if (host->prepared) @@ -818,7 +826,8 @@ request_prep_instance (struct host *host, mstime interval, mstime timeout, int r  }  int -snmp_engine_request (const char *hostname, const char *community, int version, +snmp_engine_request (const char *hostname, const char *port, +                     const char *community, int version,                       mstime interval, mstime timeout, int reqtype,                       struct asn_oid *oid, snmp_response func, void *arg)  { @@ -829,7 +838,7 @@ snmp_engine_request (const char *hostname, const char *community, int version,  	ASSERT (func);  	/* Lookup host for request */ -	host = host_instance (hostname, community, version, interval); +	host = host_instance (hostname, port, community, version, interval);  	if (!host)  		return 0; @@ -945,8 +954,8 @@ sync_response (int req, int code, struct snmp_value *value, void *data)  }  int -snmp_engine_sync (const char* host, const char* community, int version, -                  uint64_t interval, uint64_t timeout, int reqtype, +snmp_engine_sync (const char* host, const char *port, const char* community, +                  int version, uint64_t interval, uint64_t timeout, int reqtype,                    struct snmp_value *value)  {  	struct sync_data sync; @@ -958,7 +967,7 @@ snmp_engine_sync (const char* host, const char* community, int version,  	sync.code = 0;  	sync.dest = value; -	sync.id = snmp_engine_request (host, community, version, interval, timeout, +	sync.id = snmp_engine_request (host, port, community, version, interval, timeout,  	                               reqtype, &value->var, sync_response, &sync);  	if (!sync.id) diff --git a/common/snmp-engine.h b/common/snmp-engine.h index 98a45e5..59c733d 100644 --- a/common/snmp-engine.h +++ b/common/snmp-engine.h @@ -8,16 +8,16 @@ typedef void (*snmp_response) (int request, int code, struct snmp_value *value,  void snmp_engine_init (const char *bind_address, int retries); -int  snmp_engine_request (const char* host, const char* community, int version, -                          uint64_t interval, uint64_t timeout, int reqtype, +int  snmp_engine_request (const char* host, const char *port, const char* community, +                          int version, uint64_t interval, uint64_t timeout, int reqtype,                            struct asn_oid *oid, snmp_response func, void *data);  void snmp_engine_cancel (int reqid);  void snmp_engine_flush (void); -int  snmp_engine_sync (const char* host, const char* community, int version, -                       uint64_t interval, uint64_t timeout, int reqtype, +int  snmp_engine_sync (const char* host, const char *port, const char* community, +                       int version, uint64_t interval, uint64_t timeout, int reqtype,                         struct snmp_value *value);  void snmp_engine_stop (void); | 
