Change net/socket.c to use socket_*() functions
Use socket_*() functions from include/qemu/sockets.h instead of listen()/bind()/connect()/parse_host_port(). socket_*() fucntions are QAPI based and this patch performs this api conversion since everything will be using QAPI based sockets in the future. Also add a helper function socket_address_to_string() in util/qemu-sockets.c which returns the string representation of socket address. The task was listed on http://wiki.qemu.org/BiteSizedTasks page. Signed-off-by: Ashijeet Acharya <ashijeetacharya@gmail.com> Signed-off-by: Marc-André Lureau <marcandre.lureau@redhat.com> Signed-off-by: Paolo Bonzini <pbonzini@redhat.com>
This commit is contained in:
		
							parent
							
								
									d9911d14e0
								
							
						
					
					
						commit
						883e4f7624
					
				
							
								
								
									
										127
									
								
								net/socket.c
								
								
								
								
							
							
						
						
									
										127
									
								
								net/socket.c
								
								
								
								
							| 
						 | 
					@ -489,90 +489,105 @@ static int net_socket_listen_init(NetClientState *peer,
 | 
				
			||||||
{
 | 
					{
 | 
				
			||||||
    NetClientState *nc;
 | 
					    NetClientState *nc;
 | 
				
			||||||
    NetSocketState *s;
 | 
					    NetSocketState *s;
 | 
				
			||||||
    struct sockaddr_in saddr;
 | 
					    SocketAddress *saddr;
 | 
				
			||||||
    int fd, ret;
 | 
					    int ret;
 | 
				
			||||||
 | 
					    Error *local_error = NULL;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    if (parse_host_port(&saddr, host_str) < 0)
 | 
					    saddr = socket_parse(host_str, &local_error);
 | 
				
			||||||
        return -1;
 | 
					    if (saddr == NULL) {
 | 
				
			||||||
 | 
					        error_report_err(local_error);
 | 
				
			||||||
    fd = qemu_socket(PF_INET, SOCK_STREAM, 0);
 | 
					 | 
				
			||||||
    if (fd < 0) {
 | 
					 | 
				
			||||||
        perror("socket");
 | 
					 | 
				
			||||||
        return -1;
 | 
					        return -1;
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
    qemu_set_nonblock(fd);
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
    socket_set_fast_reuse(fd);
 | 
					    ret = socket_listen(saddr, &local_error);
 | 
				
			||||||
 | 
					 | 
				
			||||||
    ret = bind(fd, (struct sockaddr *)&saddr, sizeof(saddr));
 | 
					 | 
				
			||||||
    if (ret < 0) {
 | 
					    if (ret < 0) {
 | 
				
			||||||
        perror("bind");
 | 
					        qapi_free_SocketAddress(saddr);
 | 
				
			||||||
        closesocket(fd);
 | 
					        error_report_err(local_error);
 | 
				
			||||||
        return -1;
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
    ret = listen(fd, 0);
 | 
					 | 
				
			||||||
    if (ret < 0) {
 | 
					 | 
				
			||||||
        perror("listen");
 | 
					 | 
				
			||||||
        closesocket(fd);
 | 
					 | 
				
			||||||
        return -1;
 | 
					        return -1;
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    nc = qemu_new_net_client(&net_socket_info, peer, model, name);
 | 
					    nc = qemu_new_net_client(&net_socket_info, peer, model, name);
 | 
				
			||||||
    s = DO_UPCAST(NetSocketState, nc, nc);
 | 
					    s = DO_UPCAST(NetSocketState, nc, nc);
 | 
				
			||||||
    s->fd = -1;
 | 
					    s->fd = -1;
 | 
				
			||||||
    s->listen_fd = fd;
 | 
					    s->listen_fd = ret;
 | 
				
			||||||
    s->nc.link_down = true;
 | 
					    s->nc.link_down = true;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    qemu_set_fd_handler(s->listen_fd, net_socket_accept, NULL, s);
 | 
					    qemu_set_fd_handler(s->listen_fd, net_socket_accept, NULL, s);
 | 
				
			||||||
 | 
					    qapi_free_SocketAddress(saddr);
 | 
				
			||||||
    return 0;
 | 
					    return 0;
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					typedef struct {
 | 
				
			||||||
 | 
					    NetClientState *peer;
 | 
				
			||||||
 | 
					    SocketAddress *saddr;
 | 
				
			||||||
 | 
					    char *model;
 | 
				
			||||||
 | 
					    char *name;
 | 
				
			||||||
 | 
					} socket_connect_data;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					static void socket_connect_data_free(socket_connect_data *c)
 | 
				
			||||||
 | 
					{
 | 
				
			||||||
 | 
					    qapi_free_SocketAddress(c->saddr);
 | 
				
			||||||
 | 
					    g_free(c->model);
 | 
				
			||||||
 | 
					    g_free(c->name);
 | 
				
			||||||
 | 
					    g_free(c);
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					static void net_socket_connected(int fd, Error *err, void *opaque)
 | 
				
			||||||
 | 
					{
 | 
				
			||||||
 | 
					    socket_connect_data *c = opaque;
 | 
				
			||||||
 | 
					    NetSocketState *s;
 | 
				
			||||||
 | 
					    char *addr_str = NULL;
 | 
				
			||||||
 | 
					    Error *local_error = NULL;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    addr_str = socket_address_to_string(c->saddr, &local_error);
 | 
				
			||||||
 | 
					    if (addr_str == NULL) {
 | 
				
			||||||
 | 
					        error_report_err(local_error);
 | 
				
			||||||
 | 
					        closesocket(fd);
 | 
				
			||||||
 | 
					        goto end;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    s = net_socket_fd_init(c->peer, c->model, c->name, fd, true);
 | 
				
			||||||
 | 
					    if (!s) {
 | 
				
			||||||
 | 
					        closesocket(fd);
 | 
				
			||||||
 | 
					        goto end;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    snprintf(s->nc.info_str, sizeof(s->nc.info_str),
 | 
				
			||||||
 | 
					             "socket: connect to %s", addr_str);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					end:
 | 
				
			||||||
 | 
					    g_free(addr_str);
 | 
				
			||||||
 | 
					    socket_connect_data_free(c);
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
static int net_socket_connect_init(NetClientState *peer,
 | 
					static int net_socket_connect_init(NetClientState *peer,
 | 
				
			||||||
                                   const char *model,
 | 
					                                   const char *model,
 | 
				
			||||||
                                   const char *name,
 | 
					                                   const char *name,
 | 
				
			||||||
                                   const char *host_str)
 | 
					                                   const char *host_str)
 | 
				
			||||||
{
 | 
					{
 | 
				
			||||||
    NetSocketState *s;
 | 
					    socket_connect_data *c = g_new0(socket_connect_data, 1);
 | 
				
			||||||
    int fd, connected, ret;
 | 
					    int fd = -1;
 | 
				
			||||||
    struct sockaddr_in saddr;
 | 
					    Error *local_error = NULL;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    if (parse_host_port(&saddr, host_str) < 0)
 | 
					    c->peer = peer;
 | 
				
			||||||
        return -1;
 | 
					    c->model = g_strdup(model);
 | 
				
			||||||
 | 
					    c->name = g_strdup(name);
 | 
				
			||||||
 | 
					    c->saddr = socket_parse(host_str, &local_error);
 | 
				
			||||||
 | 
					    if (c->saddr == NULL) {
 | 
				
			||||||
 | 
					        goto err;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    fd = qemu_socket(PF_INET, SOCK_STREAM, 0);
 | 
					    fd = socket_connect(c->saddr, &local_error, net_socket_connected, c);
 | 
				
			||||||
    if (fd < 0) {
 | 
					    if (fd < 0) {
 | 
				
			||||||
        perror("socket");
 | 
					        goto err;
 | 
				
			||||||
        return -1;
 | 
					 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
    qemu_set_nonblock(fd);
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
    connected = 0;
 | 
					 | 
				
			||||||
    for(;;) {
 | 
					 | 
				
			||||||
        ret = connect(fd, (struct sockaddr *)&saddr, sizeof(saddr));
 | 
					 | 
				
			||||||
        if (ret < 0) {
 | 
					 | 
				
			||||||
            if (errno == EINTR || errno == EWOULDBLOCK) {
 | 
					 | 
				
			||||||
                /* continue */
 | 
					 | 
				
			||||||
            } else if (errno == EINPROGRESS ||
 | 
					 | 
				
			||||||
                       errno == EALREADY ||
 | 
					 | 
				
			||||||
                       errno == EINVAL) {
 | 
					 | 
				
			||||||
                break;
 | 
					 | 
				
			||||||
            } else {
 | 
					 | 
				
			||||||
                perror("connect");
 | 
					 | 
				
			||||||
                closesocket(fd);
 | 
					 | 
				
			||||||
                return -1;
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        } else {
 | 
					 | 
				
			||||||
            connected = 1;
 | 
					 | 
				
			||||||
            break;
 | 
					 | 
				
			||||||
        }
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
    s = net_socket_fd_init(peer, model, name, fd, connected);
 | 
					 | 
				
			||||||
    if (!s)
 | 
					 | 
				
			||||||
        return -1;
 | 
					 | 
				
			||||||
    snprintf(s->nc.info_str, sizeof(s->nc.info_str),
 | 
					 | 
				
			||||||
             "socket: connect to %s:%d",
 | 
					 | 
				
			||||||
             inet_ntoa(saddr.sin_addr), ntohs(saddr.sin_port));
 | 
					 | 
				
			||||||
    return 0;
 | 
					    return 0;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					err:
 | 
				
			||||||
 | 
					    error_report_err(local_error);
 | 
				
			||||||
 | 
					    socket_connect_data_free(c);
 | 
				
			||||||
 | 
					    return -1;
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
static int net_socket_mcast_init(NetClientState *peer,
 | 
					static int net_socket_mcast_init(NetClientState *peer,
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
		Reference in New Issue