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
|
//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
//
// copyright : (C) 2014 Eran Ifrah
// file name : clSocketClient.cpp
//
// -------------------------------------------------------------------------
// A
// _____ _ _ _ _
// / __ \ | | | | (_) |
// | / \/ ___ __| | ___| | _| |_ ___
// | | / _ \ / _ |/ _ \ | | | __/ _ )
// | \__/\ (_) | (_| | __/ |___| | || __/
// \____/\___/ \__,_|\___\_____/_|\__\___|
//
// F i l e
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
#include "clConnectionString.h"
#include "clSocketClient.h"
#ifndef _WIN32
#include <arpa/inet.h>
#include <errno.h>
#include <netinet/in.h>
#include <netinet/ip.h> /* superset of previous */
#include <sys/socket.h>
#include <sys/un.h>
#include <unistd.h>
#endif
#ifdef __WXMSW__
#define RESET_ERRNO() WSASetLastError(0)
#else
#define RESET_ERRNO() errno = 0
#endif
clSocketClient::clSocketClient() {}
clSocketClient::~clSocketClient() {}
bool clSocketClient::ConnectLocal(const wxString& socketPath)
{
DestroySocket();
#ifndef __WXMSW__
struct sockaddr_un server;
m_socket = socket(AF_UNIX, SOCK_STREAM, 0);
server.sun_family = AF_UNIX;
strcpy(server.sun_path, socketPath.mb_str(wxConvUTF8).data());
if(::connect(m_socket, (struct sockaddr*)&server, sizeof(struct sockaddr_un)) < 0) { return false; }
return true;
#else
return false;
#endif
}
bool clSocketClient::ConnectRemote(const wxString& address, int port, bool& wouldBlock, bool nonBlockingMode)
{
wouldBlock = false;
DestroySocket();
m_socket = ::socket(AF_INET, SOCK_STREAM, 0);
if(nonBlockingMode) { MakeSocketBlocking(false); }
const char* ip_addr = address.mb_str(wxConvUTF8).data();
struct sockaddr_in serv_addr;
serv_addr.sin_family = AF_INET;
serv_addr.sin_port = htons(port);
#ifndef __WXMSW__
if(inet_pton(AF_INET, ip_addr, &serv_addr.sin_addr) <= 0) {
// restore socket to blocking mode
if(nonBlockingMode) { MakeSocketBlocking(true); }
return false;
}
#else
serv_addr.sin_addr.s_addr = inet_addr(ip_addr);
#endif
RESET_ERRNO();
int rc = ::connect(m_socket, (struct sockaddr*)&serv_addr, sizeof(serv_addr));
#ifdef __WXMSW__
wouldBlock = (WSAGetLastError() == WSAEWOULDBLOCK);
#else
wouldBlock = (errno == EINPROGRESS);
#endif
// restore socket to blocking mode
if(nonBlockingMode) { MakeSocketBlocking(true); }
return rc == 0;
}
bool clSocketClient::Connect(const wxString& connectionString, bool nonBlockingMode)
{
clConnectionString cs(connectionString);
if(!cs.IsOK()) { return false; }
if(cs.GetProtocol() == clConnectionString::kUnixLocalSocket) {
return ConnectLocal(cs.GetPath());
} else {
// TCP
bool wouldBlock = false;
return ConnectRemote(cs.GetHost(), cs.GetPort(), wouldBlock, nonBlockingMode);
}
}
bool clSocketClient::ConnectNonBlocking(const wxString& connectionString, bool& wouldBlock)
{
wouldBlock = false;
clConnectionString cs(connectionString);
if(!cs.IsOK()) { return false; }
if(cs.GetProtocol() == clConnectionString::kUnixLocalSocket) {
return ConnectLocal(cs.GetPath());
} else {
// TCP
return ConnectRemote(cs.GetHost(), cs.GetPort(), wouldBlock, true);
}
}
|
