ipoe: implemented offer delaying

5 files changed, 263 insertions, 28 deletions

diff --git a/accel-pppd/accel-ppp.conf b/accel-pppd/accel-ppp.conf
index ca90ea2..127d148 100644
--- a/accel-pppd/accel-ppp.conf
+++ b/accel-pppd/accel-ppp.conf
@@ -55,6 +55,7 @@ start=dhcpv4
 #attr-l4-redirect=L4-Redirect
 #local-net=192.168.0.0/16
 #lua-file=/etc/accel-ppp.lua
+#offer-delay=0,100:100,200:200,-1:1000
 interface=eth0
 
 
diff --git a/accel-pppd/accel-ppp.conf.5 b/accel-pppd/accel-ppp.conf.5
index 607357f..a79a8d6 100644
--- a/accel-pppd/accel-ppp.conf.5
+++ b/accel-pppd/accel-ppp.conf.5
@@ -313,6 +313,11 @@ Specifies networks from which packets will be treated as unclassified. You may s
 .TP
 .BI "proto=" n
 Specifies number of protocol to be used for inserted routes.
+.TP
+.BI "offer-delay=" delay[,delay1:count1[,delay2:count2[,...]]]
+Specifies delays (also in condition of connection count) to send DHCPOFFER (ms).
+Last delay in list may be -1 which means don't accept new connections.
+List must to be sorted by count key.
 .SH [dns]
 .TP
 .BI "dns1=" x.x.x.x
diff --git a/accel-pppd/ctrl/ipoe/dhcpv4.h b/accel-pppd/ctrl/ipoe/dhcpv4.h
index b7602e1..ea39657 100644
--- a/accel-pppd/ctrl/ipoe/dhcpv4.h
+++ b/accel-pppd/ctrl/ipoe/dhcpv4.h
@@ -25,8 +25,7 @@
 #define DHCPRELEASE  7
 #define DHCPINFORM   8
 
