/* returns an initialised client, or NULL on error */
CLIENT *create_rpc_client(struct sockaddr_in *client_sock, struct addrinfo *hints, unsigned long prognum, unsigned long version, struct timeval timeout, struct sockaddr_in src_ip) {
CLIENT *client = NULL;
int sock;
long unsigned protocol; /* for portmapper */
char src[INET_ADDRSTRLEN];
char dst[INET_ADDRSTRLEN];
struct sockaddr_in getaddr; /* for getsockname */
socklen_t len = sizeof(getaddr);
/* Even if you specify a source address the portmapper will use the default one */
/* this applies to pmap_getport or clnt*_create */
/* so use our own get_rpc_port */
/* check if we need to use the portmapper, 0 = yes */
if (client_sock->sin_port == 0) {
client_sock->sin_port = htons(PMAPPORT); /* 111 */
sock = socket(AF_INET, hints->ai_socktype, 0);
if (sock < 0) {
perror("create_rpc_client(socket)");
return NULL;
}
/* set the source address if specified */
if (src_ip.sin_addr.s_addr) {
/* portmapper doesn't need a reserved port */
src_ip.sin_port = 0;
if (bind(sock, (struct sockaddr *) &src_ip, sizeof(src_ip)) == -1) {
perror("create_rpc_client(bind)");
return NULL;
}
}
if (connect(sock, (struct sockaddr *)client_sock, sizeof(struct sockaddr)) == 0) {
/* TCP */
if (hints->ai_socktype == SOCK_STREAM) {
protocol = PMAP_IPPROTO_TCP;
client = clnttcp_create(client_sock, PMAPPROG, PMAPVERS, &sock, 0, 0);
if (client == NULL) {
clnt_pcreateerror("clnttcp_create");
}
/* UDP */
} else {
protocol = PMAP_IPPROTO_UDP;
client = clntudp_create(client_sock, PMAPPROG, PMAPVERS, timeout, &sock);
if (client == NULL) {
clnt_pcreateerror("clntudp_create");
}
}
} else {
perror("create_rpc_client(connect)");
return NULL;
}
if (verbose) {
if (getsockname(sock, (struct sockaddr *)&getaddr, &len) == -1) {
perror("create_rpc_client(getsockname)");
/* this is just verbose output so don't return an error */
} else {
inet_ntop(AF_INET, (struct sockaddr_in *)&getaddr.sin_addr, src, INET_ADDRSTRLEN);
//inet_ntop(AF_INET, &(((struct sockaddr_in *)&client_sock)->sin_addr), dst, INET_ADDRSTRLEN);
inet_ntop(AF_INET, &(client_sock->sin_addr), dst, INET_ADDRSTRLEN);
debug("portmap request = %s:%u -> %s:%u\n", src, ntohs(getaddr.sin_port), dst, ntohs(client_sock->sin_port));
}
}
/* query the portmapper */
client_sock->sin_port = get_rpc_port(client, prognum, version, protocol);
/* close the portmapper connection */
client = destroy_rpc_client(client);
/* by this point we should know which port we're talking to */
debug("portmapper = %s:%u\n", dst, ntohs(client_sock->sin_port));
}
/* now make the client connection */
/* by now we should have a port defined unless the program isn't registered */
if (client_sock->sin_port) {
/* Make sure and make new sockets for each new connection */
/* clnttcp_create will happily reuse open sockets */
sock = socket(AF_INET, hints->ai_socktype, 0);
if (sock < 0) {
perror("create_rpc_client(socket)");
return NULL;
}
/* always try and bind to a low port first */
/* could check for root here but there are other mechanisms for allowing processes to bind to low ports */
if (bindresvport(sock, &src_ip) == -1) {
/* permission denied, ie we aren't root */
if (errno == EACCES) {
/* try an ephemeral port */
src_ip.sin_port = htons(0);
} else {
perror("create_rpc_client(bindresvport)");
return NULL;
}
}
/* now we're bound to a local socket, try and connect to the server */
if (connect(sock, (struct sockaddr *)client_sock, sizeof(struct sockaddr)) == 0) {
/* TCP */
if (hints->ai_socktype == SOCK_STREAM) {
/* TODO set recvsz and sendsz to the NFS blocksize */
client = clnttcp_create(client_sock, prognum, version, &sock, 0, 0);
if (client == NULL) {
clnt_pcreateerror("clnttcp_create");
}
/* UDP */
} else {
client = clntudp_create(client_sock, prognum, version, timeout, &sock);
if (client == NULL) {
clnt_pcreateerror("clntudp_create");
}
}
} else {
perror("create_rpc_client(connect)");
return NULL;
}
if (verbose) {
if (getsockname(sock, (struct sockaddr *)&getaddr, &len) == -1) {
perror("create_rpc_client(getsockname)");
/* this is just verbose output so don't return an error */
} else {
inet_ntop(AF_INET, (struct sockaddr_in *)&getaddr.sin_addr, src, INET_ADDRSTRLEN);
inet_ntop(AF_INET, &(client_sock->sin_addr), dst, INET_ADDRSTRLEN);
debug("Connected = %s:%u -> %s:%u\n", src, ntohs(getaddr.sin_port), dst, ntohs(client_sock->sin_port));
}
}
}
if (client) {
/* TODO check return values */
/* use AUTH_NONE authentication by default */
client->cl_auth = authnone_create();
/* set the RPC timeout */
clnt_control(client, CLSET_TIMEOUT, (char *)&timeout);
/* set the socket to close when the client is destroyed */
clnt_control(client, CLSET_FD_CLOSE, NULL);
}
return client;
}
inline krpc::Client server_test::connect() {
return krpc::connect("C++ClientTest", "localhost", get_rpc_port(), get_stream_port());
}
