1 files changed, 37 insertions, 2 deletions

diff --git a/sys/netgraph/ng_source.c b/sys/netgraph/ng_source.c
index 1526993..25f3aa0 100644
--- a/sys/netgraph/ng_source.c
+++ b/sys/netgraph/ng_source.c
@@ -189,6 +189,13 @@ static const struct ng_cmdlist ng_source_cmds[] = {
 	  NULL,
 	  NULL
 	},
+	{
+	  NGM_SOURCE_COOKIE,
+	  NGM_SOURCE_START_NOW,
+	  "start_now",
+	  &ng_parse_uint64_type,
+	  NULL
+	},
 	{ 0 }
 };
 
@@ -317,6 +324,30 @@ ng_source_rcvmsg(node_p node, item_p item, hook_p lasthook)
 			ng_source_start(sc);
 		    }
 		    break;
+		case NGM_SOURCE_START_NOW:
+		    {
+			u_int64_t packets = *(u_int64_t *)msg->data;
+			if (sc->output.hook == NULL) {
+				printf("%s: start on node with no output hook\n"
+				    , __FUNCTION__);
+				error = EINVAL;
+				break;
+			}
+			if (sc->node->nd_flags & NG_SOURCE_ACTIVE) {
+				error = EBUSY;
+				break;
+			}
+			/* TODO validation of packets */
+			sc->packets = packets;
+		        sc->output_ifp = NULL;
+
+			sc->node->nd_flags |= NG_SOURCE_ACTIVE;
+			timevalclear(&sc->stats.elapsedTime);
+			timevalclear(&sc->stats.endTime);
+			getmicrotime(&sc->stats.startTime);
+			sc->intr_ch = timeout(ng_source_intr, sc, 0);
+		    }
+		    break;
 		case NGM_SOURCE_STOP:
 			ng_source_stop(sc);
 			break;
@@ -600,8 +631,12 @@ ng_source_intr (void *arg)
 		return;
 	}
 
-	ifq = &sc->output_ifp->if_snd;
-	packets = ifq->ifq_maxlen - ifq->ifq_len;
+	if (sc->output_ifp != NULL) {
+		ifq = &sc->output_ifp->if_snd;
+		packets = ifq->ifq_maxlen - ifq->ifq_len;
+	} else
+		packets = sc->snd_queue.ifq_len;
+
 	ng_source_send(sc, packets, NULL);
 	if (sc->packets == 0) {
 		int s = splnet();