mirror of
https://github.com/kiddin9/op-packages.git
synced 2026-10-28 06:52:51 +08:00
63 lines
2.3 KiB
C
63 lines
2.3 KiB
C
/* SPDX-License-Identifier: Apache-2.0 */
|
|
#include <assert.h>
|
|
#include "../src/webprobe.c"
|
|
|
|
void rpc_luci_dhcp4_clients(void (*add)(const struct in_addr *, const unsigned char *)) {}
|
|
|
|
static struct web_client *add(const char *ip, const unsigned char *mac)
|
|
{
|
|
struct in_addr address;
|
|
struct web_client *client;
|
|
assert(inet_pton(AF_INET, ip, &address) == 1);
|
|
web_add_client(&address, mac);
|
|
return avl_find_element(&web_clients, &address, client, avl);
|
|
}
|
|
|
|
int main(void)
|
|
{
|
|
unsigned char mac[6] = {2, 1, 2, 3, 4, 5}, other[6] = {2, 6, 7, 8, 9, 10};
|
|
uloop_init();
|
|
avl_init(&web_clients, web_compare, false, NULL);
|
|
for (int i = 0; i < WEB_CONCURRENCY; i++) web_jobs[i].fd.fd = -1;
|
|
web_pump.cb = web_run;
|
|
web_inventory.cb = web_refresh;
|
|
assert(!add("127.0.0.1", mac));
|
|
assert(!add("0.0.0.0", mac));
|
|
assert(!add("224.0.0.1", mac));
|
|
struct web_client *c = add("192.0.2.1", mac);
|
|
assert(c && web_clients.count == 1 && !c->ready);
|
|
assert(add("192.0.2.1", mac) == c && web_clients.count == 1);
|
|
/* Publish only a complete round, with exact port-to-bit correspondence. */
|
|
c->scanning = true;
|
|
c->next_port = c->pending = WEB_PORT_COUNT;
|
|
for (int i = 0; i < WEB_PORT_COUNT; i++) {
|
|
web_jobs[0].client = c;
|
|
web_jobs[0].port = i;
|
|
web_finish(&web_jobs[0], i == 0 || i == 5);
|
|
assert(c->ready == (i == WEB_PORT_COUNT - 1));
|
|
}
|
|
assert(c->open_ports == 33 && c->next_scan > web_now());
|
|
/* Refresh failure must remove links which succeeded in a previous round. */
|
|
c->scanning = true; c->checking = 0; c->pending = 1;
|
|
web_jobs[0].client = c;
|
|
web_jobs[0].timeout.cb = web_timeout;
|
|
web_timeout(&web_jobs[0].timeout);
|
|
assert(c->open_ports == 0 && c->ready);
|
|
/* IP reassignment invalidates cached results and closes pending sockets. */
|
|
c->open_ports = 15;
|
|
web_jobs[0].client = c;
|
|
web_jobs[0].fd.fd = socket(AF_INET, SOCK_STREAM, 0);
|
|
int fd = web_jobs[0].fd.fd;
|
|
add("192.0.2.1", other);
|
|
assert(!c->ready && !c->open_ports && !web_jobs[0].client);
|
|
assert(close(fd) == -1 && errno == EBADF);
|
|
/* A lease disappearing removes its cached entry and pending jobs. */
|
|
web_refresh(&web_inventory);
|
|
assert(!web_clients.count);
|
|
uloop_timeout_cancel(&web_inventory);
|
|
uloop_timeout_cancel(&web_pump);
|
|
uloop_done();
|
|
puts("webprobe: bounded targets, round publication, timeout, IP reassignment and lease removal passed");
|
|
return 0;
|
|
}
|