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.


Enhancements to VNET and to halting:
[palacios.git] / palacios / src / devices / lnx_virtio_nic.c
index 6300eb8..5e7d4f2 100644 (file)
@@ -1,3 +1,4 @@
+
 /* 
  * This file is part of the Palacios Virtual Machine Monitor developed
  * by the V3VEE Project with funding from the United States National 
@@ -131,8 +132,6 @@ struct virtio_net_state {
     uint8_t mergeable_rx_bufs;
 
     struct v3_timer * timer;
-    struct vnet_thread * poll_thread;
-
     struct nic_statistics stats;
 
     struct v3_dev_net_ops * net_ops;
@@ -200,7 +199,7 @@ static int tx_one_pkt(struct guest_info * core,
        PrintDebug("Could not translate buffer address\n");
        return -1;
     }
-
+    
     V3_Net_Print(2, "Virtio-NIC: virtio_tx: size: %d\n", len);
     if(net_debug >= 4){
        v3_hexdump(buf, len, NULL, 0);
@@ -210,10 +209,10 @@ static int tx_one_pkt(struct guest_info * core,
        virtio->stats.tx_dropped ++;
        return -1;
     }
-
+    
     virtio->stats.tx_pkts ++;
     virtio->stats.tx_bytes += len;
-
+    
     return 0;
 }
 
@@ -227,7 +226,7 @@ static inline int copy_data_to_desc(struct guest_info * core,
                                    uint_t dst_offset){
     uint32_t len;
     uint8_t * desc_buf = NULL;
-
+    
     if (v3_gpa_to_hva(core, desc->addr_gpa, (addr_t *)&(desc_buf)) == -1) {
        PrintDebug("Could not translate buffer address\n");
        return -1;
@@ -267,81 +266,77 @@ static int handle_pkt_tx(struct guest_info * core,
                         struct virtio_net_state * virtio_state,
                         int quote)
 {
-    struct virtio_queue *q = &(virtio_state->tx_vq);
+    struct virtio_queue * q;
     int txed = 0, left = 0;
     unsigned long flags;
 
+    q = &(virtio_state->tx_vq);
     if (!q->ring_avail_addr) {
        return -1;
     }
 
-    flags = v3_lock_irqsave(virtio_state->tx_lock);
-    while (q->cur_avail_idx != q->avail->index) {
-       struct virtio_net_hdr_mrg_rxbuf * hdr = NULL;
+    while (1) {
        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];
-       int desc_cnt = get_desc_count(q, desc_idx);
+       uint16_t desc_idx, tmp_idx;
+       int desc_cnt;
+       
+       flags = v3_lock_irqsave(virtio_state->tx_lock);
 
+       if(q->cur_avail_idx == q->avail->index ||
+           (quote > 0 && txed >= quote)) {
+           left = (q->cur_avail_idx != q->avail->index);
+           v3_unlock_irqrestore(virtio_state->tx_lock, flags);
+           break;
+       }
+       
+       desc_idx = q->avail->ring[q->cur_avail_idx % q->queue_size];
+       tmp_idx = q->cur_avail_idx ++;
+       
+       v3_unlock_irqrestore(virtio_state->tx_lock, flags);
+
+       desc_cnt = get_desc_count(q, desc_idx);
        if(desc_cnt != 2){
            PrintError("VNIC: merged rx buffer not supported, desc_cnt %d\n", desc_cnt);
-           goto exit_error;
        }
 
        hdr_desc = &(q->desc[desc_idx]);
-       if (v3_gpa_to_hva(core, hdr_desc->addr_gpa, &(hdr_addr)) == -1) {
-           PrintError("Could not translate block header address\n");
-           goto exit_error;
-       }
+       if (v3_gpa_to_hva(core, hdr_desc->addr_gpa, &(hdr_addr)) != -1) {
+           struct virtio_net_hdr_mrg_rxbuf * hdr;
+           struct vring_desc * buf_desc;
 
-       hdr = (struct virtio_net_hdr_mrg_rxbuf *)hdr_addr;
-       desc_idx = hdr_desc->next;
+           hdr = (struct virtio_net_hdr_mrg_rxbuf *)hdr_addr;
+           desc_idx = hdr_desc->next;
 
-       V3_Net_Print(2, "Virtio NIC: TX hdr count : %d\n", hdr->num_buffers);
-
-       /* here we assumed that one ethernet pkt is not splitted into multiple buffer */        
-       struct vring_desc * buf_desc = &(q->desc[desc_idx]);
-       if (tx_one_pkt(core, virtio_state, buf_desc) == -1) {
-           PrintError("Virtio NIC: Error handling nic operation\n");
-           goto exit_error;
-       }
-       if(buf_desc->next & VIRTIO_NEXT_FLAG){
-           PrintError("Virtio NIC: TX more buffer need to read\n");
+           /* here we assumed that one ethernet pkt is not splitted into multiple buffer */    
+           buf_desc = &(q->desc[desc_idx]);
+           if (tx_one_pkt(core, virtio_state, buf_desc) == -1) {
+               PrintError("Virtio NIC: Fails to send packet\n");
+           }
+       } else {
+           PrintError("Could not translate block header address\n");
        }
