On 12/18/2015 3:03 PM, Yuanhan Liu wrote: > We could first check if we need realloc vq or not, if so, > reallocate it. We then do similar to vhost dev realloc. > > This could get rid of the tons of repeated "if (realloc_dev)" > and "if (realloc_vq)" statements, therefore, makes code > a bit more readable. > > Signed-off-by: Yuanhan Liu <yuanhan.liu at linux.intel.com> > --- > lib/librte_vhost/virtio-net.c | 77 > ++++++++++++++++++++----------------------- > 1 file changed, 36 insertions(+), 41 deletions(-) > > diff --git a/lib/librte_vhost/virtio-net.c b/lib/librte_vhost/virtio-net.c > index 2f83438..31ca4f7 100644 > --- a/lib/librte_vhost/virtio-net.c > +++ b/lib/librte_vhost/virtio-net.c > @@ -441,64 +441,59 @@ static struct virtio_net* > numa_realloc(struct virtio_net *dev, int index) > { > int oldnode, newnode; > - struct virtio_net *old_dev, *new_dev = NULL; > - struct vhost_virtqueue *old_vq, *new_vq = NULL; > + struct virtio_net *old_dev; > + struct vhost_virtqueue *old_vq, *vq; > int ret; > - int realloc_dev = 0, realloc_vq = 0; > > old_dev = dev; > - old_vq = dev->virtqueue[index]; > + vq = old_vq = dev->virtqueue[index]; > > - ret = get_mempolicy(&newnode, NULL, 0, old_vq->desc, > - MPOL_F_NODE | MPOL_F_ADDR); > - ret = ret | get_mempolicy(&oldnode, NULL, 0, old_dev, > + ret = get_mempolicy(&newnode, NULL, 0, old_vq->desc, > MPOL_F_NODE | MPOL_F_ADDR); > + > + /* check if we need to reallocate vq */ > + ret = get_mempolicy(&oldnode, NULL, 0, old_vq, MPOL_F_NODE | > MPOL_F_ADDR);
Why remove the ret = ret | ? Both get_mempolicy could fail. > if (ret) { > RTE_LOG(ERR, VHOST_CONFIG, > - "Unable to get vring desc or dev numa information.\n"); > + "Unable to get vq numa information.\n"); > return dev; > } > - if (oldnode != newnode) > - realloc_dev = 1; > + if (oldnode != newnode) { > + RTE_LOG(INFO, VHOST_CONFIG, > + "reallocate vq from %d to %d node\n", oldnode, newnode); > + vq = rte_malloc_socket(NULL, sizeof(*vq), 0, newnode); > + if (!vq) > + return dev; > + > + memcpy(vq, old_vq, sizeof(*vq)); > + rte_free(old_vq); > + } > > - ret = get_mempolicy(&oldnode, NULL, 0, old_vq, > - MPOL_F_NODE | MPOL_F_ADDR); > + /* check if we need to reallocate dev */ > + ret = get_mempolicy(&oldnode, NULL, 0, old_dev, MPOL_F_NODE | > MPOL_F_ADDR); > if (ret) { > RTE_LOG(ERR, VHOST_CONFIG, > - "Unable to get vq numa information.\n"); > - return dev; > + "Unable to get vring desc or dev numa information.\n"); > + goto out; > } Why vring desc in the err message? -huawei [...]