close connections in packetgen

src/core/packetgen/connection.go

@@ -64,6 +64,7 @@ func OpenConnection(ctx context.Context, c ConnectionConfig) (Connection, error)
 
 type Connection interface {
 	Write(Packet) (int, error)
+	Close() error
 	Target() string
 }
 
@@ -103,6 +104,10 @@ func (conn *rawConn) Write(packet Packet) (n int, err error) {
 	return conn.PacketConn.WriteTo(payloadBuf.Bytes(), nil, &net.IPAddr{IP: packet.IP()})
 }
 
+func (conn *rawConn) Close() error {
+	return conn.PacketConn.Close()
+}
+
 func (conn *rawConn) Target() string { return conn.target }
 
 type netConnConfig struct {
@@ -170,4 +175,8 @@ func (conn *netConn) Write(packet Packet) (n int, err error) {
 	return conn.Conn.Write(payloadBuf.Bytes())
 }
 
+func (conn *netConn) Close() error {
+	return conn.Conn.Close()
+}
+
 func (conn *netConn) Target() string { return conn.target }

src/job/packetgen.go

@@ -74,6 +74,7 @@ func sendPacket(ctx context.Context, logger *zap.Logger, jobConfig *packetgenJob
 	if err != nil {
 		return err
 	}
+	defer conn.Close()
 
 	for jobConfig.Next(ctx) {
 		packetTpl, more := <-packetsChan