2 * This program is free software; you can redistribute it and/or modify
3 * it under the terms of the GNU General Public License as published by
4 * the Free Software Foundation; either version 2 of the License.
6 * This program is distributed in the hope that it will be useful,
7 * but WITHOUT ANY WARRANTY; without even the implied warranty of
8 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
9 * GNU General Public License for more details.
11 * You should have received a copy of the GNU General Public License
12 * along with this program; if not, write to the Free Software
13 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
15 * Copyright (C) 2020 embedd.ch
16 * Copyright (C) 2020 Felix Fietkau <nbd@nbd.name>
17 * Copyright (C) 2020 John Crispin <john@phrozen.org>
24 #include <libubox/blobmsg_json.h>
30 struct ubus_context
*ubus_ctx
;
31 struct usteer_config config
= {};
32 struct blob_attr
*host_info_blob
;
33 uint64_t current_time
;
36 LIST_HEAD(node_handlers
);
38 const char * const event_types
[__EVENT_TYPE_MAX
] = {
39 [EVENT_TYPE_PROBE
] = "probe",
40 [EVENT_TYPE_AUTH
] = "auth",
41 [EVENT_TYPE_ASSOC
] = "assoc",
44 void log_msg(char *msg
)
47 syslog(LOG_INFO
, "%s\n", msg
);
49 fprintf(stderr
, "%s\n", msg
);
52 void debug_msg(int level
, const char *func
, int line
, const char *format
, ...)
56 if (config
.debug_level
< level
)
60 fprintf(stderr
, "[%s:%d] ", func
, line
);
64 vsyslog(level
>= MSG_DEBUG
? LOG_DEBUG
: LOG_INFO
, format
, ap
);
66 vfprintf(stderr
, format
, ap
);
71 void debug_msg_cont(int level
, const char *format
, ...)
75 if (config
.debug_level
< level
)
79 vfprintf(stderr
, format
, ap
);
83 void usteer_init_defaults(void)
85 memset(&config
, 0, sizeof(config
));
87 config
.sta_block_timeout
= 30 * 1000;
88 config
.local_sta_timeout
= 120 * 1000;
89 config
.local_sta_update
= 1 * 1000;
90 config
.max_retry_band
= 5;
91 config
.max_neighbor_reports
= 8;
92 config
.seen_policy_timeout
= 30 * 1000;
93 config
.band_steering_threshold
= 5;
94 config
.load_balancing_threshold
= 5;
95 config
.remote_update_interval
= 1000;
96 config
.initial_connect_delay
= 0;
97 config
.remote_node_timeout
= 10;
99 config
.roam_kick_delay
= 100;
100 config
.roam_process_timeout
= 5 * 1000;
101 config
.roam_scan_tries
= 3;
102 config
.roam_scan_interval
= 10 * 1000;
103 config
.roam_trigger_interval
= 60 * 1000;
105 config
.load_kick_enabled
= false;
106 config
.load_kick_threshold
= 75;
107 config
.load_kick_delay
= 10 * 1000;
108 config
.load_kick_min_clients
= 10;
109 config
.load_kick_reason_code
= 5; /* WLAN_REASON_DISASSOC_AP_BUSY */
111 config
.debug_level
= MSG_FATAL
;
114 void usteer_update_time(void)
118 clock_gettime(CLOCK_MONOTONIC
, &ts
);
119 current_time
= (uint64_t) ts
.tv_sec
* 1000 + ts
.tv_nsec
/ 1000000;
122 static int usage(const char *prog
)
124 fprintf(stderr
, "Usage: %s [options]\n"
126 " -v: Increase debug level (repeat for more messages):\n"
127 " 1: info messages\n"
128 " 2: debug messages\n"
129 " 3: verbose debug messages\n"
130 " 4: include network messages\n"
131 " 5: include extra testing messages\n"
132 " -i <name>: Connect to other instances on interface <name>\n"
133 " -s: Output log messages via syslog instead of stderr\n"
134 " -D <n>: Do not daemonize, wait for <n> seconds and print\n"
135 " remote hosts and nodes\n"
141 usteer_dump_timeout(struct uloop_timeout
*t
)
143 struct usteer_remote_host
*host
;
144 struct usteer_remote_node
*rn
;
145 struct blob_buf b
= {};
149 blob_buf_init(&b
, 0);
151 c
= blobmsg_open_table(&b
, "hosts");
152 avl_for_each_element(&remote_hosts
, host
, avl
)
153 usteer_dump_host(&b
, host
);
154 blobmsg_close_table(&b
, c
);
156 c
= blobmsg_open_table(&b
, "nodes");
157 for_each_remote_node(rn
)
158 usteer_dump_node(&b
, &rn
->node
);
159 blobmsg_close_table(&b
, c
);
161 str
= blobmsg_format_json(b
.head
, true);
170 int main(int argc
, char **argv
)
172 struct uloop_timeout dump_timer
;
175 usteer_init_defaults();
177 while ((ch
= getopt(argc
, argv
, "D:i:sv")) != -1) {
180 config
.debug_level
++;
183 config
.syslog
= true;
186 usteer_interface_add(optarg
);
189 dump_time
= atoi(optarg
);
192 return usage(argv
[0]);
196 openlog("usteer", 0, LOG_USER
);
198 config_set_event_log_types(NULL
);
199 usteer_update_time();
202 ubus_ctx
= ubus_connect(NULL
);
204 fprintf(stderr
, "Failed to connect to ubus\n");
208 ubus_add_uloop(ubus_ctx
);
210 dump_timer
.cb
= usteer_dump_timeout
;
211 uloop_timeout_set(&dump_timer
, dump_time
* 1000);
213 usteer_ubus_init(ubus_ctx
);
214 usteer_local_nodes_init(ubus_ctx
);