Palacios Public Git Repository

To checkout Palacios execute

  git clone http://v3vee.org/palacios/palacios.web/palacios.git
This will give you the master branch. You probably want the devel branch or one of the release branches. To switch to the devel branch, simply execute
  cd palacios
  git checkout --track -b devel origin/devel
The other branches are similar.


Add lock to vmm_queue
[palacios.git] / palacios / src / devices / lnx_virtio_nic.c
index ea998af..e83a199 100644 (file)
@@ -26,6 +26,7 @@
 #include <palacios/vm_guest_mem.h>
 #include <palacios/vmm_sprintf.h>
 #include <palacios/vmm_vnet.h>
+#include <palacios/vmm_lock.h>
 
 #include <devices/pci.h>
 
 
 //#define VIRTIO_NIC_PROFILE
 
-/* The feature bitmap for virtio net */
-#define VIRTIO_NET_F_CSUM      0       /* Host handles pkts w/ partial csum */
-#define VIRTIO_NET_F_GUEST_CSUM        1       /* Guest handles pkts w/ partial csum */
-#define VIRTIO_NET_F_MAC       5       /* Host has given MAC address. */
-#define VIRTIO_NET_F_GSO       6       /* Host handles pkts w/ any GSO type */
-#define VIRTIO_NET_F_GUEST_TSO4        7       /* Guest can handle TSOv4 in. */
-#define VIRTIO_NET_F_GUEST_TSO6        8       /* Guest can handle TSOv6 in. */
-#define VIRTIO_NET_F_GUEST_ECN 9       /* Guest can handle TSO[6] w/ ECN in. */
-#define VIRTIO_NET_F_GUEST_UFO 10      /* Guest can handle UFO in. */
-#define VIRTIO_NET_F_HOST_TSO4 11      /* Host can handle TSOv4 in. */
-#define VIRTIO_NET_F_HOST_TSO6 12      /* Host can handle TSOv6 in. */
-#define VIRTIO_NET_F_HOST_ECN  13      /* Host can handle TSO[6] w/ ECN in. */
-#define VIRTIO_NET_F_HOST_UFO  14      /* Host can handle UFO in. */
-#define VIRTIO_NET_F_MRG_RXBUF 15      /* Host can merge receive buffers. */
-#define VIRTIO_NET_F_STATUS    16      /* virtio_net_config.status available */
-#define VIRTIO_NET_F_CTRL_VQ   17      /* Control channel available */
-#define VIRTIO_NET_F_CTRL_RX   18      /* Control channel RX mode support */
-#define VIRTIO_NET_F_CTRL_VLAN 19      /* Control channel VLAN filtering */
-#define VIRTIO_NET_F_CTRL_RX_EXTRA 20  /* Extra RX mode control support */
 #define VIRTIO_NET_S_LINK_UP   1       /* Link is up */
