ede2b75f2a
FossilOrigin-Name: 4879b9ac8d498f435164e1ff2bc6edf1f9ee547178701738e7e91294fe6a4569
478 lines
9.6 KiB
C
478 lines
9.6 KiB
C
/*
|
|
This file is part of actube.
|
|
|
|
actube 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 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
libcapwap 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 Foobar. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <stdint.h>
|
|
//#include <netinet/in.h>
|
|
|
|
#include "actube.h"
|
|
#include "wtplist.h"
|
|
#include "cw/dtls.h"
|
|
#include "cw/log.h"
|
|
#include "cw/dbg.h"
|
|
#include "cw/sock.h"
|
|
#include "conf.h"
|
|
|
|
#include "socklist.h"
|
|
|
|
#include "db.h"
|
|
#include "cw/capwap_items.h"
|
|
#include "cw/capwap_cisco.h"
|
|
//#include "cw/capwap_80211.h"
|
|
#include "cw/action.h"
|
|
#include "cw/capwap_crypto.h"
|
|
#include "cw/mbag.h"
|
|
|
|
|
|
#include "ac.h"
|
|
#include "cw/format.h"
|
|
|
|
|
|
#include "dataman.h"
|
|
|
|
|
|
int ac_run();
|
|
|
|
|
|
static void *alive_thread(void *data)
|
|
{
|
|
/* Ping the database every 5 seconds */
|
|
while (1) {
|
|
sleep(5);
|
|
db_ping();
|
|
}
|
|
}
|
|
|
|
|
|
#include <getopt.h>
|
|
static int parse_args(int argc, char *argv[])
|
|
{
|
|
// int getopt_ret;
|
|
int option_index;
|
|
|
|
static struct option long_options[] = {
|
|
{"version", optional_argument, NULL, 'v'},
|
|
{0, 0, 0, 0}
|
|
};
|
|
int o;
|
|
while ((o = getopt_long(argc, argv, "v", long_options, &option_index)) != -1) {
|
|
switch (o) {
|
|
case 0:
|
|
break;
|
|
case 'v':
|
|
printf("AC-Tube 0.01, %s\n", SYS_ARCH);
|
|
exit(0);
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
#include "cw/mod.h"
|
|
extern struct mod_ac * cw_get_mod_ac(const char *name);
|
|
|
|
#include "cw/mlist.h"
|
|
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
|
|
int rc = 0;
|
|
|
|
/* parse arguments */
|
|
parse_args(argc, argv);
|
|
|
|
cw_log_name = "AC-Tube";
|
|
|
|
if (!read_config("ac.conf"))
|
|
return 1;
|
|
|
|
|
|
/* Show debug options if there are any set */
|
|
if (cw_dbg_opt_level)
|
|
cw_log(LOG_INFO, "Debug Options: %08X", cw_dbg_opt_level);
|
|
|
|
/* XXX Hard coded debug settigns, set it by config in the future */
|
|
cw_dbg_opt_display = DBG_DISP_ASC_DMP | DBG_DISP_COLORS;
|
|
|
|
/* Warn, if the "secret" debugging feature for
|
|
developers is turned on ;) */
|
|
DBGX("Attention! %s", "DBG X is ON!");
|
|
|
|
|
|
/* Initialize the database */
|
|
if (!db_init())
|
|
goto errX;
|
|
|
|
/* Start the database */
|
|
if (!db_start())
|
|
goto errX;
|
|
|
|
db_ping();
|
|
|
|
|
|
/* Start a database "pinger thread", which inserts
|
|
every xx seconds a timestamp into the DB */
|
|
pthread_t alth;
|
|
pthread_create(&alth, NULL, alive_thread, NULL);
|
|
|
|
/* Init DTLS library */
|
|
dtls_init();
|
|
|
|
|
|
|
|
|
|
ac_global_init();
|
|
if (!socklist_init())
|
|
goto errX;
|
|
|
|
if (!wtplist_init())
|
|
goto errX;
|
|
|
|
if (!dataman_list_init())
|
|
goto errX;
|
|
|
|
|
|
cw_log(LOG_INFO, "Starting AC-Tube, Name=%s, ID=%s", conf_acname, conf_acid);
|
|
rc = ac_run();
|
|
errX:
|
|
/* XXX There is more cleanup to do */
|
|
wtplist_destroy();
|
|
socklist_destroy();
|
|
return rc;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void process_ctrl_packet(int index, struct sockaddr *addr, uint8_t * buffer, int len);
|
|
|
|
void process_cw_data_packet(int index, struct sockaddr *addr, uint8_t * buffer, int len);
|
|
|
|
|
|
int ac_run()
|
|
{
|
|
|
|
if (!conf_listen_addrs_len) {
|
|
cw_log(LOG_ERR, "Fatal error: No listen addresses found.");
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
/* it is important to create the unicast sockets first,
|
|
* because when we create the mcast an bcast sockets next
|
|
* we will look for already created sockets to find a
|
|
* good unicast reply socket */
|
|
|
|
int i;
|
|
for (i = 0; i < conf_listen_addrs_len; i++) {
|
|
char addr[100];
|
|
char port[50];
|
|
int proto;
|
|
|
|
conf_parse_listen_addr(conf_listen_addrs[i], addr, port, &proto);
|
|
|
|
|
|
//struct sockaddr sa;
|
|
|
|
|
|
socklist_add_unicast(addr, port,proto);
|
|
|
|
//#ifdef WITH_LWAPP
|
|
// if (conf_lwapp)
|
|
// socklist_add_unicast(conf_listen_addrs[i], conf_lw_control_port,
|
|
// AC_PROTO_LWAPP);
|
|
//#endif
|
|
}
|
|
|
|
|
|
|
|
if (socklist_len == 0) {
|
|
cw_log(LOG_ERR, "Fatal error: Could not setup any listen socket");
|
|
return 1;
|
|
}
|
|
|
|
/* create multicast sockets */
|
|
for (i = 0; i < conf_mcast_groups_len; i++) {
|
|
|
|
socklist_add_multicast(conf_mcast_groups[i], conf_control_port,
|
|
AC_PROTO_CAPWAP);
|
|
#ifdef WITH_LWAPP
|
|
if (conf_lwapp)
|
|
socklist_add_multicast(conf_mcast_groups[i], conf_lw_control_port,
|
|
AC_PROTO_LWAPP);
|
|
#endif
|
|
|
|
}
|
|
|
|
/* broadcast socket ipv4 only */
|
|
for (i = 0; i < conf_bcast_addrs_len; i++) {
|
|
|
|
char addr[50],port[50];
|
|
int proto;
|
|
conf_parse_listen_addr(conf_bcast_addrs[i], addr, port, &proto);
|
|
|
|
socklist_add_broadcast(addr,port,proto);
|
|
#ifdef WITH_LWAPP
|
|
// printf("Adding %d\n",socklist_len);
|
|
if (conf_lwapp)
|
|
socklist_add_broadcast(conf_bcast_addrs[i], conf_lw_control_port,
|
|
AC_PROTO_LWAPP);
|
|
// printf ("SI %d, PROTO: %d\n",socklist_len-1,socklist[socklist_len-1].ac_proto);
|
|
#endif
|
|
}
|
|
|
|
|
|
|
|
//get_acinfo();
|
|
|
|
|
|
while (1) {
|
|
|
|
/* prepare fdset */
|
|
fd_set fset;
|
|
int max = 0;
|
|
FD_ZERO(&fset);
|
|
for (i = 0; i < socklist_len; i++) {
|
|
FD_SET(socklist[i].sockfd, &fset);
|
|
if (socklist[i].sockfd > max)
|
|
max = socklist[i].sockfd;
|
|
if (socklist[i].data_sockfd) {
|
|
FD_SET(socklist[i].data_sockfd,&fset);
|
|
if (socklist[i].sockfd > max)
|
|
max = socklist[i].sockfd;
|
|
}
|
|
}
|
|
|
|
/* wait for an event */
|
|
int n;
|
|
while ((n = select(max + 1, &fset, NULL, NULL, NULL)) < 0) {
|
|
if (errno != EINTR)
|
|
return n;
|
|
|
|
}
|
|
|
|
/* process the received packet */
|
|
for (i = 0; i < socklist_len; i++) {
|
|
uint8_t buffer[4096];
|
|
struct sockaddr_storage srcaddr;
|
|
socklen_t srcaddrlen;
|
|
|
|
|
|
if (FD_ISSET(socklist[i].data_sockfd, &fset)){
|
|
|
|
int len = sock_receive(socklist[i].data_sockfd,
|
|
buffer, sizeof(buffer),
|
|
0,
|
|
(struct sockaddr *) &srcaddr,
|
|
&srcaddrlen);
|
|
|
|
process_cw_data_packet(i, (struct sockaddr *) &srcaddr, buffer, len);
|
|
|
|
}
|
|
|
|
|
|
if (FD_ISSET(socklist[i].sockfd, &fset)){
|
|
|
|
int len = sock_receive(socklist[i].sockfd,
|
|
buffer, sizeof(buffer),
|
|
0,
|
|
(struct sockaddr *) &srcaddr,
|
|
&srcaddrlen);
|
|
|
|
process_ctrl_packet(i, (struct sockaddr *) &srcaddr, buffer, len);
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
void process_cw_data_packet(int index, struct sockaddr *addr, uint8_t * buffer, int len)
|
|
{
|
|
cw_dbg(DBG_X, "There is a data packet now");
|
|
|
|
dataman_list_lock();
|
|
cw_dbg(DBG_X, "Dataman list locked, now getting");
|
|
struct dataman * dm = dataman_list_get(socklist[index].data_sockfd,addr);
|
|
cw_dbg(DBG_X, "Dataman list locked, now gotted");
|
|
|
|
cw_dbg(DBG_INFO,"Packet for dataman %s,%d",sock_addr2str_p(addr),socklist[index].data_sockfd);
|
|
if (!dm) {
|
|
cw_dbg(DBG_INFO,"No dataman %s,%d",sock_addr2str_p(addr),socklist[index].data_sockfd);
|
|
dm = dataman_create(socklist[index].data_sockfd,addr);
|
|
if (!dm){
|
|
cw_log(LOG_ERR,"Can't create dataman for packet from %s",sock_addr2str_p(addr));
|
|
return;
|
|
}
|
|
dataman_list_add(dm);
|
|
|
|
dataman_start(dm);
|
|
|
|
|
|
}
|
|
dataman_list_unlock();
|
|
|
|
dataman_add_packet(dm,buffer,len);
|
|
|
|
return;
|
|
|
|
exit(0);
|
|
|
|
|
|
|
|
|
|
printf("Data packet received len = %d\n",len);
|
|
exit(0);
|
|
struct wtpman *wtpman = wtplist_get(addr);
|
|
if (!wtpman){
|
|
cw_dbg(DBG_PKT_ERR,"Discarding packet on data channel from %s - No wtpman found.",sock_addr2str(addr));
|
|
return;
|
|
}
|
|
|
|
|
|
wtpman_addpacket(wtpman, buffer, len);
|
|
}
|
|
|
|
|
|
void process_cw_ctrl_packet(int index, struct sockaddr *addr, uint8_t * buffer, int len)
|
|
{
|
|
|
|
|
|
/* first of all check preamble */
|
|
int preamble = cw_get_hdr_preamble(buffer);
|
|
|
|
if (preamble != CAPWAP_PACKET_PREAMBLE && preamble != CAPWAP_DTLS_PACKET_PREAMBLE) {
|
|
cw_dbg(DBG_PKT_ERR,
|
|
"Discarding packet from %s, wrong preamble, preamble = 0x%01X",
|
|
sock_addr2str(addr), preamble);
|
|
return;
|
|
}
|
|
|
|
|
|
wtplist_lock();
|
|
struct wtpman *wtpman = wtplist_get(addr);
|
|
if (!wtpman) {
|
|
|
|
wtpman = wtpman_create(index, addr);
|
|
|
|
|
|
if (!wtpman) {
|
|
cw_log(LOG_ERR, "Error creating wtpman: %s", strerror(errno));
|
|
wtplist_unlock();
|
|
return;
|
|
}
|
|
|
|
|
|
if (!wtplist_add(wtpman)) {
|
|
cw_log(LOG_ERR, "Error adding wtpman: Too many wtp connections");
|
|
wtpman_destroy(wtpman);
|
|
wtplist_unlock();
|
|
return;
|
|
};
|
|
|
|
wtpman_start(wtpman, preamble & 0xf);
|
|
}
|
|
|
|
|
|
wtpman_addpacket(wtpman, buffer, len);
|
|
wtplist_unlock();
|
|
}
|
|
|
|
|
|
|
|
void process_lw_ctrl_packet(int index, struct sockaddr *addr, uint8_t * buffer, int len)
|
|
{
|
|
//int sock = socklist[index].reply_sockfd;
|
|
|
|
uint8_t *m = buffer + 6;
|
|
uint32_t val = ntohl(*((uint32_t *) (m)));
|
|
|
|
|
|
printf("VAL: %08X\n", val);
|
|
|
|
|
|
|
|
/* first of all check preamble */
|
|
int version = LWTH_GET_VERSION(m);
|
|
|
|
if (version != LW_VERSION) {
|
|
// cw_log_debug1("Discarding LWAPP packet, wrong verson");
|
|
return;
|
|
}
|
|
|
|
int l = LWTH_GET_LENGTH(m);
|
|
printf("LEN = %d\n", l);
|
|
|
|
if (l + 12 != len) {
|
|
// cw_log_debug1("Discarding LWAPP packet, wrong length");
|
|
return;
|
|
}
|
|
|
|
wtplist_lock();
|
|
struct wtpman *wtpman = wtplist_get(addr);
|
|
if (!wtpman) {
|
|
|
|
wtpman = wtpman_create(index, addr);
|
|
|
|
if (!wtpman) {
|
|
cw_log(LOG_ERR, "Error creating wtpman: %s", strerror(errno));
|
|
wtplist_unlock();
|
|
return;
|
|
}
|
|
|
|
|
|
if (!wtplist_add(wtpman)) {
|
|
cw_log(LOG_ERR, "Error adding wtpman: Too many wtp connections");
|
|
wtpman_destroy(wtpman);
|
|
wtplist_unlock();
|
|
return;
|
|
};
|
|
|
|
//wtpman_lw_start(wtpman);
|
|
}
|
|
//wtpman_lw_addpacket(wtpman,buffer,len);
|
|
wtplist_unlock();
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void process_ctrl_packet(int index, struct sockaddr *addr, uint8_t * buffer, int len)
|
|
{
|
|
switch (socklist[index].ac_proto) {
|
|
case AC_PROTO_CAPWAP:
|
|
process_cw_ctrl_packet(index, addr, buffer, len);
|
|
return;
|
|
case AC_PROTO_LWAPP:
|
|
process_lw_ctrl_packet(index, addr, buffer, len);
|
|
return;
|
|
}
|
|
}
|