Important changes to repositories hosted on mbed.com
Mbed hosted mercurial repositories are deprecated and are due to be permanently deleted in July 2026.
To keep a copy of this software download the repository Zip archive or clone locally using Mercurial.
It is also possible to export all your personal repositories from the account settings page.
Dependencies: mbed myUSBHost AvailableMemory
Dependents: mbed_TANK_Kinect myBlueUSB_ros ftusbClass
Socket.cpp@7:99068afea04d, 2011-06-19 (annotated)
- Committer:
- networker
- Date:
- Sun Jun 19 12:43:19 2011 +0000
- Revision:
- 7:99068afea04d
Who changed what in which revision?
| User | Revision | Line number | New contents of line | 
|---|---|---|---|
| networker | 7:99068afea04d | 1 | /* | 
| networker | 7:99068afea04d | 2 | Copyright (c) 2010 Peter Barrett | 
| networker | 7:99068afea04d | 3 | |
| networker | 7:99068afea04d | 4 | Permission is hereby granted, free of charge, to any person obtaining a copy | 
| networker | 7:99068afea04d | 5 | of this software and associated documentation files (the "Software"), to deal | 
| networker | 7:99068afea04d | 6 | in the Software without restriction, including without limitation the rights | 
| networker | 7:99068afea04d | 7 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | 
| networker | 7:99068afea04d | 8 | copies of the Software, and to permit persons to whom the Software is | 
| networker | 7:99068afea04d | 9 | furnished to do so, subject to the following conditions: | 
| networker | 7:99068afea04d | 10 | |
| networker | 7:99068afea04d | 11 | The above copyright notice and this permission notice shall be included in | 
| networker | 7:99068afea04d | 12 | all copies or substantial portions of the Software. | 
| networker | 7:99068afea04d | 13 | |
| networker | 7:99068afea04d | 14 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | 
| networker | 7:99068afea04d | 15 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | 
| networker | 7:99068afea04d | 16 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | 
| networker | 7:99068afea04d | 17 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | 
| networker | 7:99068afea04d | 18 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | 
| networker | 7:99068afea04d | 19 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | 
| networker | 7:99068afea04d | 20 | THE SOFTWARE. | 
| networker | 7:99068afea04d | 21 | */ | 
| networker | 7:99068afea04d | 22 | |
| networker | 7:99068afea04d | 23 | #include <stdio.h> | 
| networker | 7:99068afea04d | 24 | #include <stdlib.h> | 
| networker | 7:99068afea04d | 25 | #include <stdio.h> | 
| networker | 7:99068afea04d | 26 | #include <string.h> | 
| networker | 7:99068afea04d | 27 | |
| networker | 7:99068afea04d | 28 | #include "Utils.h" | 
| networker | 7:99068afea04d | 29 | #include "Socket.h" | 
| networker | 7:99068afea04d | 30 | |
| networker | 7:99068afea04d | 31 | #define MAX_SOCKET_HANDLERS 4 | 
| networker | 7:99068afea04d | 32 | #define MAX_SOCKETS 16 | 
| networker | 7:99068afea04d | 33 | |
| networker | 7:99068afea04d | 34 | class SocketInternalPad | 
| networker | 7:99068afea04d | 35 | { | 
| networker | 7:99068afea04d | 36 | public: | 
| networker | 7:99068afea04d | 37 | SocketInternal si; | 
| networker | 7:99068afea04d | 38 | u8 pad[8]; | 
| networker | 7:99068afea04d | 39 | }; | 
| networker | 7:99068afea04d | 40 | |
| networker | 7:99068afea04d | 41 | class SocketManager | 
| networker | 7:99068afea04d | 42 | { | 
| networker | 7:99068afea04d | 43 | SocketHandler* _handlers[MAX_SOCKET_HANDLERS]; | 
| networker | 7:99068afea04d | 44 | SocketInternalPad _sockets[MAX_SOCKETS]; | 
| networker | 7:99068afea04d | 45 | |
| networker | 7:99068afea04d | 46 | public: | 
| networker | 7:99068afea04d | 47 | SocketManager() | 
| networker | 7:99068afea04d | 48 | { | 
| networker | 7:99068afea04d | 49 | memset(_handlers,0,sizeof(_handlers)); | 
| networker | 7:99068afea04d | 50 | memset(_sockets,0,sizeof(_sockets)); | 
| networker | 7:99068afea04d | 51 | } | 
| networker | 7:99068afea04d | 52 | |
| networker | 7:99068afea04d | 53 | SocketHandler* GetHandler(int type) | 
| networker | 7:99068afea04d | 54 | { | 
| networker | 7:99068afea04d | 55 | if (type < 1 || type > MAX_SOCKET_HANDLERS) | 
| networker | 7:99068afea04d | 56 | return 0; | 
| networker | 7:99068afea04d | 57 | return _handlers[type - 1]; | 
| networker | 7:99068afea04d | 58 | } | 
| networker | 7:99068afea04d | 59 | |
| networker | 7:99068afea04d | 60 | SocketInternal* GetInternal(int s) | 
| networker | 7:99068afea04d | 61 | { | 
| networker | 7:99068afea04d | 62 | if (s < 1 || s > MAX_SOCKETS) | 
| networker | 7:99068afea04d | 63 | return 0; | 
| networker | 7:99068afea04d | 64 | return &_sockets[s - 1].si; | 
| networker | 7:99068afea04d | 65 | } | 
| networker | 7:99068afea04d | 66 | |
| networker | 7:99068afea04d | 67 | int RegisterSocketHandler(int type, SocketHandler* handler) | 
| networker | 7:99068afea04d | 68 | { | 
| networker | 7:99068afea04d | 69 | if (type < 1 || type > MAX_SOCKET_HANDLERS) | 
| networker | 7:99068afea04d | 70 | return ERR_SOCKET_TYPE_NOT_FOUND; | 
| networker | 7:99068afea04d | 71 | _handlers[type - 1] = handler; | 
| networker | 7:99068afea04d | 72 | return 0; | 
| networker | 7:99068afea04d | 73 | } | 
| networker | 7:99068afea04d | 74 | |
| networker | 7:99068afea04d | 75 | int Create(int type, SocketCallback callback, void* userData) | 
| networker | 7:99068afea04d | 76 | { | 
| networker | 7:99068afea04d | 77 | SocketHandler* h = GetHandler(type); | 
| networker | 7:99068afea04d | 78 | if (!h) | 
| networker | 7:99068afea04d | 79 | return ERR_SOCKET_TYPE_NOT_FOUND; | 
| networker | 7:99068afea04d | 80 | |
| networker | 7:99068afea04d | 81 | for (int i = 0; i < MAX_SOCKETS; i++) | 
| networker | 7:99068afea04d | 82 | { | 
| networker | 7:99068afea04d | 83 | SocketInternal* si = (SocketInternal*)(_sockets+i); | 
| networker | 7:99068afea04d | 84 | if (si->ID == 0) | 
| networker | 7:99068afea04d | 85 | { | 
| networker | 7:99068afea04d | 86 | si->ID = i+1; | 
| networker | 7:99068afea04d | 87 | si->Type = type; | 
| networker | 7:99068afea04d | 88 | si->Callback = callback; | 
| networker | 7:99068afea04d | 89 | si->userData = userData; | 
| networker | 7:99068afea04d | 90 | printf("Creating socket %d for type %d, invoking 'Open' on %p (=%s)\n", si->ID, type, h, h->Name()); | 
| networker | 7:99068afea04d | 91 | return h->Create(si); | 
| networker | 7:99068afea04d | 92 | } | 
| networker | 7:99068afea04d | 93 | } | 
| networker | 7:99068afea04d | 94 | return ERR_SOCKET_NONE_LEFT; | 
| networker | 7:99068afea04d | 95 | } | 
| networker | 7:99068afea04d | 96 | |
| networker | 7:99068afea04d | 97 | int Open(int type, SocketAddrHdr* addr, SocketCallback callback, void* userData) | 
| networker | 7:99068afea04d | 98 | { | 
| networker | 7:99068afea04d | 99 | SocketHandler* h = GetHandler(type); | 
| networker | 7:99068afea04d | 100 | if (!h) | 
| networker | 7:99068afea04d | 101 | return ERR_SOCKET_TYPE_NOT_FOUND; | 
| networker | 7:99068afea04d | 102 | |
| networker | 7:99068afea04d | 103 | for (int i = 0; i < MAX_SOCKETS; i++) | 
| networker | 7:99068afea04d | 104 | { | 
| networker | 7:99068afea04d | 105 | SocketInternal* si = (SocketInternal*)(_sockets+i); | 
| networker | 7:99068afea04d | 106 | if (si->ID == 0) | 
| networker | 7:99068afea04d | 107 | { | 
| networker | 7:99068afea04d | 108 | si->ID = i+1; | 
| networker | 7:99068afea04d | 109 | si->Type = type; | 
| networker | 7:99068afea04d | 110 | si->Callback = callback; | 
| networker | 7:99068afea04d | 111 | si->userData = userData; | 
| networker | 7:99068afea04d | 112 | printf("Opening socket %d for type %d, invoking 'Open' on %p (=%s)\n", si->ID, type, h, h->Name()); | 
| networker | 7:99068afea04d | 113 | return h->Open(si,addr); | 
| networker | 7:99068afea04d | 114 | } | 
| networker | 7:99068afea04d | 115 | } | 
| networker | 7:99068afea04d | 116 | return ERR_SOCKET_NONE_LEFT; | 
| networker | 7:99068afea04d | 117 | } | 
| networker | 7:99068afea04d | 118 | |
| networker | 7:99068afea04d | 119 | int Listen(int type, int channel, SocketCallback callback, void* userData) | 
| networker | 7:99068afea04d | 120 | { | 
| networker | 7:99068afea04d | 121 | SocketHandler* h = GetHandler(type); | 
| networker | 7:99068afea04d | 122 | if (!h) | 
| networker | 7:99068afea04d | 123 | return ERR_SOCKET_TYPE_NOT_FOUND; | 
| networker | 7:99068afea04d | 124 | |
| networker | 7:99068afea04d | 125 | for (int i = 0; i < MAX_SOCKETS; i++) | 
| networker | 7:99068afea04d | 126 | { | 
| networker | 7:99068afea04d | 127 | SocketInternal* si = (SocketInternal*)(_sockets+i); | 
| networker | 7:99068afea04d | 128 | if (si->ID == 0) | 
| networker | 7:99068afea04d | 129 | { | 
| networker | 7:99068afea04d | 130 | si->ID = i+1; | 
| networker | 7:99068afea04d | 131 | si->Type = type; | 
| networker | 7:99068afea04d | 132 | si->Callback = callback; | 
| networker | 7:99068afea04d | 133 | si->userData = userData; | 
| networker | 7:99068afea04d | 134 | printf("Passively opening socket %d for type %d, invoking 'Listen' on %p (=%s)\n", si->ID, type, h, h->Name()); | 
| networker | 7:99068afea04d | 135 | int sn = h->Listen(si, channel); | 
| networker | 7:99068afea04d | 136 | if (sn < 0) | 
| networker | 7:99068afea04d | 137 | si->ID = 0;//free the socket when error | 
| networker | 7:99068afea04d | 138 | return sn; | 
| networker | 7:99068afea04d | 139 | } | 
| networker | 7:99068afea04d | 140 | } | 
| networker | 7:99068afea04d | 141 | return ERR_SOCKET_NONE_LEFT; | 
| networker | 7:99068afea04d | 142 | } | 
| networker | 7:99068afea04d | 143 | |
| networker | 7:99068afea04d | 144 | int Accept(int type, int scid, int rxid, SocketCallback callback, void* userData) | 
| networker | 7:99068afea04d | 145 | { | 
| networker | 7:99068afea04d | 146 | SocketHandler* h = GetHandler(type); | 
| networker | 7:99068afea04d | 147 | if (!h) | 
| networker | 7:99068afea04d | 148 | return ERR_SOCKET_TYPE_NOT_FOUND; | 
| networker | 7:99068afea04d | 149 | |
| networker | 7:99068afea04d | 150 | for (int i = 0; i < MAX_SOCKETS; i++) | 
| networker | 7:99068afea04d | 151 | { | 
| networker | 7:99068afea04d | 152 | SocketInternal* si = (SocketInternal*)(_sockets+i); | 
| networker | 7:99068afea04d | 153 | if (si->ID == 0) | 
| networker | 7:99068afea04d | 154 | { | 
| networker | 7:99068afea04d | 155 | si->ID = i+1; | 
| networker | 7:99068afea04d | 156 | si->Type = type; | 
| networker | 7:99068afea04d | 157 | si->Callback = callback; | 
| networker | 7:99068afea04d | 158 | si->userData = userData; | 
| networker | 7:99068afea04d | 159 | printf("Accepting socket %d for type %d, invoking 'Accept' on %p (=%s)\n", si->ID, type, h, h->Name()); | 
| networker | 7:99068afea04d | 160 | return h->Accept(si, scid, rxid); | 
| networker | 7:99068afea04d | 161 | } | 
| networker | 7:99068afea04d | 162 | } | 
| networker | 7:99068afea04d | 163 | return ERR_SOCKET_NONE_LEFT; | 
| networker | 7:99068afea04d | 164 | } | 
| networker | 7:99068afea04d | 165 | |
| networker | 7:99068afea04d | 166 | int Send(int socket, const u8* data, int len) | 
| networker | 7:99068afea04d | 167 | { | 
| networker | 7:99068afea04d | 168 | SocketInternal* si = GetInternal(socket); | 
| networker | 7:99068afea04d | 169 | if (!si || si->ID != socket) | 
| networker | 7:99068afea04d | 170 | return ERR_SOCKET_NOT_FOUND; | 
| networker | 7:99068afea04d | 171 | // printf("sending %d bytes to socket %d (ID=%d)\n", len, socket, si->ID); | 
| networker | 7:99068afea04d | 172 | return GetHandler(si->Type)->Send(si,data,len); | 
| networker | 7:99068afea04d | 173 | } | 
| networker | 7:99068afea04d | 174 | |
| networker | 7:99068afea04d | 175 | int SetOpt(int socket, int so, int* data, int len) | 
| networker | 7:99068afea04d | 176 | { | 
| networker | 7:99068afea04d | 177 | SocketInternal* si = GetInternal(socket); | 
| networker | 7:99068afea04d | 178 | if (!si || si->ID != socket) | 
| networker | 7:99068afea04d | 179 | return ERR_SOCKET_NOT_FOUND; | 
| networker | 7:99068afea04d | 180 | return GetHandler(si->Type)->SetOpt(si, so, data,len); | 
| networker | 7:99068afea04d | 181 | } | 
| networker | 7:99068afea04d | 182 | |
| networker | 7:99068afea04d | 183 | int GetOpt(int socket, int so, int* data, int len) | 
| networker | 7:99068afea04d | 184 | { | 
| networker | 7:99068afea04d | 185 | SocketInternal* si = GetInternal(socket); | 
| networker | 7:99068afea04d | 186 | if (!si || si->ID != socket) | 
| networker | 7:99068afea04d | 187 | return ERR_SOCKET_NOT_FOUND; | 
| networker | 7:99068afea04d | 188 | return GetHandler(si->Type)->GetOpt(si, so, data,len); | 
| networker | 7:99068afea04d | 189 | } | 
| networker | 7:99068afea04d | 190 | |
| networker | 7:99068afea04d | 191 | int Close(int socket) | 
| networker | 7:99068afea04d | 192 | { | 
| networker | 7:99068afea04d | 193 | SocketInternal* si = GetInternal(socket); | 
| networker | 7:99068afea04d | 194 | if (!si || si->ID != socket){ | 
| networker | 7:99068afea04d | 195 | printf("Close: socket %d not found\n", socket); | 
| networker | 7:99068afea04d | 196 | return ERR_SOCKET_NOT_FOUND; | 
| networker | 7:99068afea04d | 197 | } | 
| networker | 7:99068afea04d | 198 | printf("Close: socket %d (type=%d)\n", socket, si->Type); | 
| networker | 7:99068afea04d | 199 | si->SetState(SocketState_Closing); | 
| networker | 7:99068afea04d | 200 | int retval = GetHandler(si->Type)->Close(si); | 
| networker | 7:99068afea04d | 201 | //si->SetState(Socket_Closed); | 
| networker | 7:99068afea04d | 202 | //si->ID = 0; | 
| networker | 7:99068afea04d | 203 | return retval; | 
| networker | 7:99068afea04d | 204 | } | 
| networker | 7:99068afea04d | 205 | }; | 
| networker | 7:99068afea04d | 206 | |
| networker | 7:99068afea04d | 207 | SocketManager gSocketManager; | 
| networker | 7:99068afea04d | 208 | |
| networker | 7:99068afea04d | 209 | int Socket_Open(int type, SocketAddrHdr* addr, SocketCallback callback, void* userData) | 
| networker | 7:99068afea04d | 210 | { | 
| networker | 7:99068afea04d | 211 | return gSocketManager.Open(type,addr,callback,userData); | 
| networker | 7:99068afea04d | 212 | } | 
| networker | 7:99068afea04d | 213 | |
| networker | 7:99068afea04d | 214 | int Socket_Listen(int type, int channel, SocketCallback callback, void* userData) // Open a socket for listening | 
| networker | 7:99068afea04d | 215 | { | 
| networker | 7:99068afea04d | 216 | return gSocketManager.Listen(type,channel,callback,userData); | 
| networker | 7:99068afea04d | 217 | } | 
| networker | 7:99068afea04d | 218 | |
| networker | 7:99068afea04d | 219 | int Socket_Accept(int type, int scid, int rxid, SocketCallback callback, void* userData) // Open a socket for an incoming connection | 
| networker | 7:99068afea04d | 220 | { | 
| networker | 7:99068afea04d | 221 | return gSocketManager.Accept(type,scid,rxid,callback,userData); | 
| networker | 7:99068afea04d | 222 | } | 
| networker | 7:99068afea04d | 223 | |
| networker | 7:99068afea04d | 224 | int Socket_Send(int socket, const u8* data, int len) | 
| networker | 7:99068afea04d | 225 | { | 
| networker | 7:99068afea04d | 226 | return gSocketManager.Send(socket,data,len); | 
| networker | 7:99068afea04d | 227 | } | 
| networker | 7:99068afea04d | 228 | |
| networker | 7:99068afea04d | 229 | int Socket_SetOpt(int socket, int so, int* data, int len) | 
| networker | 7:99068afea04d | 230 | { | 
| networker | 7:99068afea04d | 231 | return gSocketManager.SetOpt(socket, so, data,len); | 
| networker | 7:99068afea04d | 232 | } | 
| networker | 7:99068afea04d | 233 | |
| networker | 7:99068afea04d | 234 | int Socket_GetOpt(int socket, int so, int* data, int len) | 
| networker | 7:99068afea04d | 235 | { | 
| networker | 7:99068afea04d | 236 | return gSocketManager.GetOpt(socket, so, data,len); | 
| networker | 7:99068afea04d | 237 | } | 
| networker | 7:99068afea04d | 238 | |
| networker | 7:99068afea04d | 239 | int Socket_Close(int socket) | 
| networker | 7:99068afea04d | 240 | { | 
| networker | 7:99068afea04d | 241 | return gSocketManager.Close(socket); | 
| networker | 7:99068afea04d | 242 | } | 
| networker | 7:99068afea04d | 243 | |
| networker | 7:99068afea04d | 244 | int RegisterSocketHandler(int type, SocketHandler* handler) | 
| networker | 7:99068afea04d | 245 | { | 
| networker | 7:99068afea04d | 246 | return gSocketManager.RegisterSocketHandler(type,handler); | 
| networker | 7:99068afea04d | 247 | } | 
| networker | 7:99068afea04d | 248 | |
| networker | 7:99068afea04d | 249 | SocketInternal* GetSocketInternal(int socket) | 
| networker | 7:99068afea04d | 250 | { | 
| networker | 7:99068afea04d | 251 | return gSocketManager.GetInternal(socket); | 
| networker | 7:99068afea04d | 252 | } | 
| networker | 7:99068afea04d | 253 |