001:
002:
003:
004:
005:
006:
007:
008:
009:
010:
011:
012:
013:
014:
015:
016:
017:
018:
019:
020:
021:
022:
023:
024:
025:
026:
027:
028:
029:
030:
031:
032:
033:
034:
035:
036:
037:
038:
039:
040:
041:
042:
043:
044:
045:
046:
047:
048:
049:
050:
051:
052:
053:
054:
055:
056:
057:
058:
059:
060:
061:
062:
063:
064:
065:
066:
067:
068:
069:
070:
071:
072:
073:
074:
075:
076:
077:
078:
079:
080:
081:
082:
083:
084:
085:
086:
087:
088:
089:
090:
091:
092:
093:
094:
095:
096:
097:
098:
099:
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:
221:
222:
223:
224:
225:
226:
227:
228:
229:
230:
231:
232:
233:
234:
235:
236:
237:
238:
239:
240:
241:
242:
243:
244:
245:
246:
247:
248:
249:
250:
251:
252:
253:
254:
255:
256:
257:
258:
259:
260:
261:
262:
263:
264:
265:
266:
267:
268:
269:
270:
271:
272:
273:
274:
275:
276:
277:
278:
279:
280:
281:
282:
283:
284:
285:
286:
287:
288:
289:
290:
291:
292:
293:
294:
295:
296:
297:
298:
299:
300:
301:
302:
303:
304:
305:
306:
307:
308:
309:
310:
311:
312:
313:
314:
315:
316:
317:
318:
319:
320:
321:
322:
323:
324:
325:
326:
327:
328:
329:
330:
331:
332:
333:
334:
335:
336:
337:
338:
339:
340:
341:
342:
343:
344:
345:
346:
347:
348:
349:
350:
351:
352:
353:
354:
355:
356:
357:
358:
359:
360:
361:
362:
363:
364:
365:
366:
367:
368:
369:
370:
371:
372:
373:
374:
375:
376:
377:
378:
379:
380:
381:
382:
383:
384:
385:
386:
387:
388:
389:
390:
391:
392:
393:
394:
395:
396:
397:
398:
399:
400:
401:
402:
403:
404:
405:
406:
407:
408:
409:
410:
411:
412:
413:
414:
415:
416:
417:
418:
419:
420:
421:
422:
423:
424:
425:
426:
427:
428:
429:
430:
431:
432:
433:
434:
435:
436:
437:
438:
439:
440:
441:
442:
443:
444:
445:
446:
447:
448:
449:
450:
451:
452:
453:
454:
455:
456:
457:
458:
459:
460:
461:
462:
463:
464:
465:
466:
467:
468:
469:
470:
471:
472:
473:
474:
475:
476:
477:
478:
479:
480:
481:
482:
483:
484:
485:
486:
487:
488:
489:
490:
491:
492:
493:
494:
495:
496:
497:
498:
499:
500:
501:
502:
503:
504:
505:
506:
507:
508:
509:
510:
511:
512:
513:
514:
515:
516:
517:
518:
519:
520:
521:
522:
523:
524:
525:
526:
527:
528:
529:
530:
531:
532:
533:
534:
535:
536:
537:
538:
539:
540:
541:
542:
543:
544:
545:
546:
547:
548:
549:
550:
551:
552:
553:
554:
555:
556:
557:
558:
559:
560:
561:
562:
563:
564:
565:
566:
567:
568:
569:
570:
571:
572:
573:
574:
575:
576:
577:
578:
579:
580:
581:
582:
583:
584:
585:
586:
587:
588:
589:
590:
591:
592:
593:
594:
595:
596:
597:
598:
599:
600:
601:
602:
603:
604:
605:
606:
607:
608:
609:
610:
611:
612:
613:
614:
615:
616:
617:
618:
619:
620:
621:
622:
623:
624:
625:
626:
627:
628:
629:
630:
631:
632:
633:
634:
635:
636:
637:
638:
639:
640:
641:
642:
643:
644:
645:
646:
647:
648:
649:
650:
651:
652:
653:
654:
655:
656:
657:
658:
659:
660:
661:
662:
663:
664:
665:
666:
667:
668:
669:
670:
671:
672:
673:
674:
675:
676:
677:
678:
679:
680:
681:
682:
683:
684:
685:
686:
687:
688:
689:
690:
691:
692:
693:
694:
695:
696:
697:
698:
699:
700:
701:
702:
703:
704:
705:
706:
707:
708:
709:
710:
711:
712:
713:
714:
715:
716:
717:
718:
719:
720:
721:
722:
723:
724:
725:
726:
727:
728:
729:
730:
731:
732:
733:
734:
735:
736:
737:
738:
739:
740:
741:
742:
743:
744:
745:
746:
747:
748:
749:
750:
751:
752:
753:
754:
755:
756:
757:
758:
759:
760:
761:
762:
763:

