Home
About
Blog
Products
Forum
Support
Contact
Sunbelt Computer Software
PL/B Language Development and Support
Home
About
Blog
Products
Forum
Support
Contact
tinyhttps/src/socket.cppm at chore/bump-mcpp-0.0.8 · mcpplibs/tinyhttps · GitHub
Skip to content
Navigation Menu
Sign in
Appearance settings
Platform
AI CODE CREATION
GitHub Copilot
Write better code with AI
GitHub Copilot app
Direct agents from issue to merge
MCP Registry
Integrate external tools
DEVELOPER WORKFLOWS
Actions
Automate any workflow
Codespaces
Instant dev environments
Issues
Plan and track work
Code Review
Manage code changes
Code Quality
Enforce quality at merge
APPLICATION SECURITY
GitHub Advanced Security
Find and fix vulnerabilities
Code security
Secure your code as you build
Secret protection
Stop leaks before they start
EXPLORE
Why GitHub
Documentation
Blog
Changelog
Marketplace
View all features
Solutions
BY COMPANY SIZE
Enterprises
Small and medium teams
Startups
Nonprofits
BY USE CASE
App Modernization
DevSecOps
DevOps
CI/CD
View all use cases
BY INDUSTRY
Healthcare
Financial services
Manufacturing
Government
View all industries
View all solutions
Resources
EXPLORE BY TOPIC
AI
Software Development
DevOps
Security
View all topics
EXPLORE BY TYPE
Customer stories
Events & webinars
Ebooks & reports
Business insights
GitHub Skills
SUPPORT & SERVICES
Documentation
Customer support
Community forum
Trust center
Partners
View all resources
Open Source
COMMUNITY
GitHub Sponsors
Fund open source developers
PROGRAMS
Security Lab
Maintainer Community
GitHub Stars
Archive Program
REPOSITORIES
Topics
Trending
Collections
Enterprise
ENTERPRISE SOLUTIONS
Enterprise platform
AI-powered developer platform
AVAILABLE ADD-ONS
GitHub Advanced Security
Enterprise-grade security features
Copilot for Business
Enterprise-grade AI features
Premium Support
Enterprise-grade 24/7 support
Pricing
Search
/
Sign in
Sign up
Appearance settings
You signed in with another tab or window.
Reload
to refresh your session.
You signed out in another tab or window.
Reload
to refresh your session.
You switched accounts on another tab or window.
Reload
to refresh your session.
Dismiss alert
{{ message }}
Uh oh!
There was an error while loading.
Please reload this page
.
mcpplibs
/
tinyhttps
Public
Notifications
You must be signed in to change notification settings
Fork
2
Star
4
Code
Issues
0
Pull requests
1
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Actions
Projects
Security and quality
Insights
Files
Expand file tree
chore/bump-mcpp-0.0.8
Breadcrumbs
tinyhttps
/
src
/
socket.cppm
Copy path
Blame
More file actions
Blame
More file actions
Latest commit
History
History
History
220 lines (186 loc) · 5.37 KB
chore/bump-mcpp-0.0.8
Breadcrumbs
tinyhttps
/
src
/
socket.cppm
Copy path
Top
File metadata and controls
Code
Blame
220 lines (186 loc) · 5.37 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
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
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
module
;
#
ifdef
_WIN32
#
include
<
winsock2.h
>
#
include
<
ws2tcpip.h
>
#
pragma
comment(lib, "ws2_32.lib")
#
else
#
include
<
sys/types.h
>
#
include
<
sys/socket.h
>
#
include
<
netdb.h
>
#
include
<
unistd.h
>
#
include
<
fcntl.h
>
#
include
<
poll.h
>
#
include
<
cerrno
>
#
endif
export
module
mcpplibs.tinyhttps:socket;
import
std;
namespace
mcpplibs
::tinyhttps {
#
ifdef
_WIN32
using
SocketHandle =
SOCKET
;
constexpr
SocketHandle
INVALID_SOCKET_FD
=
INVALID_SOCKET
;
#
else
using
SocketHandle =
int
;
constexpr
SocketHandle
INVALID_SOCKET_FD
= -
1
;
#
endif
export
class
Socket
{
public:
Socket
() =
default
;
~Socket
() {
close
();
}
//
Non-copyable
Socket
(
const
Socket&) =
delete
;
Socket&
operator
=(
const
Socket&) =
delete
;
//
Move constructor
Socket
(Socket&& other)
noexcept
: fd_(other.fd_) {
other.
fd_
=
INVALID_SOCKET_FD
;
}
//
Move assignment
Socket&
operator
=(Socket&& other)
noexcept
{
if
(
this
!= &other) {
close
();
fd_ = other.
fd_
;
other.
fd_
=
INVALID_SOCKET_FD
;
}
return
*
this
;
}
[[nodiscard]]
bool
is_valid
()
const
{
return
fd_ !=
INVALID_SOCKET_FD
;
}
bool
connect
(
const
char
* host,
int
port,
int
timeoutMs) {
//
Close existing connection if any
if
(
is_valid
()) {
close
();
}
//
Resolve address
struct
addrinfo
hints{};
hints.
ai_family
=
AF_UNSPEC
;
hints.
ai_socktype
=
SOCK_STREAM
;
hints.
ai_protocol
=
IPPROTO_TCP
;
auto
portStr =
std::to_string
(port);
struct
addrinfo
* result =
nullptr
;
int
rc = ::
getaddrinfo
(host, portStr.
c_str
(), &hints, &result);
if
(rc !=
0
|| result ==
nullptr
) {
return
false
;
}
//
Try each address
for
(
auto
* rp = result; rp !=
nullptr
; rp = rp->
ai_next
) {
SocketHandle fd = ::
socket
(rp->
ai_family
, rp->
ai_socktype
, rp->
ai_protocol
);
if
(fd ==
INVALID_SOCKET_FD
) {
continue
;
}
//
Set non-blocking
if
(!
set_non_blocking
(fd,
true
)) {
close_handle
(fd);
continue
;
}
rc = ::
connect
(fd, rp->
ai_addr
,
static_cast
<
int
>(rp->
ai_addrlen
));
bool
connected =
false
;
if
(rc ==
0
) {
connected =
true
;
}
else
{
#
ifdef
_WIN32
if
(
WSAGetLastError
() ==
WSAEWOULDBLOCK
) {
#
else
if
(errno ==
EINPROGRESS
) {
#endif
//
Wait for connection with timeout
if
(
poll_fd
(fd, timeoutMs,
false
)) {
int
err =
0
;
socklen_t
len =
sizeof
(err);
if
(::
getsockopt
(fd,
SOL_SOCKET
,
SO_ERROR
,
reinterpret_cast
<
char
*>(&err), &len) ==
0
&& err ==
0
) {
connected =
true
;
}
}
}
}
if
(connected) {
//
Restore blocking mode
set_non_blocking
(fd,
false
);
fd_ = fd;
::freeaddrinfo
(result);
return
true
;
}
close_handle
(fd);
}
::freeaddrinfo
(result);
return
false
;
}
int
read
(
char
* buf,
int
len) {
if
(!
is_valid
())
return
-
1
;
return
static_cast
<
int
>(::
recv
(fd_, buf, len,
0
));
}
int
write
(
const
char
* buf,
int
len) {
if
(!
is_valid
())
return
-
1
;
return
static_cast
<
int
>(::
send
(fd_, buf, len,
0
));
}
bool
wait_readable
(
int
timeoutMs) {
if
(!
is_valid
())
return
false
;
return
poll_fd
(fd_, timeoutMs,
true
);
}
bool
wait_writable
(
int
timeoutMs) {
if
(!
is_valid
())
return
false
;
return
poll_fd
(fd_, timeoutMs,
false
);
}
[[nodiscard]]
SocketHandle
native_handle
()
const
{
return
fd_;
}
void
close
() {
if
(
is_valid
()) {
close_handle
(fd_);
fd_ =
INVALID_SOCKET_FD
;
}
}
static
void
platform_init
() {
#
ifdef
_WIN32
WSADATA
wsaData;
WSAStartup
(
MAKEWORD
(
2
,
2
), &wsaData);
#
endif
}
static
void
platform_cleanup
() {
#
ifdef
_WIN32
WSACleanup
();
#
endif
}
private:
SocketHandle fd_ =
INVALID_SOCKET_FD
;
static
bool
set_non_blocking
(SocketHandle fd,
bool
nonBlocking) {
#
ifdef
_WIN32
u_long mode = nonBlocking ?
1
:
0
;
return
ioctlsocket
(fd,
FIONBIO
, &mode) ==
0
;
#
else
int
flags = ::
fcntl
(fd,
F_GETFL
,
0
);
if
(flags == -
1
)
return
false
;
if
(nonBlocking) {
flags |=
O_NONBLOCK
;
}
else
{
flags &= ~
O_NONBLOCK
;
}
return
::
fcntl
(fd,
F_SETFL
, flags) ==
0
;
#
endif
}
static
bool
poll_fd
(SocketHandle fd,
int
timeoutMs,
bool
forRead) {
#
ifdef
_WIN32
WSAPOLLFD
pfd{};
pfd.
fd
= fd;
pfd.
events
= forRead ?
POLLIN
:
POLLOUT
;
int
ret =
WSAPoll
(&pfd,
1
, timeoutMs);
return
ret >
0
&& (pfd.
revents
& (pfd.
events
|
POLLERR
|
POLLHUP
));
#
else
struct
pollfd
pfd{};
pfd.
fd
= fd;
pfd.
events
= forRead ?
POLLIN
:
POLLOUT
;
int
ret = ::
poll
(&pfd,
1
, timeoutMs);
return
ret >
0
&& (pfd.
revents
& (pfd.
events
|
POLLERR
|
POLLHUP
));
#
endif
}
static
void
close_handle
(SocketHandle fd) {
#
ifdef
_WIN32
::closesocket
(fd);
#
else
::close
(fd);
#
endif
}
};
}
//
namespace mcpplibs::tinyhttps
You can’t perform that action at this time.