/
socketUtility.cpp
89 lines (78 loc) · 2.34 KB
/
socketUtility.cpp
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
#include "socketUtility.h"
//creates server socket
int getServerSocket(string ip,string port){
int server_fd;
struct sockaddr_in address;
int opt = 1;
// char buffer[1024] = {0};
string s;
int portno = atoi(port.c_str());
// Creating socket file descriptor
if ((server_fd = socket(AF_INET, SOCK_STREAM, 0)) == 0)
{
perror("socket failed");
exit(EXIT_FAILURE);
}
// Forcefully attaching socket to the port 8080
if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR , &opt, sizeof(opt)))
{
perror("setsockopt");
exit(EXIT_FAILURE);
}
address.sin_family = AF_INET;
// address.sin_addr.s_addr = INADDR_ANY;
address.sin_port = htons( portno );
if(inet_pton(AF_INET, ip.c_str(), &address.sin_addr)<=0)
{
printf("\nInvalid address/ Address not supported \n");
exit(1);
}
bind(server_fd, (struct sockaddr *)&address, sizeof(address));
if (::bind(server_fd, (struct sockaddr *)&address, sizeof(address)) < 0 )
{
perror("bind failed");
exit(EXIT_FAILURE);
}
if (listen(server_fd, 10) < 0)
{
perror("listen");
exit(EXIT_FAILURE);
}
// cout<<address.sin_addr.s_addr;
return server_fd;
}
//creates client socket
int getClientSocket(string ip,string port){
int sock = 0;
int portno = atoi(port.c_str());
struct sockaddr_in serv_addr;
if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
printf("\nClient Socket creation error \n");
return -1;
// thread_exit(NULL);
}
int opt=1;
if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR , &opt, sizeof(opt)))
{
perror("setsockopt");
return -1;
// exit(EXIT_FAILURE);
}
memset(&serv_addr, '0', sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_port = htons(portno);
if(inet_pton(AF_INET, ip.c_str(), &serv_addr.sin_addr)<=0)
{
printf("\nInvalid address/ Address not supported \n");
return -1;
// exit(EXIT_FAILURE);
}
if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0)
{
printf("\nConnection Failed \n");
return -1;
// exit(EXIT_FAILURE);
}
return sock;
}