mirror of
https://github.com/ceph/ceph
synced 2025-02-13 05:49:52 +00:00
94 lines
2.2 KiB
C++
94 lines
2.2 KiB
C++
|
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
|
||
|
|
// vim: ts=8 sw=2 smarttab
|
||
|
|
|
||
|
|
#include "tcp.h"
|
||
|
|
|
||
|
|
/******************
|
||
|
|
* tcp crap
|
||
|
|
*/
|
||
|
|
|
||
|
|
/*
|
||
|
|
inlined, see tcp.h
|
||
|
|
|
||
|
|
|
||
|
|
bool tcp_read(int sd, char *buf, int len)
|
||
|
|
{
|
||
|
|
while (len > 0) {
|
||
|
|
int got = ::recv( sd, buf, len, 0 );
|
||
|
|
if (got == 0) {
|
||
|
|
generic_dout(18) << "tcp_read socket " << sd << " closed" << dendl;
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
if (got < 0) {
|
||
|
|
generic_dout(18) << "tcp_read bailing with " << got << dendl;
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
assert(got >= 0);
|
||
|
|
len -= got;
|
||
|
|
buf += got;
|
||
|
|
//generic_dout(DBL) << "tcp_read got " << got << ", " << len << " left" << dendl;
|
||
|
|
}
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
int tcp_write(int sd, char *buf, int len)
|
||
|
|
{
|
||
|
|
//generic_dout(DBL) << "tcp_write writing " << len << dendl;
|
||
|
|
assert(len > 0);
|
||
|
|
while (len > 0) {
|
||
|
|
int did = ::send( sd, buf, len, 0 );
|
||
|
|
if (did < 0) {
|
||
|
|
generic_dout(1) << "tcp_write error did = " << did << " errno " << errno << " " << strerror(errno) << dendl;
|
||
|
|
//derr(0) << "tcp_write error did = " << did << " errno " << errno << " " << strerror(errno) << dendl;
|
||
|
|
}
|
||
|
|
//assert(did >= 0);
|
||
|
|
if (did < 0) return did;
|
||
|
|
len -= did;
|
||
|
|
buf += did;
|
||
|
|
//generic_dout(DBL) << "tcp_write did " << did << ", " << len << " left" << dendl;
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
*/
|
||
|
|
|
||
|
|
int tcp_hostlookup(char *str, sockaddr_in& ta)
|
||
|
|
{
|
||
|
|
char *host = str;
|
||
|
|
char *port = 0;
|
||
|
|
|
||
|
|
for (int i=0; str[i]; i++) {
|
||
|
|
if (str[i] == ':') {
|
||
|
|
port = str+i+1;
|
||
|
|
str[i] = 0;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (!port) {
|
||
|
|
cerr << "addr '" << str << "' doesn't look like 'host:port'" << std::endl;
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
//cout << "host '" << host << "' port '" << port << "'" << std::endl;
|
||
|
|
|
||
|
|
int iport = atoi(port);
|
||
|
|
|
||
|
|
struct hostent *myhostname = gethostbyname( host );
|
||
|
|
if (!myhostname) {
|
||
|
|
cerr << "host " << host << " not found" << std::endl;
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
memset(&ta, 0, sizeof(ta));
|
||
|
|
|
||
|
|
//cout << "addrtype " << myhostname->h_addrtype << " len " << myhostname->h_length << std::endl;
|
||
|
|
|
||
|
|
ta.sin_family = myhostname->h_addrtype;
|
||
|
|
memcpy((char *)&ta.sin_addr,
|
||
|
|
myhostname->h_addr,
|
||
|
|
myhostname->h_length);
|
||
|
|
ta.sin_port = iport;
|
||
|
|
|
||
|
|
cout << "lookup '" << host << ":" << port << "' -> " << ta << std::endl;
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|