317 lines
6.7 KiB
C
317 lines
6.7 KiB
C
|
|
#include <netdb.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
|
|
|
|
#include "cw/capwap.h"
|
|
#include "cw/conn.h"
|
|
#include "cw/sock.h"
|
|
#include "cw/val.h"
|
|
#include "cw/log.h"
|
|
#include "cw/dbg.h"
|
|
#include "cw/timer.h"
|
|
#include "cw/cw.h"
|
|
#include "cw/rand.h"
|
|
|
|
#include "wtp.h"
|
|
|
|
|
|
|
|
|
|
|
|
int cw_select_ac(mavl_t local_cfg, struct cw_DiscoveryResult * dis)
|
|
{
|
|
mlistelem_t *e;
|
|
/*mavl_t iplist;*/
|
|
int en;
|
|
|
|
/*iplist = cw_ktv_create();
|
|
if (iplist == NULL){
|
|
cw_log_errno("cw_select_ac: Can't allocate memory for aciplist");
|
|
return NULL;
|
|
}
|
|
*/
|
|
|
|
en = 0;
|
|
|
|
/* for each discovery response */
|
|
mlist_foreach(e, dis->results) {
|
|
char acname[CAPWAP_MAX_AC_NAME_LEN + 1];
|
|
char key[CW_KTV_MAX_KEY_LEN];
|
|
mavl_t remote_cfg;
|
|
cw_Val_t *val, *ipval;
|
|
int prio, i;
|
|
|
|
remote_cfg = mlistelem_get_ptr(e);
|
|
|
|
/* get ac name */
|
|
val = cw_ktv_get(remote_cfg, "ac-name", CW_TYPE_BSTR16);
|
|
if (val == NULL) {
|
|
/* this should not happen, because AC Name is a
|
|
* amndatory message element */
|
|
prio = 255;
|
|
} else {
|
|
/* Get priority for AC from
|
|
* ac-name-with-priority list */
|
|
val->type->to_str(val, acname, CAPWAP_MAX_AC_NAME_LEN);
|
|
sprintf(key, "ac-name-with-priority/%s", acname);
|
|
prio = cw_ktv_get_byte(local_cfg, key, 255);
|
|
}
|
|
|
|
/* for each control ip address the AC has sent */
|
|
i = 0;
|
|
do {
|
|
sprintf(key, "%s.%d", "capwap-control-ip-address/wtps", i);
|
|
val = cw_ktv_get(remote_cfg, key, CW_TYPE_WORD);
|
|
if (val == NULL)
|
|
break;
|
|
|
|
sprintf(key, "%s.%d", "capwap-control-ip-address/address", i);
|
|
ipval = cw_ktv_get(remote_cfg, key, CW_TYPE_IPADDRESS);
|
|
|
|
sprintf(key, "%04d%05d%04d", prio, val->val.word, en);
|
|
|
|
cw_ktv_add(dis->prio_ip, key, CW_TYPE_SYSPTR, NULL, (uint8_t *) (&ipval),
|
|
sizeof(ipval));
|
|
cw_ktv_add(dis->prio_ac, key, CW_TYPE_SYSPTR, NULL, (uint8_t *)(&remote_cfg),
|
|
sizeof(remote_cfg));
|
|
i++;
|
|
en++;
|
|
} while (1);
|
|
|
|
}
|
|
|
|
|
|
|
|
return 1;
|
|
}
|
|
|
|
|
|
void cw_discovery_free_results(struct cw_DiscoveryResult * dis)
|
|
{
|
|
if (dis->prio_ac != NULL)
|
|
mavl_destroy(dis->prio_ac);
|
|
if (dis->prio_ip != NULL)
|
|
mavl_destroy(dis->prio_ip);
|
|
if (dis->results != NULL)
|
|
mlist_destroy(dis->results);
|
|
}
|
|
|
|
|
|
static void result_del(void * data)
|
|
{
|
|
|
|
mavl_t todelete = *((void**)data);
|
|
mavl_destroy(todelete);
|
|
|
|
|
|
/*mavl_t*/
|
|
}
|
|
|
|
int cw_discovery_init_results(struct cw_DiscoveryResult *dis)
|
|
{
|
|
|
|
dis->results = mlist_create(NULL, result_del, sizeof(void *));
|
|
if (dis->results==NULL)
|
|
goto errX;
|
|
dis->prio_ac=cw_ktv_create();
|
|
if (dis->prio_ac==NULL)
|
|
goto errX;
|
|
dis->prio_ip=cw_ktv_create();
|
|
if (dis->prio_ac==NULL)
|
|
goto errX;
|
|
return 1;
|
|
errX:
|
|
cw_discovery_free_results(dis);
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int run_discovery(struct cw_Conn *conn, struct cw_DiscoveryResult * dis)
|
|
{
|
|
time_t timer;
|
|
struct sockaddr_storage from;
|
|
int delay, min, max;
|
|
|
|
|
|
|
|
min = cw_ktv_get_byte(conn->local_cfg,"capwap-timers/min-discovery-interval",
|
|
CAPWAP_MIN_DISCOVERY_INTERVAL);
|
|
max = cw_ktv_get_byte(conn->local_cfg,"capwap-timers/max-discovery-interval",
|
|
CAPWAP_MAX_DISCOVERY_INTERVAL);
|
|
|
|
delay = cw_randint(min,max);
|
|
if (delay>0){
|
|
cw_dbg(DBG_INFO,"Sleeping for %d seconds before sending Discovery Request.", delay);
|
|
sleep(delay);
|
|
}
|
|
|
|
|
|
conn->capwap_state = CAPWAP_STATE_DISCOVERY;
|
|
|
|
|
|
/* create and send a discovery request message */
|
|
cw_init_request(conn, CAPWAP_MSG_DISCOVERY_REQUEST);
|
|
cw_put_msg(conn, conn->req_buffer);
|
|
conn_send_msg(conn, conn->req_buffer);
|
|
|
|
|
|
timer = cw_timer_start(cw_ktv_get_byte(conn->local_cfg, "discovery-interval",
|
|
CAPWAP_DISCOVERY_INTERVAL))-1;
|
|
|
|
/*discovery_results = mlist_create(NULL, NULL, sizeof(void *));*/
|
|
|
|
|
|
while (!cw_timer_timeout(timer)
|
|
&& conn->capwap_state == CAPWAP_STATE_DISCOVERY) {
|
|
int rc;
|
|
char addr_str[SOCK_ADDR_BUFSIZE];
|
|
|
|
conn->remote_cfg = cw_ktv_create();
|
|
if (conn->remote_cfg == NULL) {
|
|
cw_log_errno("Can't allocate memory for remote_cfg");
|
|
break;
|
|
}
|
|
|
|
rc = cw_read_from(conn, &from);
|
|
if (rc < 0) {
|
|
if (errno == EAGAIN)
|
|
continue;
|
|
|
|
cw_log(LOG_ERROR, "Error reading messages: %s", strerror(errno));
|
|
break;
|
|
}
|
|
cw_dbg(DBG_INFO, "Received Discovery Response from %s",
|
|
sock_addr2str(&from, addr_str));
|
|
mlist_append_ptr(dis->results, conn->remote_cfg);
|
|
}
|
|
|
|
cw_select_ac(conn->local_cfg, dis);
|
|
|
|
|
|
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
* Run discovery for on address (eg broadcast 255.255.255.255)
|
|
*/
|
|
int cw_run_discovery(struct cw_Conn *conn, const char *addr, const char *bindaddr,
|
|
struct cw_DiscoveryResult * dis)
|
|
{
|
|
char sock_buf[SOCK_ADDR_BUFSIZE];
|
|
struct sockaddr_storage dstaddr;
|
|
char caddr[256], control_port[64];
|
|
|
|
int rc;
|
|
/*struct cw_DiscoveryResult dis;*/
|
|
|
|
|
|
/*dis.results = mlist_create(NULL, NULL, sizeof(void *));*/
|
|
|
|
|
|
/* get addr of destination */
|
|
struct addrinfo hints;
|
|
struct addrinfo *res, *res0;
|
|
|
|
|
|
|
|
memset(&hints, 0, sizeof(hints));
|
|
|
|
hints.ai_socktype = SOCK_DGRAM;
|
|
hints.ai_family = AF_UNSPEC;
|
|
|
|
|
|
strncpy(caddr, addr, sizeof(caddr));
|
|
|
|
/* try to convert the given address string to sockaddr */
|
|
rc = sock_strtoaddr(caddr, (struct sockaddr *) &dstaddr);
|
|
if (rc) {
|
|
int iport;
|
|
/* converting to socaddr was successful */
|
|
iport = sock_getport((struct sockaddr *) &dstaddr);
|
|
if (iport != 0)
|
|
sprintf(control_port, "%d", iport);
|
|
else
|
|
sprintf(control_port, "%d", CAPWAP_CONTROL_PORT);
|
|
} else {
|
|
char *port;
|
|
/* converting went wrong, so the address string might
|
|
* be a DNS entry. Look for a colon to find the port. */
|
|
port = strchr(addr, ':');
|
|
if (port) {
|
|
strncpy(control_port, port + 1, sizeof(control_port));
|
|
caddr[port - addr] = 0;
|
|
} else {
|
|
sprintf(control_port, "%d", CAPWAP_CONTROL_PORT);
|
|
}
|
|
}
|
|
|
|
|
|
/*printf("Addr: %s, Port: %s\n",caddr,control_port); */
|
|
|
|
rc = getaddrinfo(caddr, control_port, &hints, &res0);
|
|
if (rc) {
|
|
cw_log(LOG_ERR, "Can't connect to AC %s: %s", caddr, gai_strerror(rc));
|
|
return 0;
|
|
}
|
|
|
|
for (res = res0; res; res = res->ai_next) {
|
|
|
|
int sockfd;
|
|
int opt;
|
|
sockfd = socket(res->ai_family, SOCK_DGRAM, 0);
|
|
if (sockfd == -1) {
|
|
cw_log(LOG_ERR, "Can't create socket for %s: %s", caddr,
|
|
strerror(errno));
|
|
continue;
|
|
}
|
|
|
|
opt = 1;
|
|
if (setsockopt(sockfd, SOL_SOCKET, SO_BROADCAST, &opt, sizeof(opt)) < 0) {
|
|
cw_log(LOG_ERR, "Can't set broadcast sockopt");
|
|
}
|
|
sock_set_recvtimeout(sockfd, 1);
|
|
sock_set_dontfrag(sockfd, 0);
|
|
|
|
sock_copyaddr(&conn->addr, res->ai_addr);
|
|
|
|
|
|
if (bindaddr) {
|
|
int brc;
|
|
|
|
struct sockaddr bind_address;
|
|
sock_strtoaddr(bindaddr, &bind_address);
|
|
brc = bind(sockfd, &bind_address, sock_addrlen(&bind_address));
|
|
if (brc < 0) {
|
|
cw_log(LOG_ERR, "Can't bind to %s",
|
|
sock_addr2str(&bind_address, sock_buf));
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
|
|
conn->sock = sockfd;
|
|
conn->readfrom = conn_recvfrom_packet;
|
|
|
|
|
|
cw_dbg(DBG_INFO, "Discovery to %s",
|
|
sock_addr2str_p(&conn->addr, sock_buf));
|
|
|
|
run_discovery(conn, dis);
|
|
|
|
conn->readfrom = NULL;
|
|
close(sockfd);
|
|
|
|
|
|
}
|
|
|
|
freeaddrinfo(res0);
|
|
|
|
return 0;
|
|
}
|