Report RX errors via netdev_rx_err()
[people/holger/gpxe.git] / src / arch / i386 / drivers / net / undinet.c
index 442d696..03515fd 100644 (file)
@@ -22,7 +22,8 @@
 #include <pic8259.h>
 #include <biosint.h>
 #include <pnpbios.h>
-#include <gpxe/pkbuff.h>
+#include <basemem_packet.h>
+#include <gpxe/iobuf.h>
 #include <gpxe/netdevice.h>
 #include <gpxe/if_ether.h>
 #include <gpxe/ethernet.h>
 
 /** An UNDI NIC */
 struct undi_nic {
-       /** Entry point */
-       SEGOFF16_t entry;
        /** Assigned IRQ number */
        unsigned int irq;
+       /** Currently processing ISR */
+       int isr_processing;
+       /** Bug workarounds */
+       int hacks;
 };
 
+/**
+ * @defgroup undi_hacks UNDI workarounds
+ * @{
+ */
+
+/** Work around Etherboot 5.4 bugs */
+#define UNDI_HACK_EB54         0x0001
+
+/** @} */
+
 static void undinet_close ( struct net_device *netdev );
 
 /*****************************************************************************
@@ -131,7 +144,7 @@ static union u_PXENV_ANY __data16 ( undinet_params );
  * Used as the indirection vector for all UNDI API calls.  Resides in
  * base memory.
  */