-
-/* Maximum packet size we can receive from tap device: header + 64k */
 #define VIRTIO_NET_MAX_BUFSIZE (sizeof(struct virtio_net_hdr) + (64 << 10))
 
 struct virtio_net_hdr {
@@ -66,7 +46,6 @@ struct virtio_net_hdr {
 
 #define VIRTIO_NET_HDR_GSO_NONE                0       /* Not a GSO frame */
        uint8_t gso_type;
-
        uint16_t hdr_len;               /* Ethernet + IP + tcp/udp hdrs */
        uint16_t gso_size;              /* Bytes to append to hdr_len per frame */
        uint16_t csum_start;    /* Position to start checksumming from */
@@ -94,6 +73,7 @@ struct virtio_net_state {
     struct virtio_net_config net_cfg;
     struct virtio_config virtio_cfg;
 
+    struct vm_device * dev;
     struct pci_device * pci_dev; 
     int io_range_size;
     
@@ -101,24 +81,24 @@ struct virtio_net_state {
     struct virtio_queue tx_vq;   //index 1, svq in Linux virtio driver, handle packet from guest
     struct virtio_queue ctrl_vq; //index 2, ctrol info from guest
 
+    ulong_t pkt_sent, pkt_recv, pkt_drop;
+    struct gen_queue * inpkt_q;
+
     struct v3_dev_net_ops * net_ops;
 
     void * backend_data;
     struct virtio_dev_state * virtio_dev;
     struct list_head dev_link;
-
-    ulong_t pkt_sent, pkt_recv, pkt_drop;
 };
 
+struct eth_pkt {
+    uint32_t size; //size of data
+    char data[ETHERNET_PACKET_LEN];
+}__attribute__((packed));
 
-#ifdef CONFIG_DEBUG_VIRTIO_NET
-
-static void print_packet(uchar_t *pkt, int size) {
-    PrintDebug("Virtio Nic: print_data_packet: size: %d\n", size);
-    v3_hexdump(pkt, size, NULL, 0);
-}
-
-#endif
+//temporary store all of the virtio states
+static struct virtio_net_state *temp_net_states[4]; 
+static int net_idx = 0;
 
 #if 0
 //Temporarly for debug
@@ -163,7 +143,7 @@ static int virtio_free(struct vm_device * dev)
     return -1;
 }
 
-static int virtio_reset(struct virtio_net_state * virtio) 
+static int virtio_init_state(struct virtio_net_state * virtio) 
 {
     virtio->rx_vq.ring_desc_addr = 0;
     virtio->rx_vq.ring_avail_addr = 0;
@@ -187,6 +167,11 @@ static int virtio_reset(struct virtio_net_state * virtio)
     //virtio->virtio_cfg.status = VIRTIO_NET_S_LINK_UP;
     virtio->virtio_cfg.pci_isr = 0;
 
+    virtio->pkt_sent = virtio->pkt_recv = virtio->pkt_drop = 0;
+
+    virtio->inpkt_q = v3_create_queue();
+    v3_init_queue(virtio->inpkt_q);
+
     return 0;
 }
 
@@ -205,10 +190,6 @@ static int pkt_write(struct virtio_net_state * virtio, struct vring_desc * buf_d
        return -1;
     }
 
-    PrintDebug("Length=%d\n", buf_desc->length);
-    //PrintDebug("Buffer Descriptor (ptr=%p) gpa=%p, len=%d, flags=%x, next=%d, buf address: %p, send address: %p\n", buf_desc, 
-              //(void *)(buf_desc->addr_gpa), buf_desc->length, buf_desc->flags, buf_desc->next, buf, virtio->net_ops->send);
-
     if (virtio->net_ops->send(buf, len, (void *)virtio, NULL) == -1) {
        return -1;
     }
@@ -240,7 +221,6 @@ static int copy_data_to_desc(struct virtio_net_state * virtio_state, struct vrin
        PrintError("Could not translate buffer address\n");
        return -1;
     }
-
     len = (desc->length < buf_len)?desc->length:buf_len;
     memcpy(desc_buf, buf, len);
 
@@ -260,14 +240,12 @@ static int send_pkt_to_guest(struct virtio_net_state * virtio, uchar_t * buf, ui
     PrintDebug("VIRTIO NIC:  sending packet to virtio nic %p, size:%d", virtio, size);
 
     virtio->pkt_recv ++;
-       
     if (!raw) {
        data_len -= hdr_len;
     }
 
     build_receive_header(&hdr, buf, 1);
 
-    //queue is not set yet
     if (q->ring_avail_addr == 0) {
        PrintError("Queue is not set\n");
        return -1;
@@ -290,7 +268,6 @@ static int send_pkt_to_guest(struct virtio_net_state * virtio, uchar_t * buf, ui
        }
 
        memcpy((void *)hdr_addr, &hdr, sizeof(struct virtio_net_hdr));
-
        if (offset >= data_len) {
            hdr_desc->flags &= ~VIRTIO_NEXT_FLAG;
        }
@@ -299,20 +276,16 @@ static int send_pkt_to_guest(struct virtio_net_state * virtio, uchar_t * buf, ui
            struct vring_desc * buf_desc = &(q->desc[buf_idx]);
            uint32_t len = 0;
 
-           len = copy_data_to_desc(virtio, buf_desc, buf + offset, data_len - offset);
-           
+           len = copy_data_to_desc(virtio, buf_desc, buf + offset, data_len - offset);     
            offset += len;
-
            if (offset < data_len) {
                buf_desc->flags = VIRTIO_NEXT_FLAG;             
            }
-
-           buf_desc->length = len;  // TODO: do we need this?
+           buf_desc->length = len;
        }
        
        q->used->ring[q->used->index % q->queue_size].id = q->avail->ring[q->cur_avail_idx % q->queue_size];
        q->used->ring[q->used->index % q->queue_size].length = data_len + hdr_len; // This should be the total length of data sent to guest (header+pkt_data)
-
        q->used->index++;
 
        int last_idx = q->cur_avail_idx;
@@ -321,17 +294,12 @@ static int send_pkt_to_guest(struct virtio_net_state * virtio, uchar_t * buf, ui
            q->idx_overflow = false;
     } else {
        virtio->pkt_drop++;
-       
-#ifdef VIRTIO_NIC_PROFILE
-       PrintError("Virtio NIC: %p, one pkt dropped receieved: %ld, dropped: %ld, sent: %ld curidx: %d, avaiIdx: %d\n", virtio, virtio->pkt_recv, virtio->pkt_drop, virtio->pkt_sent, q->cur_avail_idx, q->avail->index);
-#endif
-    }
 
 #ifdef VIRTIO_NIC_PROFILE
-    if ((virtio->pkt_recv % 10000) == 0){
-       PrintError("Virtio NIC: %p, receieved: %ld, dropped: %ld, sent: %ld curidx: %d, avaiIdx: %d\n", virtio, virtio->pkt_recv, virtio->pkt_drop, virtio->pkt_sent, q->cur_avail_idx, q->avail->index);
-    }
+       PrintError("Virtio NIC: %p, one pkt dropped receieved: %ld, dropped: %ld, sent: %ld curidx: %d, avaiIdx: %d\n", 
+               virtio, virtio->pkt_recv, virtio->pkt_drop, virtio->pkt_sent, q->cur_avail_idx, q->avail->index);
 #endif
+    }
 
     if (!(q->avail->flags & VIRTIO_NO_IRQ_FLAG)) {
        PrintDebug("Raising IRQ %d\n",  virtio->pci_dev->config_header.intr_line);
@@ -339,6 +307,14 @@ static int send_pkt_to_guest(struct virtio_net_state * virtio, uchar_t * buf, ui
        virtio->virtio_cfg.pci_isr = 0x1;
     }
 
+#ifdef VIRTIO_NIC_PROFILE
+    if ((virtio->pkt_recv % 10000) == 0){
+       PrintError("Virtio NIC: %p, receieved: %ld, dropped: %ld, sent: %ld\n", 
+               virtio, virtio->pkt_recv, virtio->pkt_drop, virtio->pkt_sent);
+    }
+#endif
+
+    
     return offset;
 }
 
@@ -348,19 +324,74 @@ int virtio_send(struct guest_info * vm, char *dev_name, uchar_t * buf, uint_t si
     struct vm_device * virtio_dev = v3_find_dev(vm, dev_name);
 
     // TODO: how to get virtio net state from device??
-    // this is not right now
     struct virtio_net_state * virtio_state = (struct virtio_net_state *)virtio_dev->private_data;
        
     return send_pkt_to_guest(virtio_state, buf, size, 1, NULL);
 }
 
 
+int __virtio_sendto_buf(struct virtio_net_state * net_state, uchar_t * buf, uint_t size) {
+    struct eth_pkt *pkt;
+
+    pkt = (struct eth_pkt *)V3_Malloc(sizeof(struct eth_pkt));
+    if(pkt == NULL){
+        PrintError("Virtio NIC: Memory allocate fails\n");
+        return -1;
+    }
+  
+    pkt->size = size;
+    memcpy(pkt->data, buf, size);
+    v3_enqueue(net_state->inpkt_q, (addr_t)pkt);
+       
+    PrintDebug("Virtio NIC: __virtio_sendto_buf: transmitting packet: (size:%d)\n", (int)pkt->size);
+
+    return pkt->size;
+}
+
+
 static int __virtio_dev_send(uchar_t * buf, uint32_t size, void *private_data) {
     struct virtio_net_state *virtio_state = (struct virtio_net_state *)private_data;
        
-    return send_pkt_to_guest(virtio_state, buf, size, 1, NULL);
+    return __virtio_sendto_buf(virtio_state, buf, size);
+}
+
+/*
+  *called in svm/vmx handler
+  *iteative handled the unsent packet in incoming packet queues for
+  *all virtio nic devices in this guest
+  */
+int v3_virtionic_pktprocess(struct guest_info * info)
+{
+    struct eth_pkt *pkt = NULL;
+    struct virtio_net_state *net_state;
+    int i;
+
+    for (i = 0; i < net_idx; i++) {
+        while (1) {
+            net_state = temp_net_states[i];
+            if(net_state->dev->vm != info)
+                break;
+
+            pkt = (struct eth_pkt *)v3_dequeue(net_state->inpkt_q);
+            if(pkt == NULL) 
+                break;
+                       
+            if (send_pkt_to_guest(net_state, pkt->data, pkt->size, 1, NULL)) {
+                PrintDebug("Virtio NIC: %p In pkt_handle: send one packet! pt length %d\n", 
+                               net_state, (int)pkt->size);  
+            } else {
+                PrintDebug("Virtio NIC: %p In pkt_handle: Fail to send one packet, pt length %d, discard it!\n", 
+                               net_state, (int)pkt->size); 
+            }
+       
+            V3_Free(pkt);
+        }
+    }
+    
+    return 0;
 }
 
+
 static int get_desc_count(struct virtio_queue * q, int index) {
     struct vring_desc * tmp_desc = &(q->desc[index]);
     int cnt = 1;
@@ -384,18 +415,12 @@ static int handle_pkt_tx(struct virtio_net_state * virtio_state)
     struct virtio_queue * q = &(virtio_state->tx_vq);
     struct virtio_net_hdr * hdr = NULL;
 
-    virtio_state->pkt_sent ++;
-
     if (q->avail->index < q->last_avail_idx)
        q->idx_overflow = true;
     q->last_avail_idx = q->avail->index;
 
-#ifdef VIRTIO_NIC_PROFILE
-    if(virtio_state->pkt_sent % 10000 == 0)
-       PrintError("Virtio NIC: %p, pkt_sent: %ld curidx: %d, avaiIdx: %d\n", virtio_state, virtio_state->pkt_sent, q->cur_avail_idx, q->avail->index);
-#endif 
-
-    while (q->cur_avail_idx < q->avail->index ||(q->idx_overflow && q->cur_avail_idx < (q->avail->index + 65536))) {
+    while (q->cur_avail_idx < q->avail->index || 
+                (q->idx_overflow && q->cur_avail_idx < (q->avail->index + 65536))) {
        struct vring_desc * hdr_desc = NULL;
        addr_t hdr_addr = 0;
        uint16_t desc_idx = q->avail->ring[q->cur_avail_idx % q->queue_size];
@@ -422,10 +447,10 @@ static int handle_pkt_tx(struct virtio_net_state * virtio_state)
            req_len += buf_desc->length;
            desc_idx = buf_desc->next;
        }
+       virtio_state->pkt_sent ++;
 
        q->used->ring[q->used->index % q->queue_size].id = q->avail->ring[q->cur_avail_idx % q->queue_size];
        q->used->ring[q->used->index % q->queue_size].length = req_len; // What do we set this to????
-
        q->used->index++;
 
        int last_idx = q->cur_avail_idx;
@@ -439,10 +464,14 @@ static int handle_pkt_tx(struct virtio_net_state * virtio_state)
        virtio_state->virtio_cfg.pci_isr = 0x1;
     }
 
+#ifdef VIRTIO_NIC_PROFILE
+    if(virtio_state->pkt_sent % 10000 == 0)
+       PrintError("Virtio NIC: %p, pkt_sent: %ld\n", virtio_state, virtio_state->pkt_sent);
+#endif 
+
     return 0;
 }
 
-
 static int virtio_setup_queue(struct virtio_net_state * virtio_state, struct virtio_queue * queue, addr_t pfn, addr_t page_addr) {
     queue->pfn = pfn;
                
@@ -454,19 +483,16 @@ static int virtio_setup_queue(struct virtio_net_state * virtio_state, struct vir
                
     // round up to next page boundary.
     queue->ring_used_addr = (queue->ring_used_addr + 0xfff) & ~0xfff;
-
     if (guest_pa_to_host_va(virtio_state->virtio_dev->vm, queue->ring_desc_addr, (addr_t *)&(queue->desc)) == -1) {
         PrintError("Could not translate ring descriptor address\n");
         return -1;
     }
-
  
     if (guest_pa_to_host_va(virtio_state->virtio_dev->vm, queue->ring_avail_addr, (addr_t *)&(queue->avail)) == -1) {
         PrintError("Could not translate ring available address\n");
         return -1;
     }
 
-
     if (guest_pa_to_host_va(virtio_state->virtio_dev->vm, queue->ring_used_addr, (addr_t *)&(queue->used)) == -1) {
         PrintError("Could not translate ring used address\n");
         return -1;
@@ -483,41 +509,33 @@ static int virtio_setup_queue(struct virtio_net_state * virtio_state, struct vir
     return 0;
 }
 
-
-
 static int virtio_io_write(uint16_t port, void * src, uint_t length, void * private_data) 
 {
     //struct vm_device * dev = (struct vm_device *)private_data;
     struct virtio_net_state * virtio = (struct virtio_net_state *)private_data;
     int port_idx = port % virtio->io_range_size;
 
-
     PrintDebug("VIRTIO NIC %p Write for port %d (index=%d) len=%d, value=%x\n", private_data,
               port, port_idx,  length, *(uint32_t *)src);
 
-
     switch (port_idx) {
        case GUEST_FEATURES_PORT:
            if (length != 4) {
                PrintError("Illegal write length for guest features\n");
                return -1;
-           }
-           
+           }       
            virtio->virtio_cfg.guest_features = *(uint32_t *)src;
            PrintDebug("Setting Guest Features to %x\n", virtio->virtio_cfg.guest_features);
-
            break;
+               
        case VRING_PG_NUM_PORT:
-
            if (length != 4) {
                PrintError("Illegal write length for page frame number\n");
                return -1;
            }
-
            addr_t pfn = *(uint32_t *)src;
            addr_t page_addr = (pfn << VIRTIO_PAGE_SHIFT);
            uint16_t queue_idx = virtio->virtio_cfg.vring_queue_selector;
-
            switch (queue_idx) {
                case 0:
                    virtio_setup_queue(virtio, &virtio->rx_vq, pfn, page_addr);
@@ -527,29 +545,24 @@ static int virtio_io_write(uint16_t port, void * src, uint_t length, void * priv
                    break;
                case 2:
                    virtio_setup_queue(virtio, &virtio->ctrl_vq, pfn, page_addr);
-                   break;
-                   
+                   break;          
                default:
                    break;
            }
-           
            break;
+               
        case VRING_Q_SEL_PORT:
            virtio->virtio_cfg.vring_queue_selector = *(uint16_t *)src;
-
            if (virtio->virtio_cfg.vring_queue_selector > 2) {
                PrintError("Virtio NIC device only uses 3 queue, selected %d\n", 
                           virtio->virtio_cfg.vring_queue_selector);
                return -1;
            }
-
            break;
+               
        case VRING_Q_NOTIFY_PORT: 
            {
-               uint16_t queue_idx = *(uint16_t *)src;     
-               
-               //PrintDebug("Handling Kick\n");
-               
+               uint16_t queue_idx = *(uint16_t *)src;                  
                if (queue_idx == 0){
                    PrintDebug("receive queue notification 0, packet get by Guest\n");
                } else if (queue_idx == 1){
@@ -565,18 +578,16 @@ static int virtio_io_write(uint16_t port, void * src, uint_t length, void * priv
                } else {
                    PrintError("Virtio NIC device only uses 3 queue, selected %d\n", 
                               queue_idx);
-               }
-               
+               }       
                break;          
            }
+       
        case VIRTIO_STATUS_PORT:
            virtio->virtio_cfg.status = *(uint8_t *)src;
-
            if (virtio->virtio_cfg.status == 0) {
                PrintDebug("Resetting device\n");
-               virtio_reset(virtio);
+               virtio_init_state(virtio);
            }
-
            break;
 
        case VIRTIO_ISR_PORT:
@@ -590,7 +601,6 @@ static int virtio_io_write(uint16_t port, void * src, uint_t length, void * priv
     return length;
 }
 
-
 static int virtio_io_read(uint16_t port, void * dst, uint_t length, void * private_data) 
 {
    //struct vm_device * dev = (struct vm_device *)private_data;
@@ -607,19 +617,15 @@ static int virtio_io_read(uint16_t port, void * dst, uint_t length, void * priva
                PrintError("Illegal read length for host features\n");
                return -1;
            }
-
            *(uint32_t *)dst = virtio->virtio_cfg.host_features;
-
            PrintDebug("value=0x%x\n", *(uint32_t *)dst);
-       
            break;
-       case VRING_PG_NUM_PORT:
 
+       case VRING_PG_NUM_PORT:
            if (length != 4) {
                PrintError("Illegal read length for page frame number\n");
                return -1;
            }
-
            switch (queue_idx) {
                case 0:
                    *(uint32_t *)dst = virtio->rx_vq.pfn;
@@ -633,17 +639,14 @@ static int virtio_io_read(uint16_t port, void * dst, uint_t length, void * priva
                default:
                    break;
            }
-
            PrintDebug(", value=0x%x\n", *(uint32_t *)dst);
-
            break;
-       case VRING_SIZE_PORT:
 
+       case VRING_SIZE_PORT:
            if (length != 2) {
                PrintError("Illegal read length for vring size\n");
                return -1;
            }
-
            switch (queue_idx) {
                case 0:
                    *(uint16_t *)dst = virtio->rx_vq.queue_size;
@@ -657,29 +660,23 @@ static int virtio_io_read(uint16_t port, void * dst, uint_t length, void * priva
                default:
                    break;
            }
-
            PrintDebug("queue index: %d, value=0x%x\n", (int)queue_idx, *(uint16_t *)dst);
-
            break;
-       case VIRTIO_STATUS_PORT:
 
+       case VIRTIO_STATUS_PORT:
            if (length != 1) {
                PrintError("Illegal read length for status\n");
                return -1;
            }
-
            *(uint8_t *)dst = virtio->virtio_cfg.status;
-
            PrintDebug(", value=0x%x\n", *(uint8_t *)dst);
            break;
-
+               
        case VIRTIO_ISR_PORT:
            *(uint8_t *)dst = virtio->virtio_cfg.pci_isr;
            virtio->virtio_cfg.pci_isr = 0;
            v3_pci_lower_irq(virtio->virtio_dev->pci_bus, 0, virtio->pci_dev);
-
-           PrintDebug(", value=0x%x\n", *(uint8_t *)dst);
-               
+           PrintDebug(", value=0x%x\n", *(uint8_t *)dst);              
            break;
 
        default:
@@ -698,7 +695,6 @@ static struct v3_device_ops dev_ops = {
     .stop = NULL,
 };
 
-
 static int register_dev(struct virtio_dev_state * virtio, struct virtio_net_state * net_state) 
 {
     struct pci_device * pci_dev = NULL;
@@ -709,7 +705,6 @@ static int register_dev(struct virtio_dev_state * virtio, struct virtio_net_stat
 
     // This gets the number of ports, rounded up to a power of 2
     net_state->io_range_size = 1; // must be a power of 2
-    
     while (tmp_ports > 0) {
        tmp_ports >>= 1;
        net_state->io_range_size <<= 1;
@@ -721,7 +716,6 @@ static int register_dev(struct virtio_dev_state * virtio, struct virtio_net_stat
        net_state->io_range_size >>= 1;
     }
     
-    
     for (i = 0; i < 6; i++) {
        bars[i].type = PCI_BAR_NONE;
     }
@@ -731,7 +725,6 @@ static int register_dev(struct virtio_dev_state * virtio, struct virtio_net_stat
     bars[0].type = PCI_BAR_IO;
     bars[0].default_base_port = -1;
     bars[0].num_ports = net_state->io_range_size;
-    
     bars[0].io_read = virtio_io_read;
     bars[0].io_write = virtio_io_write;
     bars[0].private_data = net_state;
@@ -754,34 +747,23 @@ static int register_dev(struct virtio_dev_state * virtio, struct virtio_net_stat
 
     pci_dev->config_header.device_id = VIRTIO_NET_DEV_ID;
     pci_dev->config_header.class = PCI_CLASS_NETWORK;
-    pci_dev->config_header.subclass = PCI_NET_SUBCLASS_OTHER;
-    
-    pci_dev->config_header.subsystem_id = VIRTIO_NET_SUBDEVICE_ID;;
-
+    pci_dev->config_header.subclass = PCI_NET_SUBCLASS_OTHER;  
+    pci_dev->config_header.subsystem_id = VIRTIO_NET_SUBDEVICE_ID;
     pci_dev->config_header.intr_pin = 1;
-
     pci_dev->config_header.max_latency = 1; // ?? (qemu does it...)
 
-
-    net_state->pci_dev = pci_dev;
-    //net_state->pci_bus = pci_bus;
-    
+    net_state->pci_dev = pci_dev;   
     net_state->virtio_cfg.host_features = 0; //no features support now
-
     net_state->rx_vq.queue_size = QUEUE_SIZE;
     net_state->tx_vq.queue_size = QUEUE_SIZE;
     net_state->ctrl_vq.queue_size = CTRL_QUEUE_SIZE;
-
     net_state->virtio_dev = virtio;
-    net_state->pkt_sent = net_state->pkt_recv = net_state->pkt_drop = 0;
-
-    virtio_reset(net_state);
+   
+    virtio_init_state(net_state);
 
     return 0;
 }
 
-
-
 static int connect_fn(struct guest_info * info, 
                      void * frontend_data, 
                      struct v3_dev_net_ops * ops, 
@@ -807,13 +789,8 @@ static int connect_fn(struct guest_info * info,
 //used when virtio_nic get a packet from guest and send it to the backend
 // send packet to all of the virtio nic devices other than the sender
 static int vnet_send(uint8_t * buf, uint32_t len, void * private_data, struct vm_device *dest_dev){
-
     PrintDebug("Virito NIC: In vnet_send: guest net state %p\n", private_data);
 
-#ifdef CONFIG_DEBUG_VIRTIO_NET
-    print_packet(buf, len);
-#endif
-
     v3_vnet_send_rawpkt(buf, len, private_data);
     return 0;
 }
@@ -827,10 +804,6 @@ static int vnet_receive(uint8_t * buf, uint32_t count, void * private_data, stru
 static int virtio_input(uchar_t * buf, uint_t len, void * private_data){
     PrintDebug("Virito NIC: In virtio_input: guest net state %p\n", private_data);
 
-#ifdef CONFIG_DEBUG_VIRTIO_NET
-    print_packet(buf, len);
-#endif
-
     return __virtio_dev_send(buf, len, private_data);
 }
 
@@ -839,36 +812,35 @@ static int virtio_input(uchar_t * buf, uint_t len, void * private_data){
 void register_virtio_to_vnet(struct vm_device  *dev, 
                                                char *dev_name,
                                                uchar_t mac[6]){
-    struct virtio_net_state * net_state  = (struct virtio_net_state *)V3_Malloc(sizeof(struct virtio_net_state));
-    memset(net_state, 0, sizeof(struct virtio_net_state));
-
+    struct virtio_net_state * net_state;
     struct virtio_dev_state *virtio_state =  (struct virtio_dev_state *)dev->private_data;
+    //uchar_t tapmac[6] = {0x00,0x02,0x55,0x67,0x42,0x39}; //for Intel-VT test HW
+    uchar_t tapmac[6] = {0x6e,0xa8,0x75,0xf4,0x82,0x95};
+    uchar_t dstmac[6] = {0xff,0xff,0xff,0xff,0xff,0xff};
+    uchar_t zeromac[6] = {0,0,0,0,0,0};
 
-    net_state->net_ops = (struct v3_dev_net_ops *)V3_Malloc(sizeof(struct v3_dev_net_ops));
 
+    net_state  = (struct virtio_net_state *)V3_Malloc(sizeof(struct virtio_net_state));
+    memset(net_state, 0, sizeof(struct virtio_net_state));
+    net_state->net_ops = (struct v3_dev_net_ops *)V3_Malloc(sizeof(struct v3_dev_net_ops));
     net_state->net_ops->send = &vnet_send;
     net_state->net_ops->receive = &vnet_receive;
+    net_state->dev = dev;
 
     register_dev(virtio_state, net_state);
-       
-    PrintDebug("Virtio NIC After register Device %s: queue size: %d, %d\n", dev->name,
-              net_state->rx_vq.queue_size, net_state->tx_vq.queue_size);
-
-    PrintDebug("VNET: connect virtio nic state %p to vnet\n", net_state);
 
-    //add a device link to link table
+    PrintDebug("Virtio NIC register Device %s: queue size: %d, %d\n", dev->name,
+              net_state->rx_vq.queue_size, net_state->tx_vq.queue_size);
+    PrintDebug("Virtio NIC: connect virtio device %s, state %p, to vnet\n", dev->name, net_state);
+       
     int idx = vnet_register_device(dev, dev_name, mac, &virtio_input, net_state);
-
-    uchar_t srcmac[6] = {0x00,0x02,0x55,0x67,0x42,0x39};
-    uchar_t dstmac[6] = {0xff,0xff,0xff,0xff,0xff,0xff};
-    uchar_t zeromac[6] = {0,0,0,0,0,0};
-
     vnet_add_route_entry(zeromac, dstmac, MAC_ANY, MAC_NONE, idx, LINK_INTERFACE, -1, LINK_INTERFACE);
-    if (idx == 0)
-       vnet_add_route_entry(zeromac, srcmac, MAC_ANY, MAC_NONE, idx, LINK_INTERFACE, -1, LINK_INTERFACE);
-    if (idx == 1)
-       vnet_add_route_entry(srcmac, zeromac, MAC_NONE, MAC_ANY, idx, LINK_INTERFACE, -1, LINK_INTERFACE);
-               
+    if (!strcmp(dev_name, "net_virtiodom0"))
+       vnet_add_route_entry(zeromac, tapmac, MAC_ANY, MAC_NONE, idx, LINK_INTERFACE, -1, LINK_INTERFACE);
+    if (!strcmp(dev_name, "net_virtio"))
+       vnet_add_route_entry(tapmac, zeromac, MAC_NONE, MAC_ANY, idx, LINK_INTERFACE, -1, LINK_INTERFACE);
+
+    temp_net_states[net_idx ++] = net_state;
 }
 
 #endif
@@ -903,8 +875,7 @@ static int virtio_init(struct guest_info * vm, v3_cfg_tree_t * cfg) {
        return -1;
     }
 
-
-    //for temporary testing, add a backend
+    //for temporary testing, virtio bridge in Palacios
 #if 0
        
     struct virtio_net_state * net_state  = (struct virtio_net_state *)V3_Malloc(sizeof(struct virtio_net_state));
@@ -922,15 +893,21 @@ static int virtio_init(struct guest_info * vm, v3_cfg_tree_t * cfg) {
 
     temp_net_states[net_idx ++] = net_state;
 
-    PrintDebug("Net_states: 0: %p, 1: %p, 2: %p\n", temp_net_states[0], temp_net_states[1], temp_net_states[2]);
+    PrintDebug("Net_states: 0: %p, 1: %p, 2: %p\n", 
+               temp_net_states[0], 
+               temp_net_states[1], 
+               temp_net_states[2]);
 
 #endif
 
 #if 1  //test interface between vnet & virtio-nic
-
     uchar_t mac[6] = {0,0,0,0,0,0};
-    register_virtio_to_vnet(dev, name,mac);
+    register_virtio_to_vnet(dev, name, mac);
 
+    PrintDebug("Net_states: 0: %p, 1: %p, 2: %p\n", 
+               temp_net_states[0], 
+               temp_net_states[1], 
+               temp_net_states[2]);
 #endif
        
     return 0;