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
.measurement_report_timeout
= 120 * 1000;
90 config
.local_sta_update
= 1 * 1000;
91 config
.max_retry_band
= 5;
92 config
.max_neighbor_reports
= 8;
93 config
.seen_policy_timeout
= 30 * 1000;
94 config
.band_steering_threshold
= 5;
95 config
.load_balancing_threshold
= 5;
96 config
.remote_update_interval
= 1000;
97 config
.initial_connect_delay
= 0;
98 config
.remote_node_timeout
= 10;
100 config
.steer_reject_timeout
= 60000;
102 config
.band_steering_interval
= 120000;
103 config
.band_steering_min_snr
= -60;
105 config
.roam_kick_delay
= 10000;
106 config
.roam_process_timeout
= 5 * 1000;
107 config
.roam_scan_tries
= 3;
108 config
.roam_scan_timeout
= 0;
109 config
.roam_scan_interval
= 10 * 1000;
110 config
.roam_trigger_interval
= 60 * 1000;
112 config
.min_snr_kick_delay
= 5 * 1000;
114 config
.load_kick_enabled
= false;
115 config
.load_kick_threshold
= 75;
116 config
.load_kick_delay
= 10 * 1000;
117 config
.load_kick_min_clients
= 10;
118 config
.load_kick_reason_code
= 5; /* WLAN_REASON_DISASSOC_AP_BUSY */
120 config
.debug_level
= MSG_FATAL
;
123 void usteer_update_time(void)
127 clock_gettime(CLOCK_MONOTONIC
, &ts
);
128 current_time
= (uint64_t) ts
.tv_sec
* 1000 + ts
.tv_nsec
/ 1000000;
131 static int usage(const char *prog
)
133 fprintf(stderr
, "Usage: %s [options]\n"
135 " -v: Increase debug level (repeat for more messages):\n"
136 " 1: info messages\n"
137 " 2: debug messages\n"
138 " 3: verbose debug messages\n"
139 " 4: include network messages\n"
140 " 5: include extra testing messages\n"
141 " -i <name>: Connect to other instances on interface <name>\n"
142 " -s: Output log messages via syslog instead of stderr\n"
143 " -D <n>: Do not daemonize, wait for <n> seconds and print\n"
144 " remote hosts and nodes\n"
150 usteer_dump_timeout(struct uloop_timeout
*t
)
152 struct usteer_remote_host
*host
;
153 struct usteer_remote_node
*rn
;
154 struct blob_buf b
= {};
158 blob_buf_init(&b
, 0);
160 c
= blobmsg_open_table(&b
, "hosts");
161 avl_for_each_element(&remote_hosts
, host
, avl
)
162 usteer_dump_host(&b
, host
);
163 blobmsg_close_table(&b
, c
);
165 c
= blobmsg_open_table(&b
, "nodes");
166 for_each_remote_node(rn
)
167 usteer_dump_node(&b
, &rn
->node
);
168 blobmsg_close_table(&b
, c
);
170 str
= blobmsg_format_json(b
.head
, true);
179 int main(int argc
, char **argv
)
181 struct uloop_timeout dump_timer
;
184 usteer_init_defaults();
186 while ((ch
= getopt(argc
, argv
, "D:i:sv")) != -1) {
189 config
.debug_level
++;
192 config
.syslog
= true;
195 usteer_interface_add(optarg
);
198 dump_time
= atoi(optarg
);
201 return usage(argv
[0]);
205 openlog("usteer", 0, LOG_USER
);
207 config_set_event_log_types(NULL
);
208 usteer_update_time();
211 ubus_ctx
= ubus_connect(NULL
);
213 fprintf(stderr
, "Failed to connect to ubus\n");
217 ubus_add_uloop(ubus_ctx
);
219 dump_timer
.cb
= usteer_dump_timeout
;
220 uloop_timeout_set(&dump_timer
, dump_time
* 1000);
222 usteer_ubus_init(ubus_ctx
);
223 usteer_local_nodes_init(ubus_ctx
);