-static SEGOFF16_t __data16 ( undinet_entry_point );
+SEGOFF16_t __data16 ( undinet_entry_point );
 #define undinet_entry_point __use_data16 ( undinet_entry_point )
 
 /**
@@ -145,7 +158,6 @@ static SEGOFF16_t __data16 ( undinet_entry_point );
  */
 static int undinet_call ( struct undi_nic *undinic, unsigned int function,
                          void *params, size_t params_len ) {
-       union u_PXENV_ANY *pxenv_any = params;
        PXENV_EXIT_t exit;
        int discard_b, discard_D;
        int rc;
@@ -153,7 +165,6 @@ static int undinet_call ( struct undi_nic *undinic, unsigned int function,
        /* Copy parameter block and entry point */
        assert ( params_len <= sizeof ( undinet_params ) );
        memcpy ( &undinet_params, params, params_len );
-       undinet_entry_point = undinic->entry;
 
        /* Call real-mode entry point.  This calling convention will
         * work with both the !PXE and the PXENV+ entry points.
@@ -180,16 +191,13 @@ static int undinet_call ( struct undi_nic *undinic, unsigned int function,
         */     
        gateA20_set();
 
-       /* Copy parameter block back */
-       memcpy ( params, &undinet_params, params_len );
-
        /* Determine return status code based on PXENV_EXIT and
         * PXENV_STATUS
         */
        if ( exit == PXENV_EXIT_SUCCESS ) {
                rc = 0;
        } else {
-               rc = -pxenv_any->Status;
+               rc = -undinet_params.Status;
                /* Paranoia; don't return success for the combination
                 * of PXENV_EXIT_FAILURE but PXENV_STATUS_SUCCESS
                 */
@@ -197,10 +205,31 @@ static int undinet_call ( struct undi_nic *undinic, unsigned int function,
                        rc = -EIO;
        }
 
+       /* If anything goes wrong, print as much debug information as
+        * it's possible to give.
+        */
        if ( rc != 0 ) {
+               SEGOFF16_t rm_params = {
+                       .segment = rm_ds,
+                       .offset = (intptr_t) &__from_data16 ( undinet_params ),
+               };
+
                DBGC ( undinic, "UNDINIC %p %s failed: %s\n", undinic,
                       undinet_function_name ( function ), strerror ( rc ) );
+               DBGC ( undinic, "UNDINIC %p parameters at %04x:%04x length "
+                      "%#02x, entry point at %04x:%04x\n", undinic,
+                      rm_params.segment, rm_params.offset, params_len,
+                      undinet_entry_point.segment,
+                      undinet_entry_point.offset );
+               DBGC ( undinic, "UNDINIC %p parameters provided:\n", undinic );
+               DBGC_HDA ( undinic, rm_params, params, params_len );
+               DBGC ( undinic, "UNDINIC %p parameters returned:\n", undinic );
+               DBGC_HDA ( undinic, rm_params, &undinet_params, params_len );
        }
+
+       /* Copy parameter block back */
+       memcpy ( params, &undinet_params, params_len );
+
        return rc;
 }
 
@@ -214,43 +243,39 @@ static int undinet_call ( struct undi_nic *undinic, unsigned int function,
 /**
  * UNDI interrupt service routine
  *
- * The UNDI ISR simply increments a counter (@c trigger_count) and
- * exits.
+ * The UNDI ISR increments a counter (@c trigger_count) and exits.
  */
-extern void undinet_isr ( void );
+extern void undiisr ( void );
+
+/** IRQ number */
+uint8_t __data16 ( undiisr_irq );
+#define undiisr_irq __use_data16 ( undiisr_irq )
 
-/** Dummy chain vector */
-static struct segoff prev_handler[ IRQ_MAX + 1 ];
+/** IRQ chain vector */
+struct segoff __data16 ( undiisr_next_handler );
+#define undiisr_next_handler __use_data16 ( undiisr_next_handler )
 
 /** IRQ trigger count */
-static volatile uint8_t __text16 ( trigger_count ) = 0;
-#define trigger_count __use_text16 ( trigger_count )
+volatile uint8_t __data16 ( undiisr_trigger_count ) = 0;
+#define undiisr_trigger_count __use_data16 ( undiisr_trigger_count )
+
+/** Last observed trigger count */
+static unsigned int last_trigger_count = 0;
 
 /**
  * Hook UNDI interrupt service routine
  *
  * @v irq              IRQ number
- *
- * The UNDI ISR specifically does @b not chain to the previous
- * interrupt handler.  BIOSes seem to install somewhat perverse
- * default interrupt handlers; some do nothing other than an iret (and
- * so will cause a screaming interrupt if there really is another
- * interrupting device) and some disable the interrupt at the PIC (and
- * so will bring our own interrupts to a shuddering halt).
  */
 static void undinet_hook_isr ( unsigned int irq ) {
 
        assert ( irq <= IRQ_MAX );
+       assert ( undiisr_irq == 0 );
 
-       __asm__ __volatile__ ( TEXT16_CODE ( "\nundinet_isr:\n\t"
-                                            "incb %%cs:%c0\n\t"
-                                            "iret\n\t" )
-                              : : "p" ( & __from_text16 ( trigger_count ) ) );
-
+       undiisr_irq = irq;
        hook_bios_interrupt ( IRQ_INT ( irq ),
-                             ( ( unsigned int ) undinet_isr ),
-                             &prev_handler[irq] );
-
+                             ( ( unsigned int ) undiisr ),
+                             &undiisr_next_handler );
 }
 
 /**
@@ -263,8 +288,9 @@ static void undinet_unhook_isr ( unsigned int irq ) {
        assert ( irq <= IRQ_MAX );
 
        unhook_bios_interrupt ( IRQ_INT ( irq ),
-                               ( ( unsigned int ) undinet_isr ),
-                               &prev_handler[irq] );
+                               ( ( unsigned int ) undiisr ),
+                               &undiisr_next_handler );
+       undiisr_irq = 0;
 }
 
 /**
@@ -273,11 +299,10 @@ static void undinet_unhook_isr ( unsigned int irq ) {
  * @ret triggered      ISR has been triggered since last check
  */
 static int undinet_isr_triggered ( void ) {
-       static unsigned int last_trigger_count = 0;
        unsigned int this_trigger_count;
 
        /* Read trigger_count.  Do this only once; it is volatile */
-       this_trigger_count = trigger_count;
+       this_trigger_count = undiisr_trigger_count;
 
        if ( this_trigger_count == last_trigger_count ) {
                /* Not triggered */
@@ -296,18 +321,6 @@ static int undinet_isr_triggered ( void ) {
  *****************************************************************************
  */
 
-/** Maximum length of a packet transmitted via the UNDI API */
-#define UNDI_PKB_LEN 1514
-
-/** A packet transmitted via the UNDI API */
-struct undi_packet {
-       uint8_t bytes[UNDI_PKB_LEN];
-};
-
-/** UNDI packet buffer */
-static struct undi_packet __data16 ( undinet_pkb );
-#define undinet_pkb __use_data16 ( undinet_pkb )
-
 /** UNDI transmit buffer descriptor */
 static struct s_PXENV_UNDI_TBD __data16 ( undinet_tbd );
 #define undinet_tbd __use_data16 ( undinet_tbd )
@@ -316,20 +329,30 @@ static struct s_PXENV_UNDI_TBD __data16 ( undinet_tbd );
  * Transmit packet
  *
  * @v netdev           Network device
- * @v pkb              Packet buffer
+ * @v iobuf            I/O buffer
  * @ret rc             Return status code
  */
 static int undinet_transmit ( struct net_device *netdev,
-                             struct pk_buff *pkb ) {
+                             struct io_buffer *iobuf ) {
        struct undi_nic *undinic = netdev->priv;
        struct s_PXENV_UNDI_TRANSMIT undi_transmit;
-       size_t len = pkb_len ( pkb );
+       size_t len = iob_len ( iobuf );
        int rc;
 
-       /* Copy packet to UNDI packet buffer */
-       if ( len > sizeof ( undinet_pkb ) )
-               len = sizeof ( undinet_pkb );
-       memcpy ( &undinet_pkb, pkb->data, len );
+       /* Technically, we ought to make sure that the previous
+        * transmission has completed before we re-use the buffer.
+        * However, many PXE stacks (including at least some Intel PXE
+        * stacks and Etherboot 5.4) fail to generate TX completions.
+        * In practice this won't be a problem, since our TX datapath
+        * has a very low packet volume and we can get away with
+        * assuming that a TX will be complete by the time we want to
+        * transmit the next packet.
+        */
+
+       /* Copy packet to UNDI I/O buffer */
+       if ( len > sizeof ( basemem_packet ) )
+               len = sizeof ( basemem_packet );
+       memcpy ( &basemem_packet, iobuf->data, len );
 
        /* Create PXENV_UNDI_TRANSMIT data structure */
        memset ( &undi_transmit, 0, sizeof ( undi_transmit ) );
@@ -344,21 +367,25 @@ static int undinet_transmit ( struct net_device *netdev,
        undinet_tbd.ImmedLength = len;
        undinet_tbd.Xmit.segment = rm_ds;
        undinet_tbd.Xmit.offset 
-               = ( ( unsigned ) & __from_data16 ( undinet_pkb ) );
+               = ( ( unsigned ) & __from_data16 ( basemem_packet ) );
 
        /* Issue PXE API call */
-       rc = undinet_call ( undinic, PXENV_UNDI_TRANSMIT, &undi_transmit,
-                           sizeof ( undi_transmit ) );
+       if ( ( rc = undinet_call ( undinic, PXENV_UNDI_TRANSMIT,
+                                  &undi_transmit,
+                                  sizeof ( undi_transmit ) ) ) != 0 )
+               goto done;
 
-       /* Free packet buffer and return */
-       free_pkb ( pkb );
+       /* Free I/O buffer */
+       netdev_tx_complete ( netdev, iobuf );
+
+ done:
        return rc;
 }
 
 /** 
  * Poll for received packets
  *
- * @v netdev   Network device
+ * @v netdev           Network device
  *
  * Fun, fun, fun.  UNDI drivers don't use polling; they use
  * interrupts.  We therefore cheat and pretend that an interrupt has
@@ -375,33 +402,38 @@ static int undinet_transmit ( struct net_device *netdev,
  * so instead of doing it the easy way we have to go to all the hassle
  * of installing a genuine interrupt service routine and dealing with
  * the wonderful 8259 Programmable Interrupt Controller.  Joy.
+ *
+ * Addendum (10/07/07).  When doing things such as iSCSI boot, in
+ * which we have to co-operate with a running OS, we can't get away
+ * with the "ISR-just-increments-a-counter-and-returns" trick at all,
+ * because it involves tying up the PIC for far too long, and other
+ * interrupt-dependent components (e.g. local disks) start breaking.
+ * We therefore implement a "proper" ISR which calls PXENV_UNDI_ISR
+ * from within interrupt context in order to deassert the device
+ * interrupt, and sends EOI if applicable.
  */
 static void undinet_poll ( struct net_device *netdev ) {
        struct undi_nic *undinic = netdev->priv;
        struct s_PXENV_UNDI_ISR undi_isr;
-       struct pk_buff *pkb = NULL;
+       struct io_buffer *iobuf = NULL;
        size_t len;
        size_t frag_len;
        int rc;
 
-       /* Do nothing unless ISR has been triggered */
-       if ( ! undinet_isr_triggered() )
-               return;
+       if ( ! undinic->isr_processing ) {
+               /* Do nothing unless ISR has been triggered */
+               if ( ! undinet_isr_triggered() )
+                       return;
 
-       /* See if this was our interrupt */
-       memset ( &undi_isr, 0, sizeof ( undi_isr ) );
-       undi_isr.FuncFlag = PXENV_UNDI_ISR_IN_START;
-       if ( ( rc = undinet_call ( undinic, PXENV_UNDI_ISR, &undi_isr,
-                                  sizeof ( undi_isr ) ) ) != 0 )
-               return;
-       if ( undi_isr.FuncFlag != PXENV_UNDI_ISR_OUT_OURS )
-               return;
-
-       /* Send EOI */
-       send_eoi ( undinic->irq );
+               /* Start ISR processing */
+               undinic->isr_processing = 1;
+               undi_isr.FuncFlag = PXENV_UNDI_ISR_IN_PROCESS;
+       } else {
+               /* Continue ISR processing */
+               undi_isr.FuncFlag = PXENV_UNDI_ISR_IN_GET_NEXT;
+       }
 
        /* Run through the ISR loop */
-       undi_isr.FuncFlag = PXENV_UNDI_ISR_IN_PROCESS;
        while ( 1 ) {
                if ( ( rc = undinet_call ( undinic, PXENV_UNDI_ISR, &undi_isr,
                                           sizeof ( undi_isr ) ) ) != 0 )
@@ -414,44 +446,54 @@ static void undinet_poll ( struct net_device *netdev ) {
                        /* Packet fragment received */
                        len = undi_isr.FrameLength;
                        frag_len = undi_isr.BufferLength;
-                       if ( ! pkb )
-                               pkb = alloc_pkb ( len );
-                       if ( ! pkb ) {
+                       if ( ! iobuf )
+                               iobuf = alloc_iob ( len );
+                       if ( ! iobuf ) {
                                DBGC ( undinic, "UNDINIC %p could not "
                                       "allocate %zd bytes for RX buffer\n",
                                       undinic, len );
-                               break;
+                               /* Fragment will be dropped */
+                               netdev_rx_err ( netdev, NULL, -ENOMEM );
+                               goto done;
                        }
-                       if ( frag_len > pkb_available ( pkb ) ) {
+                       if ( frag_len > iob_tailroom ( iobuf ) ) {
                                DBGC ( undinic, "UNDINIC %p fragment too "
                                       "large\n", undinic );
-                               frag_len = pkb_available ( pkb );
+                               frag_len = iob_tailroom ( iobuf );
                        }
-                       copy_from_real ( pkb_put ( pkb, frag_len ),
+                       copy_from_real ( iob_put ( iobuf, frag_len ),
                                         undi_isr.Frame.segment,
                                         undi_isr.Frame.offset, frag_len );
-                       if ( pkb_len ( pkb ) == len ) {
-                               netdev_rx ( netdev, pkb );
-                               pkb = NULL;
+                       if ( iob_len ( iobuf ) == len ) {
+                               /* Whole packet received; deliver it */
+                               netdev_rx ( netdev, iobuf );
+                               iobuf = NULL;
+                               /* Etherboot 5.4 fails to return all packets
+                                * under mild load; pretend it retriggered.
+                                */
+                               if ( undinic->hacks & UNDI_HACK_EB54 )
+                                       --last_trigger_count;
                        }
                        break;
                case PXENV_UNDI_ISR_OUT_DONE:
                        /* Processing complete */
+                       undinic->isr_processing = 0;
                        goto done;
                default:
                        /* Should never happen */
                        DBGC ( undinic, "UNDINIC %p ISR returned invalid "
                               "FuncFlag %04x\n", undinic, undi_isr.FuncFlag );
+                       undinic->isr_processing = 0;
                        goto done;
                }
                undi_isr.FuncFlag = PXENV_UNDI_ISR_IN_GET_NEXT;
        }
 
  done:
-       if ( pkb ) {
+       if ( iobuf ) {
                DBGC ( undinic, "UNDINIC %p returned incomplete packet\n",
                       undinic );
-               netdev_rx ( netdev, pkb );
+               netdev_rx_err ( netdev, iobuf, -EINVAL );
        }
 }
 
@@ -463,31 +505,33 @@ static void undinet_poll ( struct net_device *netdev ) {
  */
 static int undinet_open ( struct net_device *netdev ) {
        struct undi_nic *undinic = netdev->priv;
-       struct s_PXENV_UNDI_SET_STATION_ADDRESS set_address;
-       struct s_PXENV_UNDI_OPEN open;
+       struct s_PXENV_UNDI_SET_STATION_ADDRESS undi_set_address;
+       struct s_PXENV_UNDI_OPEN undi_open;
        int rc;
 
        /* Hook interrupt service routine and enable interrupt */
        undinet_hook_isr ( undinic->irq );
        enable_irq ( undinic->irq );
+       send_eoi ( undinic->irq );
 
        /* Set station address.  Required for some PXE stacks; will
         * spuriously fail on others.  Ignore failures.  We only ever
         * use it to set the MAC address to the card's permanent value
         * anyway.
         */
-       memcpy ( set_address.StationAddress, netdev->ll_addr,
-                sizeof ( set_address.StationAddress ) );
+       memcpy ( undi_set_address.StationAddress, netdev->ll_addr,
+                sizeof ( undi_set_address.StationAddress ) );
        undinet_call ( undinic, PXENV_UNDI_SET_STATION_ADDRESS,
-                      &set_address, sizeof ( set_address ) );
+                      &undi_set_address, sizeof ( undi_set_address ) );
 
        /* Open NIC */
-       memset ( &open, 0, sizeof ( open ) );
-       open.PktFilter = ( FLTR_DIRECTED | FLTR_BRDCST );
-       if ( ( rc = undinet_call ( undinic, PXENV_UNDI_OPEN, &open,
-                                  sizeof ( open ) ) ) != 0 )
+       memset ( &undi_open, 0, sizeof ( undi_open ) );
+       undi_open.PktFilter = ( FLTR_DIRECTED | FLTR_BRDCST );
+       if ( ( rc = undinet_call ( undinic, PXENV_UNDI_OPEN, &undi_open,
+                                  sizeof ( undi_open ) ) ) != 0 )
                goto err;
 
+       DBGC ( undinic, "UNDINIC %p opened\n", undinic );
        return 0;
 
  err:
@@ -502,16 +546,62 @@ static int undinet_open ( struct net_device *netdev ) {
  */
 static void undinet_close ( struct net_device *netdev ) {
        struct undi_nic *undinic = netdev->priv;
-       struct s_PXENV_UNDI_CLOSE close;
+       struct s_PXENV_UNDI_ISR undi_isr;
+       struct s_PXENV_UNDI_CLOSE undi_close;
+       int rc;
+
+       /* Ensure ISR has exited cleanly */
+       while ( undinic->isr_processing ) {
+               undi_isr.FuncFlag = PXENV_UNDI_ISR_IN_GET_NEXT;
+               if ( ( rc = undinet_call ( undinic, PXENV_UNDI_ISR, &undi_isr,
+                                          sizeof ( undi_isr ) ) ) != 0 )
+                       break;
+               switch ( undi_isr.FuncFlag ) {
+               case PXENV_UNDI_ISR_OUT_TRANSMIT:
+               case PXENV_UNDI_ISR_OUT_RECEIVE:
+                       /* Continue draining */
+                       break;
+               default:
+                       /* Stop processing */
+                       undinic->isr_processing = 0;
+                       break;
+               }
+       }
 
        /* Close NIC */
-       undinet_call ( undinic, PXENV_UNDI_CLOSE, &close, sizeof ( close ) );
+       undinet_call ( undinic, PXENV_UNDI_CLOSE, &undi_close,
+                      sizeof ( undi_close ) );
 
        /* Disable interrupt and unhook ISR */
        disable_irq ( undinic->irq );
        undinet_unhook_isr ( undinic->irq );
+
+       DBGC ( undinic, "UNDINIC %p closed\n", undinic );
+}
+
+/**
+ * Enable/disable interrupts
+ *
+ * @v netdev           Net device
+ * @v enable           Interrupts should be enabled
+ */
+static void undinet_irq ( struct net_device *netdev, int enable ) {
+       struct undi_nic *undinic = netdev->priv;
+
+       /* Cannot support interrupts yet */
+       DBGC ( undinic, "UNDINIC %p cannot %s interrupts\n",
+              undinic, ( enable ? "enable" : "disable" ) );
 }
 
+/** UNDI network device operations */
+static struct net_device_operations undinet_operations = {
+       .open           = undinet_open,
+       .close          = undinet_close,
+       .transmit       = undinet_transmit,
+       .poll           = undinet_poll,
+       .irq            = undinet_irq,
+};
+
 /**
  * Probe UNDI device
  *
@@ -525,6 +615,7 @@ int undinet_probe ( struct undi_device *undi ) {
        struct s_PXENV_UNDI_STARTUP undi_startup;
        struct s_PXENV_UNDI_INITIALIZE undi_initialize;
        struct s_PXENV_UNDI_GET_INFORMATION undi_info;
+       struct s_PXENV_UNDI_GET_IFACE_INFO undi_iface;
        struct s_PXENV_UNDI_SHUTDOWN undi_shutdown;
        struct s_PXENV_UNDI_CLEANUP undi_cleanup;
        struct s_PXENV_STOP_UNDI stop_undi;
@@ -534,21 +625,28 @@ int undinet_probe ( struct undi_device *undi ) {
        netdev = alloc_etherdev ( sizeof ( *undinic ) );
        if ( ! netdev )
                return -ENOMEM;
+       netdev_init ( netdev, &undinet_operations );
        undinic = netdev->priv;
        undi_set_drvdata ( undi, netdev );
+       netdev->dev = &undi->dev;
        memset ( undinic, 0, sizeof ( *undinic ) );
-       undinic->entry = undi->entry;
+       undinet_entry_point = undi->entry;
+       DBGC ( undinic, "UNDINIC %p using UNDI %p\n", undinic, undi );
 
        /* Hook in UNDI stack */
-       memset ( &start_undi, 0, sizeof ( start_undi ) );
-       start_undi.AX = undi->pci_busdevfn;
-       start_undi.BX = undi->isapnp_csn;
-       start_undi.DX = undi->isapnp_read_port;
-       start_undi.ES = BIOS_SEG;
-       start_undi.DI = find_pnp_bios();
-       if ( ( rc = undinet_call ( undinic, PXENV_START_UNDI, &start_undi,
-                                  sizeof ( start_undi ) ) ) != 0 )
-               goto err_start_undi;
+       if ( ! ( undi->flags & UNDI_FL_STARTED ) ) {
+               memset ( &start_undi, 0, sizeof ( start_undi ) );
+               start_undi.AX = undi->pci_busdevfn;
+               start_undi.BX = undi->isapnp_csn;
+               start_undi.DX = undi->isapnp_read_port;
+               start_undi.ES = BIOS_SEG;
+               start_undi.DI = find_pnp_bios();
+               if ( ( rc = undinet_call ( undinic, PXENV_START_UNDI,
+                                          &start_undi,
+                                          sizeof ( start_undi ) ) ) != 0 )
+                       goto err_start_undi;
+       }
+       undi->flags |= UNDI_FL_STARTED;
 
        /* Bring up UNDI stack */
        memset ( &undi_startup, 0, sizeof ( undi_startup ) );
@@ -573,22 +671,33 @@ int undinet_probe ( struct undi_device *undi ) {
                       undinic, undinic->irq );
                goto err_bad_irq;
        }
-       DBGC ( undinic, "UNDINIC %p (%s) using IRQ %d\n",
+       DBGC ( undinic, "UNDINIC %p is %s on IRQ %d\n",
               undinic, eth_ntoa ( netdev->ll_addr ), undinic->irq );
 
-       /* Point to NIC specific routines */
-       netdev->open     = undinet_open;
-       netdev->close    = undinet_close;
-       netdev->transmit = undinet_transmit;
-       netdev->poll     = undinet_poll;
+       /* Get interface information */
+       memset ( &undi_iface, 0, sizeof ( undi_iface ) );
+       if ( ( rc = undinet_call ( undinic, PXENV_UNDI_GET_IFACE_INFO,
+                                  &undi_iface,
+                                  sizeof ( undi_iface ) ) ) != 0 )
+               goto err_undi_get_iface_info;
+       DBGC ( undinic, "UNDINIC %p has type %s and link speed %ld\n",
+              undinic, undi_iface.IfaceType, undi_iface.LinkSpeed );
+       if ( strncmp ( ( ( char * ) undi_iface.IfaceType ), "Etherboot",
+                      sizeof ( undi_iface.IfaceType ) ) == 0 ) {
+               DBGC ( undinic, "UNDINIC %p Etherboot 5.4 workaround enabled\n",
+                      undinic );
+               undinic->hacks |= UNDI_HACK_EB54;
+       }
 
        /* Register network device */
        if ( ( rc = register_netdev ( netdev ) ) != 0 )
                goto err_register;
 
+       DBGC ( undinic, "UNDINIC %p added\n", undinic );
        return 0;
 
  err_register:
+ err_undi_get_iface_info:
  err_bad_irq:
  err_undi_get_information:
  err_undi_initialize:
@@ -605,7 +714,8 @@ int undinet_probe ( struct undi_device *undi ) {
        undinet_call ( undinic, PXENV_STOP_UNDI, &stop_undi,
                       sizeof ( stop_undi ) );
  err_start_undi:
-       free_netdev ( netdev );
+       netdev_nullify ( netdev );
+       netdev_put ( netdev );
        undi_set_drvdata ( undi, NULL );
        return rc;
 }
@@ -637,7 +747,14 @@ void undinet_remove ( struct undi_device *undi ) {
        memset ( &stop_undi, 0, sizeof ( stop_undi ) );
        undinet_call ( undinic, PXENV_STOP_UNDI, &stop_undi,
                       sizeof ( stop_undi ) );
+       undi->flags &= ~UNDI_FL_STARTED;
+
+       /* Clear entry point */
+       memset ( &undinet_entry_point, 0, sizeof ( undinet_entry_point ) );
 
        /* Free network device */
-       free_netdev ( netdev );
+       netdev_nullify ( netdev );
+       netdev_put ( netdev );
+
+       DBGC ( undinic, "UNDINIC %p removed\n", undinic );
 }