-struct dhcpv4_hdr
-{
+struct dhcpv4_hdr {
 	uint8_t op;
 	uint8_t htype;
 	uint8_t hlen;
@@ -44,16 +43,14 @@ struct dhcpv4_hdr
 	uint8_t magic[4];
 } __packed;
 
-struct dhcpv4_option
-{
+struct dhcpv4_option {
 	struct list_head entry;
 	uint8_t type;
 	uint8_t len;
 	uint8_t *data;
 };
 
-struct dhcpv4_packet
-{
+struct dhcpv4_packet {
 	struct dhcpv4_hdr *hdr;
 	struct list_head options;
 	struct dhcpv4_option *client_id;
@@ -66,8 +63,7 @@ struct dhcpv4_packet
 	uint8_t data[0];
 };
 
-struct dhcpv4_iprange
-{
+struct dhcpv4_iprange {
 	struct list_head entry;
 	uint32_t routerip;
 	uint32_t startip;
@@ -78,8 +74,7 @@ struct dhcpv4_iprange
 	unsigned long free[0];
 }; 
 
-struct dhcpv4_serv
-{
+struct dhcpv4_serv {
 	struct triton_context_t *ctx;
 	struct triton_md_handler_t hnd;
 	int raw_sock;
@@ -88,8 +83,7 @@ struct dhcpv4_serv
 	struct dhcpv4_iprange *range;
 };
 
-struct dhcpv4_relay
-{
+struct dhcpv4_relay {
 	struct list_head entry;
 	struct triton_context_t ctx;
 	struct triton_md_handler_t hnd;
diff --git a/accel-pppd/ctrl/ipoe/ipoe.c b/accel-pppd/ctrl/ipoe/ipoe.c
index 94f50fd..f4701d1 100644
--- a/accel-pppd/ctrl/ipoe/ipoe.c
+++ b/accel-pppd/ctrl/ipoe/ipoe.c
@@ -79,11 +79,14 @@ static int conf_lease_timeout = 660;
 static int conf_verbose;
 static const char *conf_agent_remote_id;
 static int conf_proto;
+static LIST_HEAD(conf_offer_delay);
 
 static unsigned int stat_starting;
 static unsigned int stat_active;
+static unsigned int stat_delayed_offer;
 
 static mempool_t ses_pool;
+static mempool_t disc_item_pool;
 
 static LIST_HEAD(serv_list);
 
@@ -117,6 +120,19 @@ struct gw_addr {
 	int mask1;
 };
 
+struct disc_item {
+	struct list_head entry;
+	struct dhcpv4_packet *pack;
+	struct timespec ts;
+};
+
+struct delay {
+	struct list_head entry;
+	unsigned int conn_cnt;
+	int delay;
+};
+
+
 static pthread_mutex_t uc_lock = PTHREAD_MUTEX_INITIALIZER;
 static LIST_HEAD(uc_list);
 static int uc_size;
@@ -132,6 +148,8 @@ static void ipoe_drop_sessions(struct ipoe_serv *serv, struct ipoe_session *skip
 static void ipoe_serv_close(struct triton_context_t *ctx);
 static void __ipoe_session_activate(struct ipoe_session *ses);
 static void ipoe_ses_recv_dhcpv4(struct dhcpv4_serv *dhcpv4, struct dhcpv4_packet *pack);
+static void __ipoe_recv_dhcpv4(struct dhcpv4_serv *dhcpv4, struct dhcpv4_packet *pack, int force);
+static int get_offer_delay();
 
 static struct ipoe_session *ipoe_session_lookup(struct ipoe_serv *serv, struct dhcpv4_packet *pack, struct ipoe_session **opt82_ses)
 {
@@ -183,7 +201,7 @@ static struct ipoe_session *ipoe_session_lookup(struct ipoe_serv *serv, struct d
 		if (opt82_match && opt82_ses)
 			*opt82_ses = ses;
 			
-		if (memcmp(pack->hdr->chaddr, ses->hwaddr, 6))
+		if (memcmp(pack->hdr->chaddr, ses->hwaddr, ETH_ALEN))
 			continue;
 	
 		res = ses;
@@ -385,6 +403,8 @@ static void ipoe_session_start(struct ipoe_session *ses)
 	struct ifreq ifr;
 	struct unit_cache *uc;
 
+	__sync_add_and_fetch(&stat_starting, 1);
+	
 	if (!ses->ses.username) {
 		strncpy(ses->ses.ifname, ses->serv->ifname, AP_IFNAME_LEN);
 		
@@ -692,6 +712,10 @@ static void __ipoe_session_activate(struct ipoe_session *ses)
 	if (ses->l4_redirect)
 		ipoe_change_l4_redirect(ses, 0);
 
+	__sync_sub_and_fetch(&stat_starting, 1);
+	__sync_add_and_fetch(&stat_active, 1);
+	ses->started = 1;
+
 	ap_session_activate(&ses->ses);
 
 	if (ses->dhcpv4_request) {
@@ -786,6 +810,11 @@ static void ipoe_session_free(struct ipoe_session *ses)
 {
 	struct unit_cache *uc;
 
+	if (ses->started)
+		__sync_sub_and_fetch(&stat_active, 1);
+	else
+		__sync_sub_and_fetch(&stat_starting, 1);
+	
 	if (ses->timer.tpd)
 		triton_timer_del(&ses->timer);
 
@@ -1056,10 +1085,88 @@ static void ipoe_ses_recv_dhcpv4(struct dhcpv4_serv *dhcpv4, struct dhcpv4_packe
 	}
 }
 
-static void ipoe_recv_dhcpv4(struct dhcpv4_serv *dhcpv4, struct dhcpv4_packet *pack)
+static void ipoe_serv_disc_timer(struct triton_timer_t *t)
+{
+	struct ipoe_serv *serv = container_of(t, typeof(*serv), disc_timer);
+	struct disc_item *d;
+	struct timespec ts;
+	int delay, offer_delay;
+
+	clock_gettime(CLOCK_MONOTONIC, &ts);
+	
+	while (!list_empty(&serv->disc_list)) {	
+	  d = list_entry(serv->disc_list.next, typeof(*d), entry);
+			
+		delay = (ts.tv_sec - d->ts.tv_sec) * 1000 + (ts.tv_nsec - d->ts.tv_nsec) / 1000000;
+		offer_delay = get_offer_delay();
+
+		if (delay < offer_delay - 1) {
+			delay = offer_delay - delay;
+			t->expire_tv.tv_sec = delay / 1000;
+			t->expire_tv.tv_usec = (delay % 1000) * 1000;
+			triton_timer_mod(t, 0);
+			return;
+		}
+	
+		__ipoe_recv_dhcpv4(serv->dhcpv4, d->pack, 1);
+
+		list_del(&d->entry);
+		dhcpv4_packet_free(d->pack);
+		mempool_free(d);
+
+		__sync_sub_and_fetch(&stat_delayed_offer, 1);
+	}
+
+	triton_timer_del(t);
+}
+
+static void ipoe_serv_add_disc(struct ipoe_serv *serv, struct dhcpv4_packet *pack, int offer_delay)
+{
+	struct disc_item *d = mempool_alloc(disc_item_pool);
+
+	if (!d)
+		return;
+		
+	__sync_add_and_fetch(&stat_delayed_offer, 1);
+	
+	dhcpv4_packet_ref(pack);
+	d->pack = pack;
+	clock_gettime(CLOCK_MONOTONIC, &d->ts);
+	list_add_tail(&d->entry, &serv->disc_list);
+		
+	if (!serv->disc_timer.tpd) {
+		serv->disc_timer.expire_tv.tv_sec = offer_delay / 1000;
+		serv->disc_timer.expire_tv.tv_usec = (offer_delay % 1000) * 1000;
+		triton_timer_add(&serv->ctx, &serv->disc_timer, 0);
+	}
+}
+
+static void ipoe_serv_check_disc(struct ipoe_serv *serv, struct dhcpv4_packet *pack)
+{
+	struct disc_item *d;
+
+	list_for_each_entry(d, &serv->disc_list, entry) {
+		if (d->pack->hdr->xid != pack->hdr->xid)
+			continue;
+
+		if (memcmp(d->pack->hdr->chaddr, pack->hdr->chaddr, ETH_ALEN))
+			continue;
+		
+		list_del(&d->entry);
+		dhcpv4_packet_free(d->pack);
+		mempool_free(d);
+		
+		__sync_sub_and_fetch(&stat_delayed_offer, 1);
+		
+		break;
+	}
+}
+
+static void __ipoe_recv_dhcpv4(struct dhcpv4_serv *dhcpv4, struct dhcpv4_packet *pack, int force)
 {
 	struct ipoe_serv *serv = container_of(dhcpv4->ctx, typeof(*serv), ctx);
 	struct ipoe_session *ses, *opt82_ses;
+	int offer_delay;
 	//struct dhcpv4_packet *reply;
 
 	if (ap_shutdown)
@@ -1078,6 +1185,15 @@ static void ipoe_recv_dhcpv4(struct dhcpv4_serv *dhcpv4, struct dhcpv4_packet *p
 				triton_context_call(&opt82_ses->ctx, (triton_event_func)__ipoe_session_terminate, &opt82_ses->ses);
 			}
 
+			offer_delay = get_offer_delay();
+			if (offer_delay == -1)
+				goto out;
+
+			if (offer_delay && !force) {
+				ipoe_serv_add_disc(serv, pack, offer_delay);
+				goto out;
+			}
+
 			ses = ipoe_session_create_dhcpv4(serv, pack);
 			if (ses) {
 				dhcpv4_packet_ref(pack);
@@ -1117,17 +1233,20 @@ static void ipoe_recv_dhcpv4(struct dhcpv4_serv *dhcpv4, struct dhcpv4_packet *p
 			}
 		}
 	} else if (pack->msg_type == DHCPREQUEST) {
+		ipoe_serv_check_disc(serv, pack);
+		
 		ses = ipoe_session_lookup(serv, pack, &opt82_ses);
 
 		if (!ses) {
 			if (conf_verbose) {
-				log_info2("recv ");
-				dhcpv4_print_packet(pack, 0, log_info2);
+				log_debug("recv ");
+				dhcpv4_print_packet(pack, 0, log_debug);
 			}
-				
-			if (serv->opt_shared == 0)
+
+			if (serv->opt_shared == 0) {
 				ipoe_drop_sessions(serv, NULL);
-			else if (opt82_ses) {
+				//dhcpv4_send_nak(dhcpv4, pack);
+			} else if (opt82_ses) {
 				if (conf_verbose) {
 					log_switch(dhcpv4->ctx, &opt82_ses->ses);
 					log_ppp_warn("mac change detected\n");
@@ -1135,11 +1254,10 @@ static void ipoe_recv_dhcpv4(struct dhcpv4_serv *dhcpv4, struct dhcpv4_packet *p
 				
 				triton_context_call(&opt82_ses->ctx, (triton_event_func)__ipoe_session_terminate, &opt82_ses->ses);
 			}
-
-			dhcpv4_send_nak(dhcpv4, pack);
 		} else {
 			if (pack->hdr->ciaddr == ses->yiaddr && pack->hdr->xid != ses->xid)
 				ses->xid = pack->hdr->xid;
+			
 			if ((pack->server_id && (pack->server_id != ses->siaddr || pack->request_ip != ses->yiaddr)) ||
 				(pack->hdr->ciaddr && (pack->hdr->xid != ses->xid || pack->hdr->ciaddr != ses->yiaddr)) ||
 				(opt82_ses && ses != opt82_ses) || (!opt82_ses && pack->relay_agent)) {
@@ -1154,7 +1272,8 @@ static void ipoe_recv_dhcpv4(struct dhcpv4_serv *dhcpv4, struct dhcpv4_packet *p
 
 				if (pack->server_id == ses->siaddr)
 					dhcpv4_send_nak(dhcpv4, pack);
-				else if (ses->serv->dhcpv4_relay)
+
+				if (ses->serv->dhcpv4_relay)
 					dhcpv4_relay_send(ses->serv->dhcpv4_relay, pack, 0, ses->serv->ifname, conf_agent_remote_id);
 				
 				triton_context_call(&ses->ctx, (triton_event_func)__ipoe_session_terminate, &ses->ses);
@@ -1190,6 +1309,11 @@ out:
 	pthread_mutex_unlock(&serv->lock);
 }
 
+static void ipoe_recv_dhcpv4(struct dhcpv4_serv *dhcpv4, struct dhcpv4_packet *pack)
+{
+	__ipoe_recv_dhcpv4(dhcpv4, pack, 0);
+}
+
 static int parse_dhcpv4_mask(uint32_t mask)
 {
 	int i;
@@ -1489,6 +1613,17 @@ static void ipoe_serv_close(struct triton_context_t *ctx)
 	if (serv->arp)
 		arpd_stop(serv->arp);
 
+	while (!list_empty(&serv->disc_list)) {
+		struct disc_item *d = list_entry(serv->disc_list.next, typeof(*d), entry);
+		list_del(&d->entry);
+		dhcpv4_packet_free(d->pack);
+		mempool_free(d);
+		__sync_sub_and_fetch(&stat_delayed_offer, 1);
+	}
+
+	if (serv->disc_timer.tpd)
+		triton_timer_del(&serv->disc_timer);
+
 	triton_context_unregister(ctx);
 
 	_free(serv->ifname);
@@ -1520,6 +1655,7 @@ static int show_stat_exec(const char *cmd, char * const *fields, int fields_cnt,
 	cli_send(client, "ipoe:\r\n");
 	cli_sendv(client,"  starting: %u\r\n", stat_starting);
 	cli_sendv(client,"  active: %u\r\n", stat_active);
+	cli_sendv(client,"  delayed: %u\r\n", stat_delayed_offer);
 
 	return CLI_CMD_OK;
 }
@@ -1568,6 +1704,24 @@ struct ipoe_serv *ipoe_find_serv(const char *ifname)
 	return NULL;
 }
 
+static int get_offer_delay()
+{
+	struct delay *r, *prev = NULL;
+
+	list_for_each_entry(r, &conf_offer_delay, entry) {
+		if (!prev || stat_active >= r->conn_cnt) {
+			prev = r;
+			continue;
+		}
+		break;
+	}
+
+	if (prev)
+		return prev->delay;
+	
+	return 0;
+}
+
 static void add_interface(const char *ifname, int ifindex, const char *opt)
 {
 	char *str0 = NULL, *str, *ptr1, *ptr2;
@@ -1735,7 +1889,9 @@ static void add_interface(const char *ifname, int ifindex, const char *opt)
 	serv->active = 1;
 	INIT_LIST_HEAD(&serv->sessions);
 	INIT_LIST_HEAD(&serv->addr_list);
+	INIT_LIST_HEAD(&serv->disc_list);
 	memcpy(serv->hwaddr, ifr.ifr_hwaddr.sa_data, ETH_ALEN);
+	serv->disc_timer.expire = ipoe_serv_disc_timer;
 	pthread_mutex_init(&serv->lock, NULL);
 
 	triton_context_register(&serv->ctx, NULL);
@@ -1994,6 +2150,81 @@ static void load_radius_attrs(void)
 }
 #endif
 
+static void strip(char *str)
+{
+	char *ptr = str;
+	char *endptr = strchr(str, 0);
+	while (1) {
+		ptr = strchr(ptr, ' ');
+		if (ptr)
+			memmove(ptr, ptr + 1, endptr - ptr - 1);
+		else
+			break;
+	}
+}
+
+int parse_offer_delay(const char *str)
+{
+	char *str1;
+	char *ptr1, *ptr2, *ptr3, *endptr;
+	struct delay *r;
+
+	while (!list_empty(&conf_offer_delay)) {
+		r = list_entry(conf_offer_delay.next, typeof(*r), entry);
+		list_del(&r->entry);
+		_free(r);
+	}
+
+	if (!str)
+		return 0;
+
+	str1 = _strdup(str);
+	strip(str1);
+
+	ptr1 = str1;
+
+	while (1) {
+		ptr2 = strchr(ptr1, ',');
+		if (ptr2)
+			*ptr2 = 0;
+		ptr3 = strchr(ptr1, ':');
+		if (ptr3)
+			*ptr3 = 0;
+
+		r = _malloc(sizeof(*r));
+		memset(r, 0, sizeof(*r));
+
+		r->delay = strtol(ptr1, &endptr, 10);
+		if (*endptr)
+			goto out_err;
+
+		if (list_empty(&conf_offer_delay))
+			r->conn_cnt = 0;
+		else {
+			if (!ptr3)
+				goto out_err;
+			r->conn_cnt = strtol(ptr3 + 1, &endptr, 10);
+			if (*endptr)
+				goto out_err;
+		}
+
+		list_add_tail(&r->entry, &conf_offer_delay);
+
+		if (!ptr2)
+			break;
+
+		ptr1 = ptr2 + 1;
+	}
+
+	_free(str1);
+	return 0;
+
+out_err:
+	_free(str1);
+	log_error("ipoe: failed to parse offer-delay\n");
+	return -1;
+}
+
 static void load_config(void)
 {
 	const char *opt;
@@ -2150,6 +2381,8 @@ static void load_config(void)
 	if (triton_module_loaded("radius"))
 		load_radius_attrs();
 #endif
+
+	parse_offer_delay(conf_get_opt("ipoe", "offer-delay"));
 	
 	load_interfaces(s);
 	load_local_nets(s);
@@ -2167,6 +2400,7 @@ static struct triton_timer_t l4_redirect_timer = {
 static void ipoe_init(void)
 {
 	ses_pool = mempool_create(sizeof(struct ipoe_session));
+	disc_item_pool = mempool_create(sizeof(struct disc_item));
 	uc_pool = mempool_create(sizeof(struct unit_cache));
 
 	triton_context_register(&l4_redirect_ctx, NULL);
diff --git a/accel-pppd/ctrl/ipoe/ipoe.h b/accel-pppd/ctrl/ipoe/ipoe.h
index 7385598..7806eb6 100644
--- a/accel-pppd/ctrl/ipoe/ipoe.h
+++ b/accel-pppd/ctrl/ipoe/ipoe.h
@@ -15,8 +15,7 @@
 
 struct arp_serv;
 
-struct ipoe_serv
-{
+struct ipoe_serv {
 	struct list_head entry;
 	struct triton_context_t ctx;
 	char *ifname;
@@ -27,6 +26,8 @@ struct ipoe_serv
 	struct dhcpv4_serv *dhcpv4;
 	struct dhcpv4_relay *dhcpv4_relay;
 	struct arp_serv *arp;
+	struct list_head disc_list;
+	struct triton_timer_t disc_timer;
 	pthread_mutex_t lock;
 	int opt_mode;
 	uint32_t opt_src;
@@ -40,8 +41,7 @@ struct ipoe_serv
 	int active:1;
 };
 
-struct ipoe_session
-{
+struct ipoe_session {
 	struct list_head entry;
 	struct triton_context_t ctx;
 	struct triton_timer_t timer;
@@ -68,6 +68,7 @@ struct ipoe_session
 	int ifindex;
 	struct ipv4db_item_t ipv4;
 	int ifcfg:1;
+	int started:1;
 	int terminating:1;
 	int dhcp_addr:1;
 	int relay_addr:1;
@@ -75,8 +76,7 @@ struct ipoe_session
 	int l4_redirect_set:1;
 };
 
-struct ipoe_session_info
-{
+struct ipoe_session_info {
 	struct list_head entry;
 	int ifindex;
 	uint32_t addr;
@@ -111,5 +111,6 @@ void ipoe_nl_get_sessions(struct list_head *list);
 
 struct arp_serv *arpd_start(struct ipoe_serv *ipoe);
 void arpd_stop(struct arp_serv *arp);
+
 #endif