-           
-       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 = buf_desc->length; /* What do we set this to???? */
-       q->used->index ++;
+
+       flags = v3_lock_irqsave(virtio_state->tx_lock);
        
-       q->cur_avail_idx ++;
+       q->used->ring[q->used->index % q->queue_size].id = 
+           q->avail->ring[tmp_idx % q->queue_size];
        
-       if(++txed >= quote && quote > 0){
-           left = (q->cur_avail_idx != q->avail->index);
-           break;
-       }
-    }
-
-    v3_unlock_irqrestore(virtio_state->tx_lock, flags);
+       q->used->index ++;
+       
+       v3_unlock_irqrestore(virtio_state->tx_lock, flags);
 
+       txed ++;
+    }
+        
     if (txed && !(q->avail->flags & VIRTIO_NO_IRQ_FLAG)) {
        v3_pci_raise_irq(virtio_state->virtio_dev->pci_bus, 
-                        0, virtio_state->pci_dev);
+                        virtio_state->pci_dev, 0);
        virtio_state->virtio_cfg.pci_isr = 0x1;
        virtio_state->stats.rx_interrupts ++;
     }
 
-    V3_Print("Virtio Intr Line %d\n", virtio_state->pci_dev->config_header.intr_line);
-
-    if(txed > 0) {
-       V3_Net_Print(2, "Virtio Handle TX: txed pkts: %d, left %d\n", txed, left);
-    }
-
     return left;
-
-exit_error:
-       
-    v3_unlock_irqrestore(virtio_state->tx_lock, flags);
-    return -1;
 }
 
 
@@ -425,8 +420,8 @@ static int virtio_io_write(struct guest_info *core,
                                       pfn, page_addr);
                    if(virtio->tx_notify == 0){
                        disable_cb(&virtio->tx_vq);
-                       vnet_thread_wakeup(virtio->poll_thread);
                    }
+                   virtio->status = 1;
                    break;
                case 2:
                    virtio_setup_queue(core, virtio, 
@@ -558,7 +553,7 @@ static int virtio_io_read(struct guest_info *core,
            *(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);
+                            virtio->pci_dev, 0);
            break;
 
        case VIRTIO_NET_CONFIG ... VIRTIO_NET_CONFIG + ETH_ALEN:
@@ -713,7 +708,7 @@ static int virtio_rx(uint8_t * buf, uint32_t size, void * private_data) {
                     virtio->pci_dev->config_header.intr_line);
 
        virtio->virtio_cfg.pci_isr = 0x1;       
-       v3_pci_raise_irq(virtio->virtio_dev->pci_bus, 0, virtio->pci_dev);
+       v3_pci_raise_irq(virtio->virtio_dev->pci_bus, virtio->pci_dev, 0);
        virtio->stats.rx_interrupts ++;
     }
 
