1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
2 // vim: ts=8 sw=2 smarttab
4 * Ceph - scalable distributed file system
6 * Copyright (C) 2016 XSKY <haomai@xsky.com>
8 * Author: Haomai Wang <haomaiwang@gmail.com>
10 * This is free software; you can redistribute it and/or
11 * modify it under the terms of the GNU Lesser General Public
12 * License version 2.1, as published by the Free Software
13 * Foundation. See file COPYING.
17 #include "msg/async/net_handler.h"
18 #include "RDMAStack.h"
20 #define dout_subsys ceph_subsys_ms
22 #define dout_prefix *_dout << " RDMAServerSocketImpl "
24 RDMAServerSocketImpl::RDMAServerSocketImpl(CephContext *cct, Infiniband* i, RDMADispatcher *s, RDMAWorker *w, entity_addr_t& a)
25 : cct(cct), net(cct), server_setup_socket(-1), infiniband(i), dispatcher(s), worker(w), sa(a)
29 int RDMAServerSocketImpl::listen(entity_addr_t &sa, const SocketOptions &opt)
32 server_setup_socket = net.create_socket(sa.get_family(), true);
33 if (server_setup_socket < 0) {
35 lderr(cct) << __func__ << " failed to create server socket: "
36 << cpp_strerror(errno) << dendl;
40 rc = net.set_nonblock(server_setup_socket);
45 rc = net.set_socket_options(server_setup_socket, opt.nodelay, opt.rcbuf_size);
49 net.set_close_on_exec(server_setup_socket);
51 rc = ::bind(server_setup_socket, sa.get_sockaddr(), sa.get_sockaddr_len());
54 ldout(cct, 10) << __func__ << " unable to bind to " << sa.get_sockaddr()
55 << " on port " << sa.get_port() << ": " << cpp_strerror(errno) << dendl;
59 rc = ::listen(server_setup_socket, cct->_conf->ms_tcp_listen_backlog);
62 lderr(cct) << __func__ << " unable to listen on " << sa << ": " << cpp_strerror(errno) << dendl;
66 ldout(cct, 20) << __func__ << " bind to " << sa.get_sockaddr() << " on port " << sa.get_port() << dendl;
70 ::close(server_setup_socket);
71 server_setup_socket = -1;
75 int RDMAServerSocketImpl::accept(ConnectedSocket *sock, const SocketOptions &opt, entity_addr_t *out, Worker *w)
77 ldout(cct, 15) << __func__ << dendl;
81 socklen_t slen = sizeof(ss);
82 int sd = ::accept(server_setup_socket, (sockaddr*)&ss, &slen);
87 net.set_close_on_exec(sd);
88 int r = net.set_nonblock(sd);
94 r = net.set_socket_options(sd, opt.nodelay, opt.rcbuf_size);
100 assert(NULL != out); //out should not be NULL in accept connection
102 out->set_sockaddr((sockaddr*)&ss);
103 net.set_priority(sd, opt.priority, out->get_family());
105 RDMAConnectedSocketImpl* server;
106 //Worker* w = dispatcher->get_stack()->get_worker();
107 server = new RDMAConnectedSocketImpl(cct, infiniband, dispatcher, dynamic_cast<RDMAWorker*>(w));
108 server->set_accept_fd(sd);
109 ldout(cct, 20) << __func__ << " accepted a new QP, tcp_fd: " << sd << dendl;
110 std::unique_ptr<RDMAConnectedSocketImpl> csi(server);
111 *sock = ConnectedSocket(std::move(csi));
116 void RDMAServerSocketImpl::abort_accept()
118 if (server_setup_socket >= 0)
119 ::close(server_setup_socket);