Imported Upstream version 2.6.2
This commit is contained in:
Arnaud Quette
parent
a367d9bc54
commit
45043b58d0
246 changed files with 18228 additions and 1415 deletions
338
tools/nut-scanner/scan_avahi.c
Normal file
338
tools/nut-scanner/scan_avahi.c
Normal file
|
|
@ -0,0 +1,338 @@
|
|||
/* scan_avahi.c: detect NUT avahi services
|
||||
*
|
||||
* Copyright (C) 2011 - Frederic Bohe <fredericbohe@eaton.com>
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation; either version 2 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
*/
|
||||
#include "common.h"
|
||||
|
||||
#ifdef HAVE_AVAHI_CLIENT_CLIENT_H
|
||||
#include "nut-scan.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <assert.h>
|
||||
#include <stdlib.h>
|
||||
#include <time.h>
|
||||
|
||||
#include <avahi-client/client.h>
|
||||
#include <avahi-client/lookup.h>
|
||||
|
||||
#include <avahi-common/simple-watch.h>
|
||||
#include <avahi-common/malloc.h>
|
||||
#include <avahi-common/error.h>
|
||||
|
||||
static AvahiSimplePoll *simple_poll = NULL;
|
||||
static nutscan_device_t * dev_ret = NULL;
|
||||
static long avahi_usec_timeout = 0;
|
||||
|
||||
static void update_device(const char * host_name,const char *ip, uint16_t port,char * text, int proto)
|
||||
{
|
||||
nutscan_device_t * dev = NULL;
|
||||
|
||||
char * t = NULL;
|
||||
char * t_saveptr = NULL;
|
||||
char * phrase = NULL;
|
||||
char * phrase_saveptr = NULL;
|
||||
char * word = NULL;
|
||||
char * value = NULL;
|
||||
char * device = NULL;
|
||||
char * device_saveptr = NULL;
|
||||
int device_found = 0;
|
||||
char buf[6];
|
||||
int buf_size;
|
||||
|
||||
if( text == NULL ) {
|
||||
return;
|
||||
}
|
||||
|
||||
t = strdup(text);
|
||||
phrase = strtok_r(t,"\"",&t_saveptr);
|
||||
while(phrase != NULL ) {
|
||||
word = strtok_r(phrase,"=",&phrase_saveptr);
|
||||
if( word == NULL ) {
|
||||
phrase = strtok_r(NULL,"\"",&t_saveptr);
|
||||
continue;
|
||||
}
|
||||
value = strtok_r(NULL,"=",&phrase_saveptr);
|
||||
if( value == NULL ) {
|
||||
phrase = strtok_r(NULL,"\"",&t_saveptr);
|
||||
continue;
|
||||
}
|
||||
|
||||
if( strcmp(word,"device_list") != 0 ) {
|
||||
phrase = strtok_r(NULL,"\"",&t_saveptr);
|
||||
continue;
|
||||
}
|
||||
|
||||
device = strtok_r(value,";",&device_saveptr);
|
||||
while( device != NULL ) {
|
||||
device_found = 1;
|
||||
dev = nutscan_new_device();
|
||||
dev->type = TYPE_NUT;
|
||||
dev->driver = strdup("nutclient");
|
||||
if( proto == AVAHI_PROTO_INET) {
|
||||
nutscan_add_option_to_device(dev,"desc","IPv4");
|
||||
}
|
||||
if( proto == AVAHI_PROTO_INET6 ) {
|
||||
nutscan_add_option_to_device(dev,"desc","IPv6");
|
||||
}
|
||||
|
||||
if( port != PORT) {
|
||||
/* +5+1+1+1 is for :
|
||||
- port number (max 65535 so 5 characters),
|
||||
- '@' and ':' characters
|
||||
- terminating 0 */
|
||||
buf_size = strlen(device)+strlen(host_name)+
|
||||
5+1+1+1;
|
||||
dev->port=malloc(buf_size);
|
||||
if(dev->port) {
|
||||
snprintf(dev->port,buf_size,"%s@%s:%u",
|
||||
device,host_name,port);
|
||||
}
|
||||
}
|
||||
else {
|
||||
/*+1+1 is for '@' character and terminating 0 */
|
||||
buf_size = strlen(device)+strlen(host_name)+1+1;
|
||||
dev->port=malloc(buf_size);
|
||||
if(dev->port) {
|
||||
snprintf(dev->port,buf_size,"%s@%s",
|
||||
device,host_name);
|
||||
}
|
||||
}
|
||||
if( dev->port ) {
|
||||
dev_ret = nutscan_add_device_to_device(dev_ret,dev);
|
||||
}
|
||||
else {
|
||||
nutscan_free_device(dev);
|
||||
}
|
||||
device = strtok_r(NULL,";",&device_saveptr);
|
||||
};
|
||||
|
||||
phrase = strtok_r(NULL,"\"",&t_saveptr);
|
||||
};
|
||||
free(t);
|
||||
|
||||
/* If no device published in avahi data, try to get the device by
|
||||
connecting directly to upsd */
|
||||
if( !device_found) {
|
||||
snprintf(buf,sizeof(buf),"%u",port);
|
||||
dev = nutscan_scan_nut(ip,ip,buf,avahi_usec_timeout);
|
||||
if(dev) {
|
||||
dev_ret = nutscan_add_device_to_device(dev_ret,dev);
|
||||
}
|
||||
/* add an upsd entry without associated device */
|
||||
else {
|
||||
dev = nutscan_new_device();
|
||||
dev->type = TYPE_NUT;
|
||||
dev->driver = strdup("nutclient");
|
||||
if( proto == AVAHI_PROTO_INET) {
|
||||
nutscan_add_option_to_device(dev,"desc","IPv4");
|
||||
}
|
||||
if( proto == AVAHI_PROTO_INET6 ) {
|
||||
nutscan_add_option_to_device(dev,"desc","IPv6");
|
||||
}
|
||||
if( port != PORT) {
|
||||
/*+1+1 is for ':' character and terminating 0 */
|
||||
/*buf is the string containing the port number*/
|
||||
buf_size = strlen(host_name)+strlen(buf)+1+1;
|
||||
dev->port=malloc(buf_size);
|
||||
if(dev->port) {
|
||||
snprintf(dev->port,buf_size,"%s:%s",
|
||||
host_name,buf);
|
||||
}
|
||||
}
|
||||
else {
|
||||
dev->port=strdup(host_name);
|
||||
}
|
||||
if( dev->port ) {
|
||||
dev_ret = nutscan_add_device_to_device(dev_ret,dev);
|
||||
}
|
||||
else {
|
||||
nutscan_free_device(dev);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void resolve_callback(
|
||||
AvahiServiceResolver *r,
|
||||
AVAHI_GCC_UNUSED AvahiIfIndex interface,
|
||||
AVAHI_GCC_UNUSED AvahiProtocol protocol,
|
||||
AvahiResolverEvent event,
|
||||
const char *name,
|
||||
const char *type,
|
||||
const char *domain,
|
||||
const char *host_name,
|
||||
const AvahiAddress *address,
|
||||
uint16_t port,
|
||||
AvahiStringList *txt,
|
||||
AvahiLookupResultFlags flags,
|
||||
AVAHI_GCC_UNUSED void* userdata) {
|
||||
|
||||
assert(r);
|
||||
|
||||
/* Called whenever a service has been resolved successfully or timed out */
|
||||
|
||||
switch (event) {
|
||||
case AVAHI_RESOLVER_FAILURE:
|
||||
fprintf(stderr, "(Resolver) Failed to resolve service '%s' of type '%s' in domain '%s': %s\n", name, type, domain, avahi_strerror(avahi_client_errno(avahi_service_resolver_get_client(r))));
|
||||
break;
|
||||
|
||||
case AVAHI_RESOLVER_FOUND: {
|
||||
char a[AVAHI_ADDRESS_STR_MAX], *t;
|
||||
|
||||
/* fprintf(stderr, "Service '%s' of type '%s' in domain '%s':\n", name, type, domain); */
|
||||
|
||||
avahi_address_snprint(a, sizeof(a), address);
|
||||
t = avahi_string_list_to_string(txt);
|
||||
/*
|
||||
fprintf(stderr,
|
||||
"\t%s:%u (%s)\n"
|
||||
"\tTXT=%s\n"
|
||||
"\tcookie is %u\n"
|
||||
"\tis_local: %i\n"
|
||||
"\tour_own: %i\n"
|
||||
"\twide_area: %i\n"
|
||||
"\tmulticast: %i\n"
|
||||
"\tcached: %i\n",
|
||||
host_name, port, a,
|
||||
t,
|
||||
avahi_string_list_get_service_cookie(txt),
|
||||
!!(flags & AVAHI_LOOKUP_RESULT_LOCAL),
|
||||
!!(flags & AVAHI_LOOKUP_RESULT_OUR_OWN),
|
||||
!!(flags & AVAHI_LOOKUP_RESULT_WIDE_AREA),
|
||||
!!(flags & AVAHI_LOOKUP_RESULT_MULTICAST),
|
||||
!!(flags & AVAHI_LOOKUP_RESULT_CACHED));
|
||||
*/
|
||||
update_device(host_name,a,port,t,address->proto);
|
||||
avahi_free(t);
|
||||
}
|
||||
}
|
||||
|
||||
avahi_service_resolver_free(r);
|
||||
}
|
||||
|
||||
static void browse_callback(
|
||||
AvahiServiceBrowser *b,
|
||||
AvahiIfIndex interface,
|
||||
AvahiProtocol protocol,
|
||||
AvahiBrowserEvent event,
|
||||
const char *name,
|
||||
const char *type,
|
||||
const char *domain,
|
||||
AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
|
||||
void* userdata) {
|
||||
|
||||
AvahiClient *c = userdata;
|
||||
assert(b);
|
||||
|
||||
/* Called whenever a new services becomes available on the LAN or is removed from the LAN */
|
||||
|
||||
switch (event) {
|
||||
case AVAHI_BROWSER_FAILURE:
|
||||
|
||||
fprintf(stderr, "(Browser) %s\n", avahi_strerror(avahi_client_errno(avahi_service_browser_get_client(b))));
|
||||
avahi_simple_poll_quit(simple_poll);
|
||||
return;
|
||||
|
||||
case AVAHI_BROWSER_NEW:
|
||||
/* fprintf(stderr, "(Browser) NEW: service '%s' of type '%s' in domain '%s'\n", name, type, domain); */
|
||||
|
||||
/* We ignore the returned resolver object. In the callback
|
||||
function we free it. If the server is terminated before
|
||||
the callback function is called the server will free
|
||||
the resolver for us. */
|
||||
|
||||
if (!(avahi_service_resolver_new(c, interface, protocol, name, type, domain, AVAHI_PROTO_UNSPEC, 0, resolve_callback, c)))
|
||||
fprintf(stderr, "Failed to resolve service '%s': %s\n", name, avahi_strerror(avahi_client_errno(c)));
|
||||
|
||||
break;
|
||||
|
||||
case AVAHI_BROWSER_REMOVE:
|
||||
fprintf(stderr, "(Browser) REMOVE: service '%s' of type '%s' in domain '%s'\n", name, type, domain);
|
||||
break;
|
||||
|
||||
case AVAHI_BROWSER_ALL_FOR_NOW:
|
||||
avahi_simple_poll_quit(simple_poll);
|
||||
case AVAHI_BROWSER_CACHE_EXHAUSTED:
|
||||
/* fprintf(stderr, "(Browser) %s\n", event == AVAHI_BROWSER_CACHE_EXHAUSTED ? "CACHE_EXHAUSTED" : "ALL_FOR_NOW"); */
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static void client_callback(AvahiClient *c, AvahiClientState state, AVAHI_GCC_UNUSED void * userdata) {
|
||||
assert(c);
|
||||
|
||||
/* Called whenever the client or server state changes */
|
||||
|
||||
if (state == AVAHI_CLIENT_FAILURE) {
|
||||
fprintf(stderr, "Server connection failure: %s\n", avahi_strerror(avahi_client_errno(c)));
|
||||
avahi_simple_poll_quit(simple_poll);
|
||||
}
|
||||
}
|
||||
|
||||
nutscan_device_t * nutscan_scan_avahi(long usec_timeout)
|
||||
{
|
||||
/* Example service publication
|
||||
* $ avahi-publish -s nut _upsd._tcp 3493 txtvers=1 protovers=1.0.0 device_list="dev1;dev2"
|
||||
*/
|
||||
AvahiClient *client = NULL;
|
||||
AvahiServiceBrowser *sb = NULL;
|
||||
int error;
|
||||
int ret = 1;
|
||||
|
||||
avahi_usec_timeout = usec_timeout;
|
||||
|
||||
/* Allocate main loop object */
|
||||
if (!(simple_poll = avahi_simple_poll_new())) {
|
||||
fprintf(stderr, "Failed to create simple poll object.\n");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
/* Allocate a new client */
|
||||
client = avahi_client_new(avahi_simple_poll_get(simple_poll), 0, client_callback, NULL, &error);
|
||||
|
||||
/* Check wether creating the client object succeeded */
|
||||
if (!client) {
|
||||
fprintf(stderr, "Failed to create client: %s\n", avahi_strerror(error));
|
||||
goto fail;
|
||||
}
|
||||
|
||||
/* Create the service browser */
|
||||
if (!(sb = avahi_service_browser_new(client, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "_upsd._tcp", NULL, 0, browse_callback, client))) {
|
||||
fprintf(stderr, "Failed to create service browser: %s\n", avahi_strerror(avahi_client_errno(client)));
|
||||
goto fail;
|
||||
}
|
||||
|
||||
/* Run the main loop */
|
||||
avahi_simple_poll_loop(simple_poll);
|
||||
|
||||
ret = 0;
|
||||
|
||||
fail:
|
||||
|
||||
/* Cleanup things */
|
||||
if (sb)
|
||||
avahi_service_browser_free(sb);
|
||||
|
||||
if (client)
|
||||
avahi_client_free(client);
|
||||
|
||||
if (simple_poll)
|
||||
avahi_simple_poll_free(simple_poll);
|
||||
|
||||
return dev_ret;
|
||||
}
|
||||
#endif /* HAVE_AVAHI_CLIENT_CLIENT_H */
|
||||
Loading…
Add table
Add a link
Reference in a new issue