Endpoint.cpp

00001 /* Copyright (C) 2012 mbed.org, MIT License
00002  *
00003  * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
00004  * and associated documentation files (the "Software"), to deal in the Software without restriction,
00005  * including without limitation the rights to use, copy, modify, merge, publish, distribute,
00006  * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
00007  * furnished to do so, subject to the following conditions:
00008  *
00009  * The above copyright notice and this permission notice shall be included in all copies or
00010  * substantial portions of the Software.
00011  *
00012  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
00013  * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
00014  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
00015  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
00016  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
00017  */
00018 #include "Socket/Socket.h"
00019 #include "Socket/Endpoint.h"
00020 #include <cstring>
00021 #include <cstdio>
00022 
00023 Endpoint::Endpoint()  {
00024     reset_address();
00025 }
00026 Endpoint::~Endpoint() {}
00027 
00028 void Endpoint::reset_address(void) {
00029     std::memset(&_remoteHost, 0, sizeof(struct sockaddr_in));
00030     _ipAddress[0] = '\0';
00031 }
00032 
00033 #include "stdio.h"
00034 
00035 int Endpoint::set_address(const char* host, const int port) {
00036     reset_address();
00037 
00038     // IP Address
00039     char address[5];
00040     char *p_address = address;
00041 
00042     // Dot-decimal notation
00043     int result = std::sscanf(host, "%3u.%3u.%3u.%3u",
00044         (unsigned int*)&address[0], (unsigned int*)&address[1],
00045         (unsigned int*)&address[2], (unsigned int*)&address[3]);
00046 
00047     if (result != 4) {
00048         // Resolve address with DNS
00049         struct hostent *host_address = lwip_gethostbyname(host);
00050         if (host_address == NULL)
00051             return -1; //Could not resolve address
00052         p_address = (char*)host_address->h_addr_list[0];
00053     }
00054     std::memcpy((char*)&_remoteHost.sin_addr.s_addr, p_address, 4);
00055 
00056     // Address family
00057     _remoteHost.sin_family = AF_INET;
00058 
00059     // Set port
00060     _remoteHost.sin_port = htons(port);
00061 
00062     return 0;
00063 }
00064 
00065 int Endpoint::set_address(in_addr host) {
00066     reset_address();
00067     inet_ntoa_r(host, _ipAddress, sizeof(_ipAddress));
00068     return 0;
00069 }
00070 
00071 char* Endpoint::get_address() {
00072     if ((_ipAddress[0] == '\0') && (_remoteHost.sin_addr.s_addr != 0))
00073             inet_ntoa_r(_remoteHost.sin_addr, _ipAddress, sizeof(_ipAddress));
00074     return _ipAddress;
00075 }
00076 
00077 int   Endpoint::get_port() {
00078     return ntohs(_remoteHost.sin_port);
00079 }