const char* aio_stream::get_peer(bool full /* = false */) const
{
if (stream_ == NULL)
return dummy_;
ACL_VSTREAM* vs = acl_aio_vstream(stream_);
const char* ptr = ACL_VSTREAM_PEER(vs);
if (ptr == NULL || *ptr == 0)
{
char buf[256];
ACL_SOCKET fd = ACL_VSTREAM_SOCK(vs);
if (acl_getpeername(fd, buf, sizeof(buf)) == -1)
return dummy_;
acl_vstream_set_peer(vs, buf);
}
ptr = ACL_VSTREAM_PEER(vs);
if (full)
return ptr;
else if (peer_ip_[0] != 0)
return peer_ip_;
return const_cast<aio_stream*> (this)->get_ip(ptr,
const_cast<aio_stream*>(this)->peer_ip_, sizeof(peer_ip_));
}
const char* socket_stream::get_peer_ip() const
{
if (stream_ == NULL)
return dummy_;
if (peer_ip_[0] != 0)
return peer_ip_;
char* ptr = ACL_VSTREAM_PEER(stream_);
if (ptr == NULL || *ptr == 0)
{
char buf[64];
if (acl_getpeername(ACL_VSTREAM_SOCK(stream_),
buf, sizeof(buf)) == -1)
{
return dummy_;
}
acl_vstream_set_peer(stream_, buf);
}
return const_cast<socket_stream*> (this)->get_ip(
ACL_VSTREAM_PEER(stream_),
const_cast<socket_stream*> (this)->peer_ip_,
sizeof(peer_ip_));
}
ACL_SOCKET acl_inet_accept_ex(ACL_SOCKET listen_fd, char *ipbuf, size_t size)
{
struct sockaddr_in client_addr;
socklen_t addr_len;
ACL_SOCKET fd;
memset(&client_addr, 0, sizeof(client_addr));
addr_len = sizeof(client_addr);
/* when client_addr not null and protocol is AF_INET, acl_sane_accept
* will set nodelay on the accepted socket, 2008.9.4, zsx
*/
fd = acl_sane_accept(listen_fd, (struct sockaddr *)&client_addr, &addr_len);
if (fd == ACL_SOCKET_INVALID)
return fd;
if (ipbuf != NULL && size > 0 && acl_getpeername(fd, ipbuf, size) < 0)
ipbuf[0] = 0;
return fd;
}
const char* socket_stream::get_peer(bool full /* = false */) const
{
if (stream_ == NULL)
return dummy_;
// xxx: acl_vstream 中没有对此地址赋值
char* ptr = ACL_VSTREAM_PEER(stream_);
if (ptr == NULL || *ptr == 0)
{
char buf[64];
if (acl_getpeername(ACL_VSTREAM_SOCK(stream_),
buf, sizeof(buf)) == -1)
{
return dummy_;
}
acl_vstream_set_peer(stream_, buf);
}
if (full)
return ACL_VSTREAM_PEER(stream_);
else
return get_peer_ip();
}
ACL_VSTREAM *acl_vstream_accept_ex(ACL_VSTREAM *listen_stream,
ACL_VSTREAM *client_stream, char *addr, int size)
{
const char *myname = "acl_vstream_accept_ex";
ACL_SOCKET connfd = ACL_SOCKET_INVALID;
ACL_SOCKET servfd = ACL_VSTREAM_SOCK(listen_stream);
char buf[256];
if ((listen_stream->type & ACL_VSTREAM_TYPE_LISTEN_INET)) {
#ifdef WIN32
if (!(listen_stream->type & ACL_VSTREAM_TYPE_LISTEN_IOCP))
connfd = acl_inet_accept_ex(servfd, buf, sizeof(buf));
else if (listen_stream->iocp_sock == ACL_SOCKET_INVALID)
return NULL;
else {
int ret;
connfd = listen_stream->iocp_sock;
listen_stream->iocp_sock = ACL_SOCKET_INVALID;
/* iocp 方式下，需调用下面过程以允许调用
* getpeername/getsockname
*/
ret = setsockopt(connfd, SOL_SOCKET,
SO_UPDATE_ACCEPT_CONTEXT,
(char *)&servfd, sizeof(servfd));
buf[0] = 0;
if (ret != SOCKET_ERROR)
acl_getpeername(connfd, buf, sizeof(buf));
}
#else
connfd = acl_inet_accept_ex(servfd, buf, sizeof(buf));
#endif
if (connfd != ACL_SOCKET_INVALID && addr != NULL && size > 0)
ACL_SAFE_STRNCPY(addr, buf, size);
#ifdef ACL_UNIX
} else if ((listen_stream->type & ACL_VSTREAM_TYPE_LISTEN_UNIX)) {
connfd = acl_unix_accept(servfd);
if (acl_getpeername(connfd, buf, sizeof(buf)) < 0)
buf[0] = 0;
if (addr)
ACL_SAFE_STRNCPY(addr, buf, size);
#endif
} else
acl_msg_fatal("%s(%d)->%s: invalid listen stream(%d)",
__FILE__, __LINE__, myname, listen_stream->flag);
if (connfd == ACL_SOCKET_INVALID)
return NULL;
if (client_stream != NULL) {
acl_vstream_reset(client_stream);
ACL_VSTREAM_SET_SOCK(client_stream, connfd);
} else {
client_stream = acl_vstream_fdopen(connfd,
ACL_VSTREAM_FLAG_RW,
(int) listen_stream->read_buf_len,
listen_stream->rw_timeout,
ACL_VSTREAM_TYPE_SOCK);
/* 让 client_stream 的 context 成员继承 listen_stream 的
* context 成员变量.
*/
client_stream->context = listen_stream->context;
}
if (client_stream == NULL)
return NULL;
acl_vstream_set_peer(client_stream, buf);
return client_stream;
}
