Socket.h

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 #ifndef SOCKET_H_
00019 #define SOCKET_H_
00020 
00021 #include "lwip/sockets.h"
00022 #include "lwip/netdb.h"
00023 
00024 //DNS
00025 inline struct hostent *gethostbyname(const char *name) {
00026   return lwip_gethostbyname(name);
00027 }
00028 
00029 inline int gethostbyname_r(const char *name, struct hostent *ret, char *buf, size_t buflen, struct hostent **result, int *h_errnop) {
00030   return lwip_gethostbyname_r(name, ret, buf, buflen, result, h_errnop);
00031 }
00032 
00033 class TimeInterval;
00034 
00035 /** Socket file descriptor and select wrapper
00036   */
00037 class Socket {
00038 public:
00039     /** Socket
00040      */
00041     Socket();
00042     
00043     /** Set blocking or non-blocking mode of the socket and a timeout on
00044         blocking socket operations
00045     \param blocking  true for blocking mode, false for non-blocking mode.
00046     \param timeout   timeout in ms [Default: (1500)ms].
00047     */
00048     void set_blocking(bool blocking, unsigned int timeout=1500);
00049     
00050     /** Close the socket file descriptor
00051      */
00052     int close();
00053     
00054     ~Socket();
00055     
00056 protected:
00057     int _sock_fd;
00058     int init_socket(int type);
00059     
00060     int wait_readable(TimeInterval& timeout);
00061     int wait_writable(TimeInterval& timeout);
00062     
00063     bool _blocking;
00064     unsigned int _timeout;
00065     
00066 private:
00067     int select(struct timeval *timeout, bool read, bool write);
00068 };
00069 
00070 /** Time interval class used to specify timeouts
00071  */
00072 class TimeInterval {
00073     friend class Socket;
00074 
00075 public:
00076     /** Time Interval
00077      \param ms time interval expressed in milliseconds
00078       */
00079     TimeInterval(unsigned int ms);
00080     
00081 private:
00082     struct timeval _time;
00083 };
00084 
00085 #endif /* SOCKET_H_ */