/*
* Copyright (c), Zeriph Enterprises
* All rights reserved.
*
* Contributor(s):
* Zechariah Perez, omni (at) zeriph (dot) com
*
* THIS SOFTWARE IS PROVIDED BY ZERIPH AND CONTRIBUTORS "AS IS" AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL ZERIPH AND CONTRIBUTORS BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#if !defined(OMNI_SOCKET_HPP)
#define OMNI_SOCKET_HPP 1
#include <omni/types/net_t.hpp>
#include <omni/net/endpoint_descriptor.hpp>
#if defined(OMNI_SAFE_SOCKET)
#include <omni/sync/basic_lock.hpp>
#endif
// TODO: see https://learn.microsoft.com/en-us/dotnet/api/system.net.sockets.socket?view=net-9.0#methods .. add what makes sense
namespace omni {
namespace net {
class socket
{
public:
socket(omni::net::socket_type type, omni::net::protocol_type protocol);
socket(omni::net::socket_type type, omni::net::protocol_type protocol, omni::net::socket_create_options create_ops);
~socket();
omni::net::address_family address_family() const;
omni::net::socket_error accept(omni::net::endpoint_descriptor& remote_ep);
omni::net::socket_error bind();
omni::net::socket_error bind(uint16_t port);
omni::net::socket_error bind(uint32_t ip, uint16_t port);
omni::net::socket_error bind(const std::string& ip, uint16_t port);
omni::net::socket_error bind(const std::wstring& ip, uint16_t port);
omni::net::socket_error bind(const std::string& ip);
omni::net::socket_error bind(const std::wstring& ip);
uint32_t bound_endpoint() const;
uint16_t bound_port() const;
omni::net::socket_error close();
omni::net::socket_error close(uint16_t timeout);
omni::net::socket_error connect();
omni::net::socket_error connect(uint32_t ip, uint16_t port);
omni::net::socket_error connect(const std::string& ip, uint16_t port);
omni::net::socket_error connect(const std::wstring& ip, uint16_t port);
omni::net::socket_error connect_host(const std::string& host, uint16_t port);
omni::net::socket_error connect_host(const std::wstring& host, uint16_t port);
omni::net::socket_error disconnect(bool reuse);
uint32_t endpoint() const;
omni::net::socket_error get_socket_option(omni::net::socket_option_level op_level, int32_t op_name, int32_t& op_val);
omni::net::socket_error get_socket_option(omni::net::socket_option_level op_level, omni::net::socket_option op_name, int32_t& op_val);
omni::net::socket_error get_socket_option(omni::net::socket_option_level op_level, omni::net::tcp_option op_name, int32_t& op_val);
omni::net::socket_t native_handle() const;
omni::net::socket_error open();
omni::net::socket_error open(omni::net::socket_type type, omni::net::protocol_type protocol);
omni::net::socket_error ioc(uint32_t op_code, omni::net::xfr_t* val);
bool is_bound() const;
bool is_connected() const;
bool is_open() const;
bool is_shutdown() const;
bool is_listening() const;
omni::net::socket_error listen();
omni::net::socket_error listen(int32_t backlog);
omni::net::socket_error last_error() const;
uint16_t port() const;
omni::net::protocol_type protocol() const;
template < typename T >
omni::net::socket_error receive(std::vector<T>& buffer, uint32_t& received)
{
// if CXX buffer.data
return this->_receive(&buffer[0], (buffer.size() * sizeof(T)), omni::net::socket_flags::NONE, received);
}
template < typename T >
omni::net::socket_error receive(std::vector<T>& buffer, omni::net::socket_flags flags, uint32_t& received)
{
// if CXX buffer.data
return this->_receive(&buffer[0], (buffer.size() * sizeof(T)), flags, received);
}
template < std::size_t SZ >
omni::net::socket_error receive(char (&buffer)[SZ], uint32_t& received)
{
return this->_receive(buffer, SZ, omni::net::socket_flags::NONE, received);
}
template < std::size_t SZ >
omni::net::socket_error receive(char (&buffer)[SZ], omni::net::socket_flags flags, uint32_t& received)
{
return this->_receive(buffer, SZ, flags, received);
}
omni::net::socket_error unsafe_receive(char* buffer, uint32_t buffer_size, uint32_t& received)
{
return this->_receive(buffer, buffer_size, omni::net::socket_flags::NONE, received);
}
omni::net::socket_error unsafe_receive(char* buffer, uint32_t buffer_size, omni::net::socket_flags flags, uint32_t& received)
{
return this->_receive(buffer, buffer_size, flags, received);
}
template < typename T >
omni::net::socket_error receive_from(std::vector<T>& buffer, uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(&buffer[0], (buffer.size() * sizeof(T)), omni::net::socket_flags::NONE, fip, fport, received);
}
template < typename T >
omni::net::socket_error receive_from(std::vector<T>& buffer, omni::net::socket_flags flags, uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(&buffer[0], (buffer.size() * sizeof(T)), flags, fip, fport, received);
}
template < typename T >
omni::net::socket_error receive_from(std::vector<T>& buffer, std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(&buffer[0], (buffer.size() * sizeof(T)), omni::net::socket_flags::NONE, from_ip, from_port, received);
}
template < typename T >
omni::net::socket_error receive_from(std::vector<T>& buffer, omni::net::socket_flags flags, std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(&buffer[0], (buffer.size() * sizeof(T)), flags, from_ip, from_port, received);
}
template < std::size_t SZ >
omni::net::socket_error receive_from(char (&buffer)[SZ], uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(buffer, SZ, omni::net::socket_flags::NONE, fip, fport, received);
}
template < std::size_t SZ >
omni::net::socket_error receive_from(char (&buffer)[SZ], omni::net::socket_flags flags, uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(buffer, SZ, flags, fip, fport, received);
}
template < std::size_t SZ >
omni::net::socket_error receive_from(char (&buffer)[SZ], std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(buffer, SZ, omni::net::socket_flags::NONE, from_ip, from_port, received);
}
template < std::size_t SZ >
omni::net::socket_error receive_from(char (&buffer)[SZ], omni::net::socket_flags flags, std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(buffer, SZ, flags, from_ip, from_port, received);
}
omni::net::socket_error unsafe_receive_from(char* buffer, uint32_t buffer_size, uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(buffer, buffer_size, omni::net::socket_flags::NONE, fip, fport, received);
}
omni::net::socket_error unsafe_receive_from(char* buffer, uint32_t buffer_size, omni::net::socket_flags flags, uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(buffer, buffer_size, flags, fip, fport, received);
}
omni::net::socket_error unsafe_receive_from(char* buffer, uint32_t buffer_size, std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(buffer, buffer_size, omni::net::socket_flags::NONE, from_ip, from_port, received);
}
omni::net::socket_error unsafe_receive_from(char* buffer, uint32_t buffer_size, omni::net::socket_flags flags, std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(buffer, buffer_size, flags, from_ip, from_port, received);
}
template < typename T >
omni::net::socket_error send(const std::vector<T>& buffer, uint32_t& sent)
{
return this->_send(&buffer[0], (buffer.size() * sizeof(T)), omni::net::socket_flags::NONE, sent);
}
template < typename T >
omni::net::socket_error send(const std::vector<T>& buffer, omni::net::socket_flags flags, uint32_t& sent)
{
return this->_send(&buffer[0], (buffer.size() * sizeof(T)), flags, sent);
}
template < std::size_t SZ >
omni::net::socket_error send(const char (&buffer)[SZ], uint32_t& sent)
{
return this->_send(buffer, SZ, omni::net::socket_flags::NONE, sent);
}
template < std::size_t SZ >
omni::net::socket_error send(const char (&buffer)[SZ], omni::net::socket_flags flags, uint32_t& sent)
{
return this->_send(buffer, SZ, flags, sent);
}
omni::net::socket_error send(const std::string& buffer, uint32_t& sent)
{
return this->unsafe_send(buffer.data(), buffer.size(), sent);
}
omni::net::socket_error unsafe_send(const char* buffer, std::size_t buffer_size, uint32_t& sent)
{
return this->_send(buffer, buffer_size, omni::net::socket_flags::NONE, sent);
}
omni::net::socket_error unsafe_send(const char* buffer, std::size_t buffer_size, omni::net::socket_flags flags, uint32_t& sent)
{
return this->_send(buffer, buffer_size, flags, sent);
}
template < typename T >
omni::net::socket_error send_to(const std::vector<T>& buffer, uint32_t len, const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, len, omni::net::socket_flags::NONE, ip, port, sent);
}
template < typename T >
omni::net::socket_error send_to(const std::vector<T>& buffer, uint32_t len, omni::net::socket_flags flags, const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, len, flags, ip, port, sent);
}
template < std::size_t SZ >
omni::net::socket_error send_to(const char (&buffer)[SZ], const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, SZ, omni::net::socket_flags::NONE, ip, port, sent);
}
template < std::size_t SZ >
omni::net::socket_error send_to(const char (&buffer)[SZ], omni::net::socket_flags flags, const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, SZ, flags, ip, port, sent);
}
omni::net::socket_error unsafe_send_to(const char* buffer, uint32_t len, const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, len, omni::net::socket_flags::NONE, ip, port, sent);
}
omni::net::socket_error unsafe_send_to(const char* buffer, uint32_t len, omni::net::socket_flags flags, const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, len, flags, ip, port, sent);
}
omni::net::socket_error set_socket_option(omni::net::socket_option_level op_level, int32_t op_name, int32_t op_val);
omni::net::socket_error set_socket_option(omni::net::socket_option_level op_level, omni::net::socket_option op_name, int32_t op_val);
omni::net::socket_error set_socket_option(omni::net::socket_option_level op_level, omni::net::tcp_option op_name, int32_t op_val);
omni::net::socket& set_protocol_type(omni::net::protocol_type protocol);
omni::net::socket& set_socket_type(omni::net::socket_type type);
omni::net::socket_error set_blocking_mode(omni::net::blocking_mode mode);
omni::net::socket_error shutdown(omni::net::socket_shutdown how);
omni::net::socket_type socket_type() const;
void swap(omni::net::socket& other);
omni::string_t to_string_t() const;
std::string to_string() const;
std::wstring to_wstring() const;
operator std::string() const
{
return this->to_string();
}
operator std::wstring() const
{
return this->to_wstring();
}
OMNI_MEMBERS_FW(omni::net::socket) // disposing,name,type(),hash()
OMNI_OSTREAM_FW(omni::net::socket)
private:
omni::net::socket_t m_socket;
omni::net::sockaddr_in_t m_addr;
omni::net::protocol_type m_proto;
omni::net::socket_type m_type;
omni::net::socket_error m_last_err;
uint32_t m_ep4;
uint16_t m_port;
uint16_t m_conops;
#if defined(OMNI_OS_WIN)
omni::net::wsa_info m_wsa;
#endif
#if defined(OMNI_SAFE_SOCKET)
mutable omni::sync::basic_lock m_mtx;
#endif
socket(); // = delete
socket(const omni::net::socket& cp); // = delete
omni::net::socket& operator= (const omni::net::socket& other); // = delete
omni::net::socket_error _close(uint16_t timeout, bool shutdown);
omni::net::socket_error _receive(void* buffer, std::size_t buffer_size, omni::net::socket_flags flags, uint32_t& received);
omni::net::socket_error _receive_from(void* buffer, std::size_t buffer_size, omni::net::socket_flags flags, std::string& from_ip, uint16_t& from_port, uint32_t& received);
omni::net::socket_error _send(const void* buffer, std::size_t buffer_size, omni::net::socket_flags flags, uint32_t& sent);
omni::net::socket_error _send_to(const void* buffer, std::size_t buffer_size, omni::net::socket_flags flags, const std::string& ip, uint16_t port, uint32_t& sent);
};
class socket6
{
public:
socket6(omni::net::socket_type type,
omni::net::protocol_type protocol);
socket6(omni::net::socket_type type,
omni::net::protocol_type protocol,
bool allow_ip4_fallback);
socket6(omni::net::address_family family,
omni::net::socket_type type,
omni::net::protocol_type protocol);
socket6(omni::net::address_family family,
omni::net::socket_type type,
omni::net::protocol_type protocol,
bool allow_ip4_fallback);
socket6(omni::net::address_family family,
omni::net::socket_type type,
omni::net::protocol_type protocol,
omni::net::socket_create_options create_ops);
socket6(omni::net::address_family family,
omni::net::socket_type type,
omni::net::protocol_type protocol,
omni::net::socket_create_options create_ops,
bool allow_ip4_fallback);
~socket6();
omni::net::address_family address_family() const;
omni::net::socket_error accept(omni::net::endpoint_descriptor6& remote_ep);
omni::net::socket_error bind();
omni::net::socket_error bind(uint16_t port);
omni::net::socket_error bind(uint32_t ip, uint16_t port);
omni::net::socket_error bind(const omni::net::ip6_binary_address& ip, uint16_t port);
omni::net::socket_error bind4(const std::string& ip, uint16_t port);
omni::net::socket_error bind4(const std::wstring& ip, uint16_t port);
omni::net::socket_error bind4(const std::string& ip);
omni::net::socket_error bind4(const std::wstring& ip);
omni::net::socket_error bind6(const std::string& ip, uint16_t port);
omni::net::socket_error bind6(const std::wstring& ip, uint16_t port);
omni::net::socket_error bind6(const std::string& ip);
omni::net::socket_error bind6(const std::wstring& ip);
std::string bound_endpoint() const;
uint16_t bound_port() const;
omni::net::socket_error close();
omni::net::socket_error close(uint16_t timeout);
omni::net::socket_error connect();
omni::net::socket_error connect(uint32_t ip, uint16_t port);
omni::net::socket_error connect(const omni::net::ip6_binary_address& ip, uint16_t port);
omni::net::socket_error connect4(const std::string& ip, uint16_t port);
omni::net::socket_error connect4(const std::wstring& ip, uint16_t port);
omni::net::socket_error connect6(const std::string& ip, uint16_t port);
omni::net::socket_error connect6(const std::wstring& ip, uint16_t port);
omni::net::socket_error connect_host(const std::string& host, uint16_t port);
omni::net::socket_error connect_host(const std::wstring& host, uint16_t port);
omni::net::socket_error disconnect(bool reuse);
std::string endpoint() const;
omni::net::socket_error get_bound_endpoint4_binary(uint32_t& ep) const;
omni::net::socket_error get_bound_endpoint6_binary(omni::net::ip6_binary_address& ep) const;
omni::net::socket_error get_endpoint4_binary(uint32_t& ep) const;
omni::net::socket_error get_endpoint6_binary(omni::net::ip6_binary_address& ep) const;
omni::net::socket_error get_socket_option(omni::net::socket_option_level op_level, int32_t op_name, int32_t& op_val);
omni::net::socket_error get_socket_option(omni::net::socket_option_level op_level, omni::net::socket_option op_name, int32_t& op_val);
omni::net::socket_error get_socket_option(omni::net::socket_option_level op_level, omni::net::tcp_option op_name, int32_t& op_val);
omni::net::socket_t native_handle() const;
omni::net::socket_error open();
omni::net::socket_error open(omni::net::socket_type type, omni::net::protocol_type protocol);
omni::net::socket_error open(omni::net::address_family family, omni::net::socket_type type, omni::net::protocol_type protocol);
omni::net::socket_error ioc(uint32_t op_code, omni::net::xfr_t* val);
bool is_ip4() const;
bool is_ip6() const;
bool is_bound() const;
bool is_connected() const;
bool is_open() const;
bool is_shutdown() const;
bool is_listening() const;
omni::net::socket_error listen();
omni::net::socket_error listen(int32_t backlog);
omni::net::socket_error last_error() const;
uint16_t port() const;
omni::net::protocol_type protocol() const;
template < typename T >
omni::net::socket_error receive(std::vector<T>& buffer, uint32_t& received)
{
// if CXX buffer.data
return this->_receive(&buffer[0], (buffer.size() * sizeof(T)), omni::net::socket_flags::NONE, received);
}
template < typename T >
omni::net::socket_error receive(std::vector<T>& buffer, omni::net::socket_flags flags, uint32_t& received)
{
// if CXX buffer.data
return this->_receive(&buffer[0], (buffer.size() * sizeof(T)), flags, received);
}
template < std::size_t SZ >
omni::net::socket_error receive(char (&buffer)[SZ], uint32_t& received)
{
return this->_receive(buffer, SZ, omni::net::socket_flags::NONE, received);
}
template < std::size_t SZ >
omni::net::socket_error receive(char (&buffer)[SZ], omni::net::socket_flags flags, uint32_t& received)
{
return this->_receive(buffer, SZ, flags, received);
}
omni::net::socket_error unsafe_receive(char* buffer, uint32_t buffer_size, uint32_t& received)
{
return this->_receive(buffer, buffer_size, omni::net::socket_flags::NONE, received);
}
omni::net::socket_error unsafe_receive(char* buffer, uint32_t buffer_size, omni::net::socket_flags flags, uint32_t& received)
{
return this->_receive(buffer, buffer_size, flags, received);
}
template < typename T >
omni::net::socket_error receive_from(std::vector<T>& buffer, uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(&buffer[0], (buffer.size() * sizeof(T)), omni::net::socket_flags::NONE, fip, fport, received);
}
template < typename T >
omni::net::socket_error receive_from(std::vector<T>& buffer, omni::net::socket_flags flags, uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(&buffer[0], (buffer.size() * sizeof(T)), flags, fip, fport, received);
}
template < typename T >
omni::net::socket_error receive_from(std::vector<T>& buffer, std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(&buffer[0], (buffer.size() * sizeof(T)), omni::net::socket_flags::NONE, from_ip, from_port, received);
}
template < typename T >
omni::net::socket_error receive_from(std::vector<T>& buffer, omni::net::socket_flags flags, std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(&buffer[0], (buffer.size() * sizeof(T)), flags, from_ip, from_port, received);
}
template < std::size_t SZ >
omni::net::socket_error receive_from(char (&buffer)[SZ], uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(buffer, SZ, omni::net::socket_flags::NONE, fip, fport, received);
}
template < std::size_t SZ >
omni::net::socket_error receive_from(char (&buffer)[SZ], omni::net::socket_flags flags, uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(buffer, SZ, flags, fip, fport, received);
}
template < std::size_t SZ >
omni::net::socket_error receive_from(char (&buffer)[SZ], std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(buffer, SZ, omni::net::socket_flags::NONE, from_ip, from_port, received);
}
template < std::size_t SZ >
omni::net::socket_error receive_from(char (&buffer)[SZ], omni::net::socket_flags flags, std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(buffer, SZ, flags, from_ip, from_port, received);
}
omni::net::socket_error unsafe_receive_from(char* buffer, uint32_t buffer_size, uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(buffer, buffer_size, omni::net::socket_flags::NONE, fip, fport, received);
}
omni::net::socket_error unsafe_receive_from(char* buffer, uint32_t buffer_size, omni::net::socket_flags flags, uint32_t& received)
{
uint16_t fport = 0;
std::string fip;
return this->_receive_from(buffer, buffer_size, flags, fip, fport, received);
}
omni::net::socket_error unsafe_receive_from(char* buffer, uint32_t buffer_size, std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(buffer, buffer_size, omni::net::socket_flags::NONE, from_ip, from_port, received);
}
omni::net::socket_error unsafe_receive_from(char* buffer, uint32_t buffer_size, omni::net::socket_flags flags, std::string& from_ip, uint16_t& from_port, uint32_t& received)
{
return this->_receive_from(buffer, buffer_size, flags, from_ip, from_port, received);
}
template < typename T >
omni::net::socket_error send(const std::vector<T>& buffer, uint32_t& sent)
{
return this->_send(&buffer[0], (buffer.size() * sizeof(T)), omni::net::socket_flags::NONE, sent);
}
template < typename T >
omni::net::socket_error send(const std::vector<T>& buffer, omni::net::socket_flags flags, uint32_t& sent)
{
return this->_send(&buffer[0], (buffer.size() * sizeof(T)), flags, sent);
}
template < std::size_t SZ >
omni::net::socket_error send(const char (&buffer)[SZ], uint32_t& sent)
{
return this->_send(buffer, SZ, omni::net::socket_flags::NONE, sent);
}
template < std::size_t SZ >
omni::net::socket_error send(const char (&buffer)[SZ], omni::net::socket_flags flags, uint32_t& sent)
{
return this->_send(buffer, SZ, flags, sent);
}
omni::net::socket_error send(const std::string& buffer, uint32_t& sent)
{
return this->unsafe_send(buffer.data(), buffer.size(), sent);
}
omni::net::socket_error unsafe_send(const char* buffer, std::size_t buffer_size, uint32_t& sent)
{
return this->_send(buffer, buffer_size, omni::net::socket_flags::NONE, sent);
}
omni::net::socket_error unsafe_send(const char* buffer, std::size_t buffer_size, omni::net::socket_flags flags, uint32_t& sent)
{
return this->_send(buffer, buffer_size, flags, sent);
}
template < typename T >
omni::net::socket_error send_to(const std::vector<T>& buffer, uint32_t len, const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, len, omni::net::socket_flags::NONE, ip, port, sent);
}
template < typename T >
omni::net::socket_error send_to(const std::vector<T>& buffer, uint32_t len, omni::net::socket_flags flags, const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, len, flags, ip, port, sent);
}
template < std::size_t SZ >
omni::net::socket_error send_to(const char (&buffer)[SZ], const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, SZ, omni::net::socket_flags::NONE, ip, port, sent);
}
template < std::size_t SZ >
omni::net::socket_error send_to(const char (&buffer)[SZ], omni::net::socket_flags flags, const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, SZ, flags, ip, port, sent);
}
omni::net::socket_error unsafe_send_to(const char* buffer, uint32_t len, const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, len, omni::net::socket_flags::NONE, ip, port, sent);
}
omni::net::socket_error unsafe_send_to(const char* buffer, uint32_t len, omni::net::socket_flags flags, const std::string& ip, uint16_t port, uint32_t& sent)
{
return this->_send_to(buffer, len, flags, ip, port, sent);
}
void set_allow_ip4_fallback(bool set);
omni::net::socket_error set_socket_option(omni::net::socket_option_level op_level, int32_t op_name, int32_t op_val);
omni::net::socket_error set_socket_option(omni::net::socket_option_level op_level, omni::net::socket_option op_name, int32_t op_val);
omni::net::socket_error set_socket_option(omni::net::socket_option_level op_level, omni::net::tcp_option op_name, int32_t op_val);
omni::net::socket6& set_address_family(omni::net::address_family family);
omni::net::socket6& set_protocol_type(omni::net::protocol_type protocol);
omni::net::socket6& set_socket_type(omni::net::socket_type type);
omni::net::socket_error set_blocking_mode(omni::net::blocking_mode mode);
omni::net::socket_error shutdown(omni::net::socket_shutdown how);
omni::net::socket_type socket_type() const;
void swap(omni::net::socket6& other);
omni::string_t to_string_t() const;
std::string to_string() const;
std::wstring to_wstring() const;
operator std::string() const
{
return this->to_string();
}
operator std::wstring() const
{
return this->to_wstring();
}
OMNI_MEMBERS_FW(omni::net::socket6) // disposing,name,type(),hash()
OMNI_OSTREAM_FW(omni::net::socket6)
private:
omni::net::socket_t m_socket;
omni::net::dual_socket_addr_t m_addr;
omni::net::address_family m_family;
omni::net::protocol_type m_proto;
omni::net::socket_type m_type;
omni::net::dual_socket_endpoint_t m_ep;
mutable omni::net::socket_error m_last_err;
uint16_t m_port;
uint16_t m_conops;
#if defined(OMNI_OS_WIN)
omni::net::wsa_info m_wsa;
#endif
#if defined(OMNI_SAFE_SOCKET6)
mutable omni::sync::basic_lock m_mtx;
#endif
socket6(); // = delete
socket6(const omni::net::socket6& cp); // = delete
omni::net::socket6& operator= (const omni::net::socket6& other); // = delete
omni::net::socket_error _bind();
omni::net::socket_error _close(uint16_t timeout, bool shutdown);
omni::net::socket_error _receive(void* buffer, std::size_t buffer_size, omni::net::socket_flags flags, uint32_t& received);
omni::net::socket_error _receive_from(void* buffer, std::size_t buffer_size, omni::net::socket_flags flags, std::string& from_ip, uint16_t& from_port, uint32_t& received);
omni::net::socket_error _send(const void* buffer, std::size_t buffer_size, omni::net::socket_flags flags, uint32_t& sent);
omni::net::socket_error _send_to(const void* buffer, std::size_t buffer_size, omni::net::socket_flags flags, const std::string& ip, uint16_t port, uint32_t& sent);
};
}
namespace ipc {
#if defined(OMNI_ALLOW_UNIX_SOCKET)
class unix_socket
{
public:
unix_socket(omni::net::socket_type type);
unix_socket(omni::net::socket_type type, omni::net::protocol_type protocol);
unix_socket(omni::net::socket_type type, omni::net::protocol_type protocol, omni::net::socket_create_options create_ops);
~unix_socket();
omni::net::address_family address_family() const;
omni::net::socket_error accept(omni::ipc::unix_endpoint_descriptor& remote_ep);
omni::net::socket_error bind();
omni::net::socket_error bind(const std::string& ep);
omni::net::socket_error bind(const std::wstring& ep);
omni::net::socket_error bind(const char* ep);
template < std::size_t SZ >
omni::net::socket_error bind(const char (&ep)[SZ]) { return this->bind(std::string(ep)); }
std::string bound_endpoint() const;
omni::net::socket_error close();
omni::net::socket_error close(uint16_t timeout);
omni::net::socket_error connect();
omni::net::socket_error connect(const std::string& ep);
omni::net::socket_error connect(const std::wstring& ep);
omni::net::socket_error connect(const char* ep);
template < std::size_t SZ >
omni::net::socket_error connect(const char (&ep)[SZ]) { return this->connect(std::string(ep)); }
omni::net::socket_error disconnect(bool reuse);
std::string endpoint() const;
omni::net::socket_error get_socket_option(omni::net::socket_option_level op_level, int32_t op_name, int32_t& op_val);
omni::net::socket_error get_socket_option(omni::net::socket_option_level op_level, omni::net::tcp_option op_name, int32_t& op_val);
omni::net::socket_t native_handle() const;
omni::net::socket_error open();
omni::net::socket_error open(omni::net::socket_type type, omni::net::protocol_type protocol);
omni::net::socket_error ioc(uint32_t op_code, omni::net::xfr_t* val);
bool is_bound() const;
bool is_connected() const;
bool is_open() const;
bool is_shutdown() const;
bool is_listening() const;
omni::net::socket_error listen();
omni::net::socket_error listen(int32_t backlog);
omni::net::socket_error last_error() const;
omni::net::protocol_type protocol() const;
template < typename T >
omni::net::socket_error receive(std::vector<T>& buffer, uint32_t& received)
{
// if CXX buffer.data
return this->_receive(&buffer[0], (buffer.size() * sizeof(T)), omni::net::socket_flags::NONE, received);
}
template < typename T >
omni::net::socket_error receive(std::vector<T>& buffer, omni::net::socket_flags flags, uint32_t& received)
{
// if CXX buffer.data
return this->_receive(&buffer[0], (buffer.size() * sizeof(T)), flags, received);
}
template < std::size_t SZ >
omni::net::socket_error receive(char (&buffer)[SZ], uint32_t& received)
{
return this->_receive(buffer, SZ, omni::net::socket_flags::NONE, received);
}
template < std::size_t SZ >
omni::net::socket_error receive(char (&buffer)[SZ], omni::net::socket_flags flags, uint32_t& received)
{
return this->_receive(buffer, SZ, flags, received);
}
omni::net::socket_error unsafe_receive(char* buffer, uint32_t buffer_size, uint32_t& received)
{
return this->_receive(buffer, buffer_size, omni::net::socket_flags::NONE, received);
}
omni::net::socket_error unsafe_receive(char* buffer, uint32_t buffer_size, omni::net::socket_flags flags, uint32_t& received)
{
return this->_receive(buffer, buffer_size, flags, received);
}
template < typename T >
omni::net::socket_error send(const std::vector<T>& buffer, uint32_t& sent)
{
return this->_send(&buffer[0], (buffer.size() * sizeof(T)), omni::net::socket_flags::NONE, sent);
}
template < typename T >
omni::net::socket_error send(const std::vector<T>& buffer, omni::net::socket_flags flags, uint32_t& sent)
{
return this->_send(&buffer[0], (buffer.size() * sizeof(T)), flags, sent);
}
template < std::size_t SZ >
omni::net::socket_error send(const char (&buffer)[SZ], uint32_t& sent)
{
return this->_send(buffer, SZ, omni::net::socket_flags::NONE, sent);
}
template < std::size_t SZ >
omni::net::socket_error send(const char (&buffer)[SZ], omni::net::socket_flags flags, uint32_t& sent)
{
return this->_send(buffer, SZ, flags, sent);
}
omni::net::socket_error send(const std::string& buffer, uint32_t& sent)
{
return this->unsafe_send(buffer.data(), buffer.size(), sent);
}
omni::net::socket_error unsafe_send(const char* buffer, std::size_t buffer_size, uint32_t& sent)
{
return this->_send(buffer, buffer_size, omni::net::socket_flags::NONE, sent);
}
omni::net::socket_error unsafe_send(const char* buffer, std::size_t buffer_size, omni::net::socket_flags flags, uint32_t& sent)
{
return this->_send(buffer, buffer_size, flags, sent);
}
omni::net::socket_error set_socket_option(omni::net::socket_option_level op_level, int32_t op_name, int32_t op_val);
omni::net::socket_error set_socket_option(omni::net::socket_option_level op_level, omni::net::tcp_option op_name, int32_t op_val);
omni::ipc::unix_socket& set_protocol_type(omni::net::protocol_type protocol);
omni::ipc::unix_socket& set_socket_type(omni::net::socket_type type);
omni::net::socket_error set_blocking_mode(omni::net::blocking_mode mode);
omni::net::socket_error shutdown(omni::net::socket_shutdown how);
omni::net::socket_type socket_type() const;
void swap(omni::ipc::unix_socket& other);
omni::string_t to_string_t() const;
std::string to_string() const;
std::wstring to_wstring() const;
operator std::string() const
{
return this->to_string();
}
operator std::wstring() const
{
return this->to_wstring();
}
OMNI_MEMBERS_FW(omni::ipc::unix_socket) // disposing,name,type(),hash()
OMNI_OSTREAM_FW(omni::ipc::unix_socket)
private:
omni::net::socket_t m_socket;
omni::net::sockaddr_un_t m_addr;
omni::net::protocol_type m_proto;
omni::net::socket_type m_type;
omni::net::socket_error m_last_err;
char m_ep[OMNI_UNIX_SOCKET_MAX_PATH];
uint16_t m_conops;
#if defined(OMNI_OS_WIN)
omni::net::wsa_info m_wsa;
#endif
#if defined(OMNI_SAFE_UNIX_SOCKET)
mutable omni::sync::basic_lock m_mtx;
#endif
unix_socket(); // = delete
unix_socket(const omni::ipc::unix_socket& cp); // = delete
omni::ipc::unix_socket& operator= (const omni::ipc::unix_socket& other); // = delete
omni::net::socket_error _close(uint16_t timeout, bool shutdown);
omni::net::socket_error _receive(void* buffer, std::size_t buffer_size, omni::net::socket_flags flags, uint32_t& received);
omni::net::socket_error _send(const void* buffer, std::size_t buffer_size, omni::net::socket_flags flags, uint32_t& sent);
};
#endif
}
}
namespace std {
inline void swap(omni::net::socket& o1, omni::net::socket& o2)
{
o1.swap(o2);
}
inline void swap(omni::net::socket6& o1, omni::net::socket6& o2)
{
o1.swap(o2);
}
#if defined(OMNI_ALLOW_UNIX_SOCKET)
inline void swap(omni::ipc::unix_socket& o1, omni::ipc::unix_socket& o2)
{
o1.swap(o2);
}
#endif
}
#endif // OMNI_SOCKET_HPP