Socket.cpp 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
  1. /*
  2. * Author: Patrick-Christopher Mattulat
  3. * Company: Lynar Studios
  4. * E-Mail: webmaster@lynarstudios.com
  5. * Created: 2020-12-06
  6. * Changed: 2020-12-06
  7. *
  8. * */
  9. #include <ls_std/network/Socket.hpp>
  10. #if defined(unix) || defined(__APPLE__)
  11. #include <sys/socket.h>
  12. #include <resolv.h>
  13. #include <unistd.h>
  14. #endif
  15. #ifdef _WIN32
  16. #include <http.h>
  17. #endif
  18. ls_std::Socket::Socket() : ls_std::Class("Socket")
  19. {}
  20. bool ls_std::Socket::close() const
  21. {
  22. bool closed;
  23. #if defined(unix) || defined(__APPLE__)
  24. int result = ::close(this->descriptor);
  25. closed = result == 0;
  26. #endif
  27. return closed;
  28. }
  29. bool ls_std::Socket::create(ls_std::AddressFamily _addressFamily, ls_std::SocketType _socketType, ls_std::NetworkProtocol _networkProtocol)
  30. {
  31. bool created;
  32. #if defined(unix) || defined(__APPLE__)
  33. this->descriptor = ls_std::Socket::_create(_addressFamily, _socketType, _networkProtocol);
  34. created = this->descriptor >= 0;
  35. #endif
  36. #ifdef _WIN32
  37. this->descriptor = ls_std::Socket::_create(_addressFamily, _socketType, _networkProtocol);
  38. created = this->descriptor >= 0;
  39. #endif
  40. return created;
  41. }
  42. int ls_std::Socket::_convertAddressFamily(ls_std::AddressFamily _addressFamily)
  43. {
  44. int addressFamily;
  45. switch(_addressFamily)
  46. {
  47. case ls_std::ADDRESS_FAMILY_IP_V4:
  48. {
  49. addressFamily = AF_INET;
  50. } break;
  51. case ls_std::ADDRESS_FAMILY_IP_V6:
  52. {
  53. addressFamily = AF_INET6;
  54. } break;
  55. }
  56. return addressFamily;
  57. }
  58. int ls_std::Socket::_convertNetworkProtocol(ls_std::NetworkProtocol _networkProtocol)
  59. {
  60. int networkProtocol;
  61. switch(_networkProtocol)
  62. {
  63. case ls_std::NETWORK_PROTOCOL_TCP_IP:
  64. {
  65. networkProtocol = IPPROTO_TCP;
  66. } break;
  67. case ls_std::NETWORK_PROTOCOL_UDP_IP:
  68. {
  69. networkProtocol = IPPROTO_UDP;
  70. } break;
  71. }
  72. return networkProtocol;
  73. }
  74. int ls_std::Socket::_convertSocketType(ls_std::SocketType _socketType)
  75. {
  76. int socketType;
  77. switch(_socketType)
  78. {
  79. case ls_std::SOCKET_TYPE_STREAM:
  80. {
  81. socketType = SOCK_STREAM;
  82. } break;
  83. case ls_std::SOCKET_TYPE_DATAGRAM:
  84. {
  85. socketType = SOCK_DGRAM;
  86. } break;
  87. }
  88. return socketType;
  89. }
  90. #if defined(unix) || defined(__APPLE__)
  91. int ls_std::Socket::_create(ls_std::AddressFamily _addressFamily, ls_std::SocketType _socketType, ls_std::NetworkProtocol _networkProtocol)
  92. {
  93. int addressFamily = ls_std::Socket::_convertAddressFamily(_addressFamily);
  94. int socketType = ls_std::Socket::_convertSocketType(_socketType);
  95. int networkProtocol = ls_std::Socket::_convertNetworkProtocol(_networkProtocol);
  96. return socket(addressFamily, socketType, networkProtocol);
  97. }
  98. #endif
  99. #ifdef _WIN32
  100. SOCKET ls_std::Socket::_create(ls_std::AddressFamily _addressFamily, ls_std::SocketType _socketType, ls_std::NetworkProtocol _networkProtocol)
  101. {
  102. int addressFamily = ls_std::Socket::_convertAddressFamily(_addressFamily);
  103. int socketType = ls_std::Socket::_convertSocketType(_socketType);
  104. int networkProtocol = ls_std::Socket::_convertNetworkProtocol(_networkProtocol);
  105. return socket(addressFamily, socketType, networkProtocol);
  106. }
  107. #endif