mirror of
https://github.com/ceph/ceph
synced 2025-02-10 04:19:56 +00:00
9213a23f14
git-svn-id: https://ceph.svn.sf.net/svnroot/ceph@1138 29311d96-e01e-0410-9327-a35deaab8ce9
88 lines
2.0 KiB
C++
88 lines
2.0 KiB
C++
|
|
#include "tcp.h"
|
|
|
|
/******************
|
|
* tcp crap
|
|
*/
|
|
|
|
bool tcp_read(int sd, char *buf, int len)
|
|
{
|
|
while (len > 0) {
|
|
int got = ::recv( sd, buf, len, 0 );
|
|
if (got == 0) {
|
|
dout(18) << "tcp_read socket " << sd << " closed" << endl;
|
|
return false;
|
|
}
|
|
if (got < 0) {
|
|
dout(18) << "tcp_read bailing with " << got << endl;
|
|
return false;
|
|
}
|
|
assert(got >= 0);
|
|
len -= got;
|
|
buf += got;
|
|
//dout(DBL) << "tcp_read got " << got << ", " << len << " left" << endl;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
int tcp_write(int sd, char *buf, int len)
|
|
{
|
|
//dout(DBL) << "tcp_write writing " << len << endl;
|
|
assert(len > 0);
|
|
while (len > 0) {
|
|
int did = ::send( sd, buf, len, 0 );
|
|
if (did < 0) {
|
|
dout(1) << "tcp_write error did = " << did << " errno " << errno << " " << strerror(errno) << endl;
|
|
//cerr << "tcp_write error did = " << did << " errno " << errno << " " << strerror(errno) << endl;
|
|
}
|
|
//assert(did >= 0);
|
|
if (did < 0) return did;
|
|
len -= did;
|
|
buf += did;
|
|
//dout(DBL) << "tcp_write did " << did << ", " << len << " left" << endl;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
int tcp_hostlookup(char *str, tcpaddr_t& 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'" << endl;
|
|
return -1;
|
|
}
|
|
//cout << "host '" << host << "' port '" << port << "'" << endl;
|
|
|
|
int iport = atoi(port);
|
|
|
|
struct hostent *myhostname = gethostbyname( host );
|
|
if (!myhostname) {
|
|
cerr << "host " << host << " not found" << endl;
|
|
return -1;
|
|
}
|
|
|
|
memset(&ta, 0, sizeof(ta));
|
|
|
|
//cout << "addrtype " << myhostname->h_addrtype << " len " << myhostname->h_length << 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 << endl;
|
|
|
|
return 0;
|
|
}
|