diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/include/gpxe/netdevice.h | 5 | ||||
| -rw-r--r-- | src/net/netdevice.c | 4 |
2 files changed, 9 insertions, 0 deletions
diff --git a/src/include/gpxe/netdevice.h b/src/include/gpxe/netdevice.h index b7ec83d96..7551be74d 100644 --- a/src/include/gpxe/netdevice.h +++ b/src/include/gpxe/netdevice.h @@ -10,6 +10,7 @@ #include <stdint.h> #include <gpxe/list.h> #include <gpxe/tables.h> +#include <gpxe/hotplug.h> struct pk_buff; struct net_device; @@ -137,6 +138,10 @@ struct ll_protocol { struct net_device { /** List of network devices */ struct list_head list; + + /** List of persistent reference holders */ + struct list_head references; + /** Transmit packet * * @v netdev Network device diff --git a/src/net/netdevice.c b/src/net/netdevice.c index 3f8e54bb1..3949d26de 100644 --- a/src/net/netdevice.c +++ b/src/net/netdevice.c @@ -161,6 +161,7 @@ struct net_device * alloc_netdev ( size_t priv_size ) { netdev = calloc ( 1, sizeof ( *netdev ) + priv_size ); if ( netdev ) { + INIT_LIST_HEAD ( &netdev->references ); INIT_LIST_HEAD ( &netdev->rx_queue ); netdev->priv = ( ( ( void * ) netdev ) + sizeof ( *netdev ) ); } @@ -201,6 +202,9 @@ void unregister_netdev ( struct net_device *netdev ) { free_pkb ( pkb ); } + /* Kill off any persistent references to this device */ + forget_references ( &netdev->references ); + /* Remove from device list */ list_del ( &netdev->list ); DBG ( "%s unregistered\n", netdev_name ( netdev ) ); |