bgplg/socket.cpp at master · sem-hub/bgplg · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
#include <iostream>
#include <errno.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <strings.h>
#include <netinet/in.h>
#include "socket.hpp"

using namespace std;

Socket::Socket() : created_(false), type_(0), terminate_(false)
{
}

Socket::Socket(const Socket& s) : s_(s.s_), created_(true), terminate_(false)
{
}

Socket::Socket(int s) : s_(s), created_(true), terminate_(false)
{
}

void
Socket::close()
{
    if(created_)
	::close(s_);
}

void
Socket::create(int type)
{
    type_ = type;
    s_ = socket(AF_INET, type, 0);
    if(s_ < 0)
	throw SocketOsError("socket", errno);
    created_ = true;
}

void
Socket::listen(in_addr_t bindip, in_port_t bindport)
{
    int n = 1;
    struct sockaddr_in  sin;

    if(!created_)
	throw SocketNotCreated();

    if(::setsockopt(s_, SOL_SOCKET, SO_REUSEADDR, &n, sizeof(n)) == -1)
	throw SocketOsError("setsockopt", errno);
    bzero(&sin, sizeof(sin));
    sin.sin_family = AF_INET;
    sin.sin_addr.s_addr = bindip;
    sin.sin_port = bindport;

    if (::bind(s_, (struct sockaddr *)&sin, sizeof(sin)) == -1)
	throw SocketOsError("bind", errno);

    if (::listen(s_, 1024) == -1)
	throw SocketOsError("listen", errno);
}

bool
Socket::is_ready()
{
    fd_set	fdset;
    struct timeval tm = { 0, 100 };

    if(!created_)
	throw SocketNotCreated();

    FD_ZERO(&fdset);
    FD_SET(s_, &fdset);
    if((select(s_ + 1, &fdset, NULL, NULL, &tm)) > 0 && FD_ISSET(s_, &fdset))
	return true;

    return false;
}

Socket*
Socket::accept()
{
    struct sockaddr_in  claddr;
    socklen_t len = sizeof(claddr);
    int newsocket;

    if(!created_)
	throw SocketNotCreated();

    if((newsocket=::accept(s_, (struct sockaddr *)&claddr, &len)) < 0)
	throw SocketOsError("accept", errno);

    return new Socket(newsocket);
}

size_t
Socket::read(char *buf, size_t size)
{
    size_t s = ::recv(s_, buf, size, 0);
    if(s < 0)
	throw SocketOsError("recv", errno);

    return s;
}

void
Socket::read_assure(char *buf, size_t size)
{
    size_t m = 0;
    ssize_t n;

    if(!created_)
	throw SocketNotCreated();

    do {
	n = ::recv(s_, buf+m, size-m, 0);
	if(n < 0 && errno != EAGAIN)
	    throw SocketNotEnoughData();
	m += n;
    } while(m < size && !terminate_);
}

void
Socket::write(const char *buf, size_t size)
{
    if(!created_)
	throw SocketNotCreated();

    if(::write(s_, buf, size) < (ssize_t)size)
	throw SocketOsError("write", errno);
}

in_addr_t
Socket::get_peer()
{
    struct sockaddr_in peername;
    socklen_t len = sizeof(peername);

    if(!created_)
	throw SocketNotCreated();

    if(::getpeername(s_, (struct sockaddr*)&peername, &len) < 0) {
	// XXX cout << "errno=" << errno << endl;
	throw SocketOsError("getpeername", errno);
    }

    return ntohl(peername.sin_addr.s_addr);
}