@@ -760,7 +755,12 @@ static struct v3_device_ops dev_ops = {
 static int virtio_poll(int quote, void * data){
     struct virtio_net_state * virtio  = (struct virtio_net_state *)data;
 
-    return handle_pkt_tx(&(virtio->vm->cores[0]), virtio, quote);
+    if (virtio->status) {
+
+       return handle_pkt_tx(&(virtio->vm->cores[0]), virtio, quote);
+    } 
+
+    return 0;
 }
 
 static int register_dev(struct virtio_dev_state * virtio, 
@@ -803,7 +803,7 @@ static int register_dev(struct virtio_dev_state * virtio,
     pci_dev = v3_pci_register_device(virtio->pci_bus, PCI_STD_DEVICE, 
                                     0, PCI_AUTO_DEV_NUM, 0,
                                     "LNX_VIRTIO_NIC", bars,
-                                    NULL, NULL, NULL, net_state);
+                                    NULL, NULL, NULL, NULL, net_state);
     
     if (!pci_dev) {
        PrintError("Virtio NIC: Could not register PCI Device\n");
@@ -838,7 +838,7 @@ static int register_dev(struct virtio_dev_state * virtio,
     return 0;
 }
 
-
+#if 0
 #define RATE_UPPER_THRESHOLD 10  /* 10000 pkts per second, around 100Mbits */
 #define RATE_LOWER_THRESHOLD 1
 #define PROFILE_PERIOD 10000 /*us*/
@@ -870,7 +870,6 @@ static void virtio_nic_timer(struct guest_info * core,
            V3_Print("Virtio NIC: Switch TX to VMM driven mode\n");
            disable_cb(&(net_state->tx_vq));
            net_state->tx_notify = 0;
-           vnet_thread_wakeup(net_state->poll_thread);
        }
 
        if(tx_rate < RATE_LOWER_THRESHOLD && net_state->tx_notify == 0){
@@ -907,7 +906,7 @@ static void virtio_nic_timer(struct guest_info * core,
 static struct v3_timer_ops timer_ops = {
     .update_timer = virtio_nic_timer,
 };
-
+#endif
 
 static int connect_fn(struct v3_vm_info * info, 
                      void * frontend_data, 
@@ -917,6 +916,11 @@ static int connect_fn(struct v3_vm_info * info,
     struct virtio_dev_state * virtio = (struct virtio_dev_state *)frontend_data;
     struct virtio_net_state * net_state  = (struct virtio_net_state *)V3_Malloc(sizeof(struct virtio_net_state));
 
+    if (!net_state) {
+       PrintError("Cannot allocate in connect\n");
+       return -1;
+    }
+
     memset(net_state, 0, sizeof(struct virtio_net_state));
     register_dev(virtio, net_state);
 
@@ -924,11 +928,11 @@ static int connect_fn(struct v3_vm_info * info,
     net_state->net_ops = ops;
     net_state->backend_data = private_data;
     net_state->virtio_dev = virtio;
-    net_state->tx_notify = 0;
-    net_state->rx_notify = 0;
+    
+    net_state->tx_notify = 1;
+    net_state->rx_notify = 1;
        
-    net_state->timer = v3_add_timer(&(info->cores[0]),
-                                   &timer_ops,net_state);
+    //net_state->timer = v3_add_timer(&(info->cores[0]), &timer_ops,net_state);
 
     ops->recv = virtio_rx;
     ops->poll = virtio_poll;
@@ -936,9 +940,14 @@ static int connect_fn(struct v3_vm_info * info,
     ops->config.poll = 1;
     ops->config.quote = 64;
     ops->config.fnt_mac = V3_Malloc(ETH_ALEN);  
-    memcpy(ops->config.fnt_mac, virtio->mac, ETH_ALEN);
 
-    net_state->status = 1;
+    if (!ops->config.fnt_mac) { 
+       PrintError("Cannot allocate in connect\n");
+       // should unregister here
+       return -1;
+    }
+
+    memcpy(ops->config.fnt_mac, virtio->mac, ETH_ALEN);
 
     return 0;
 }
@@ -957,6 +966,12 @@ static int virtio_init(struct v3_vm_info * vm, v3_cfg_tree_t * cfg) {
     }
 
     virtio_state  = (struct virtio_dev_state *)V3_Malloc(sizeof(struct virtio_dev_state));
+
+    if (!virtio_state) {
+       PrintError("Cannot allocate in init\n");
+       return -1;
+    }
+
     memset(virtio_state, 0, sizeof(struct virtio_dev_state));
 
     INIT_LIST_HEAD(&(virtio_state->dev_list));