s_socket.c 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615
  1. /* apps/s_socket.c - socket-related functions used by s_client and s_server */
  2. /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
  3. * All rights reserved.
  4. *
  5. * This package is an SSL implementation written
  6. * by Eric Young (eay@cryptsoft.com).
  7. * The implementation was written so as to conform with Netscapes SSL.
  8. *
  9. * This library is free for commercial and non-commercial use as long as
  10. * the following conditions are aheared to. The following conditions
  11. * apply to all code found in this distribution, be it the RC4, RSA,
  12. * lhash, DES, etc., code; not just the SSL code. The SSL documentation
  13. * included with this distribution is covered by the same copyright terms
  14. * except that the holder is Tim Hudson (tjh@cryptsoft.com).
  15. *
  16. * Copyright remains Eric Young's, and as such any Copyright notices in
  17. * the code are not to be removed.
  18. * If this package is used in a product, Eric Young should be given attribution
  19. * as the author of the parts of the library used.
  20. * This can be in the form of a textual message at program startup or
  21. * in documentation (online or textual) provided with the package.
  22. *
  23. * Redistribution and use in source and binary forms, with or without
  24. * modification, are permitted provided that the following conditions
  25. * are met:
  26. * 1. Redistributions of source code must retain the copyright
  27. * notice, this list of conditions and the following disclaimer.
  28. * 2. Redistributions in binary form must reproduce the above copyright
  29. * notice, this list of conditions and the following disclaimer in the
  30. * documentation and/or other materials provided with the distribution.
  31. * 3. All advertising materials mentioning features or use of this software
  32. * must display the following acknowledgement:
  33. * "This product includes cryptographic software written by
  34. * Eric Young (eay@cryptsoft.com)"
  35. * The word 'cryptographic' can be left out if the rouines from the library
  36. * being used are not cryptographic related :-).
  37. * 4. If you include any Windows specific code (or a derivative thereof) from
  38. * the apps directory (application code) you must include an acknowledgement:
  39. * "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
  40. *
  41. * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
  42. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  43. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  44. * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  45. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  46. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  47. * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  48. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  49. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  50. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  51. * SUCH DAMAGE.
  52. *
  53. * The licence and distribution terms for any publically available version or
  54. * derivative of this code cannot be changed. i.e. this code cannot simply be
  55. * copied and put under another distribution licence
  56. * [including the GNU Public Licence.]
  57. */
  58. #include <stdio.h>
  59. #include <stdlib.h>
  60. #include <string.h>
  61. #include <errno.h>
  62. #include <signal.h>
  63. /* With IPv6, it looks like Digital has mixed up the proper order of
  64. recursive header file inclusion, resulting in the compiler complaining
  65. that u_int isn't defined, but only if _POSIX_C_SOURCE is defined, which
  66. is needed to have fileno() declared correctly... So let's define u_int */
  67. #if defined(OPENSSL_SYS_VMS_DECC) && !defined(__U_INT)
  68. #define __U_INT
  69. typedef unsigned int u_int;
  70. #endif
  71. #define USE_SOCKETS
  72. #define NON_MAIN
  73. #include "apps.h"
  74. #undef USE_SOCKETS
  75. #undef NON_MAIN
  76. #include "s_apps.h"
  77. #include <openssl/ssl.h>
  78. #ifdef FLAT_INC
  79. #include "e_os.h"
  80. #else
  81. #include "../e_os.h"
  82. #endif
  83. #ifndef OPENSSL_NO_SOCK
  84. #if defined(OPENSSL_SYS_NETWARE) && defined(NETWARE_BSDSOCK)
  85. #include "netdb.h"
  86. #endif
  87. static struct hostent *GetHostByName(char *name);
  88. #if defined(OPENSSL_SYS_WINDOWS) || (defined(OPENSSL_SYS_NETWARE) && !defined(NETWARE_BSDSOCK))
  89. static void ssl_sock_cleanup(void);
  90. #endif
  91. static int ssl_sock_init(void);
  92. static int init_client_ip(int *sock,unsigned char ip[4], int port, int type);
  93. static int init_server(int *sock, int port, int type);
  94. static int init_server_long(int *sock, int port,char *ip, int type);
  95. static int do_accept(int acc_sock, int *sock, char **host);
  96. static int host_ip(char *str, unsigned char ip[4]);
  97. #ifdef OPENSSL_SYS_WIN16
  98. #define SOCKET_PROTOCOL 0 /* more microsoft stupidity */
  99. #else
  100. #define SOCKET_PROTOCOL IPPROTO_TCP
  101. #endif
  102. #if defined(OPENSSL_SYS_NETWARE) && !defined(NETWARE_BSDSOCK)
  103. static int wsa_init_done=0;
  104. #endif
  105. #ifdef OPENSSL_SYS_WINDOWS
  106. static struct WSAData wsa_state;
  107. static int wsa_init_done=0;
  108. #ifdef OPENSSL_SYS_WIN16
  109. static HWND topWnd=0;
  110. static FARPROC lpTopWndProc=NULL;
  111. static FARPROC lpTopHookProc=NULL;
  112. extern HINSTANCE _hInstance; /* nice global CRT provides */
  113. static LONG FAR PASCAL topHookProc(HWND hwnd, UINT message, WPARAM wParam,
  114. LPARAM lParam)
  115. {
  116. if (hwnd == topWnd)
  117. {
  118. switch(message)
  119. {
  120. case WM_DESTROY:
  121. case WM_CLOSE:
  122. SetWindowLong(topWnd,GWL_WNDPROC,(LONG)lpTopWndProc);
  123. ssl_sock_cleanup();
  124. break;
  125. }
  126. }
  127. return CallWindowProc(lpTopWndProc,hwnd,message,wParam,lParam);
  128. }
  129. static BOOL CALLBACK enumproc(HWND hwnd,LPARAM lParam)
  130. {
  131. topWnd=hwnd;
  132. return(FALSE);
  133. }
  134. #endif /* OPENSSL_SYS_WIN32 */
  135. #endif /* OPENSSL_SYS_WINDOWS */
  136. #ifdef OPENSSL_SYS_WINDOWS
  137. static void ssl_sock_cleanup(void)
  138. {
  139. if (wsa_init_done)
  140. {
  141. wsa_init_done=0;
  142. #ifndef OPENSSL_SYS_WINCE
  143. WSACancelBlockingCall();
  144. #endif
  145. WSACleanup();
  146. }
  147. }
  148. #elif defined(OPENSSL_SYS_NETWARE) && !defined(NETWARE_BSDSOCK)
  149. static void sock_cleanup(void)
  150. {
  151. if (wsa_init_done)
  152. {
  153. wsa_init_done=0;
  154. WSACleanup();
  155. }
  156. }
  157. #endif
  158. static int ssl_sock_init(void)
  159. {
  160. #ifdef WATT32
  161. extern int _watt_do_exit;
  162. _watt_do_exit = 0;
  163. if (sock_init())
  164. return (0);
  165. #elif defined(OPENSSL_SYS_WINDOWS)
  166. if (!wsa_init_done)
  167. {
  168. int err;
  169. #ifdef SIGINT
  170. signal(SIGINT,(void (*)(int))ssl_sock_cleanup);
  171. #endif
  172. wsa_init_done=1;
  173. memset(&wsa_state,0,sizeof(wsa_state));
  174. if (WSAStartup(0x0101,&wsa_state)!=0)
  175. {
  176. err=WSAGetLastError();
  177. BIO_printf(bio_err,"unable to start WINSOCK, error code=%d\n",err);
  178. return(0);
  179. }
  180. #ifdef OPENSSL_SYS_WIN16
  181. EnumTaskWindows(GetCurrentTask(),enumproc,0L);
  182. lpTopWndProc=(FARPROC)GetWindowLong(topWnd,GWL_WNDPROC);
  183. lpTopHookProc=MakeProcInstance((FARPROC)topHookProc,_hInstance);
  184. SetWindowLong(topWnd,GWL_WNDPROC,(LONG)lpTopHookProc);
  185. #endif /* OPENSSL_SYS_WIN16 */
  186. }
  187. #elif defined(OPENSSL_SYS_NETWARE) && !defined(NETWARE_BSDSOCK)
  188. WORD wVerReq;
  189. WSADATA wsaData;
  190. int err;
  191. if (!wsa_init_done)
  192. {
  193. # ifdef SIGINT
  194. signal(SIGINT,(void (*)(int))sock_cleanup);
  195. # endif
  196. wsa_init_done=1;
  197. wVerReq = MAKEWORD( 2, 0 );
  198. err = WSAStartup(wVerReq,&wsaData);
  199. if (err != 0)
  200. {
  201. BIO_printf(bio_err,"unable to start WINSOCK2, error code=%d\n",err);
  202. return(0);
  203. }
  204. }
  205. #endif /* OPENSSL_SYS_WINDOWS */
  206. return(1);
  207. }
  208. int init_client(int *sock, char *host, int port, int type)
  209. {
  210. unsigned char ip[4];
  211. if (!host_ip(host,&(ip[0])))
  212. {
  213. return(0);
  214. }
  215. return(init_client_ip(sock,ip,port,type));
  216. }
  217. static int init_client_ip(int *sock, unsigned char ip[4], int port, int type)
  218. {
  219. unsigned long addr;
  220. struct sockaddr_in them;
  221. int s,i;
  222. if (!ssl_sock_init()) return(0);
  223. memset((char *)&them,0,sizeof(them));
  224. them.sin_family=AF_INET;
  225. them.sin_port=htons((unsigned short)port);
  226. addr=(unsigned long)
  227. ((unsigned long)ip[0]<<24L)|
  228. ((unsigned long)ip[1]<<16L)|
  229. ((unsigned long)ip[2]<< 8L)|
  230. ((unsigned long)ip[3]);
  231. them.sin_addr.s_addr=htonl(addr);
  232. if (type == SOCK_STREAM)
  233. s=socket(AF_INET,SOCK_STREAM,SOCKET_PROTOCOL);
  234. else /* ( type == SOCK_DGRAM) */
  235. s=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP);
  236. if (s == INVALID_SOCKET) { perror("socket"); return(0); }
  237. #if defined(SO_KEEPALIVE) && !defined(OPENSSL_SYS_MPE)
  238. if (type == SOCK_STREAM)
  239. {
  240. i=0;
  241. i=setsockopt(s,SOL_SOCKET,SO_KEEPALIVE,(char *)&i,sizeof(i));
  242. if (i < 0) { perror("keepalive"); return(0); }
  243. }
  244. #endif
  245. if (connect(s,(struct sockaddr *)&them,sizeof(them)) == -1)
  246. { closesocket(s); perror("connect"); return(0); }
  247. *sock=s;
  248. return(1);
  249. }
  250. int do_server(int port, int type, int *ret, int (*cb)(char *hostname, int s, unsigned char *context), unsigned char *context)
  251. {
  252. int sock;
  253. char *name = NULL;
  254. int accept_socket = 0;
  255. int i;
  256. if (!init_server(&accept_socket,port,type)) return(0);
  257. if (ret != NULL)
  258. {
  259. *ret=accept_socket;
  260. /* return(1);*/
  261. }
  262. for (;;)
  263. {
  264. if (type==SOCK_STREAM)
  265. {
  266. if (do_accept(accept_socket,&sock,&name) == 0)
  267. {
  268. SHUTDOWN(accept_socket);
  269. return(0);
  270. }
  271. }
  272. else
  273. sock = accept_socket;
  274. i=(*cb)(name,sock, context);
  275. if (name != NULL) OPENSSL_free(name);
  276. if (type==SOCK_STREAM)
  277. SHUTDOWN2(sock);
  278. if (i < 0)
  279. {
  280. SHUTDOWN2(accept_socket);
  281. return(i);
  282. }
  283. }
  284. }
  285. static int init_server_long(int *sock, int port, char *ip, int type)
  286. {
  287. int ret=0;
  288. struct sockaddr_in server;
  289. int s= -1,i;
  290. if (!ssl_sock_init()) return(0);
  291. memset((char *)&server,0,sizeof(server));
  292. server.sin_family=AF_INET;
  293. server.sin_port=htons((unsigned short)port);
  294. if (ip == NULL)
  295. server.sin_addr.s_addr=INADDR_ANY;
  296. else
  297. /* Added for T3E, address-of fails on bit field (beckman@acl.lanl.gov) */
  298. #ifndef BIT_FIELD_LIMITS
  299. memcpy(&server.sin_addr.s_addr,ip,4);
  300. #else
  301. memcpy(&server.sin_addr,ip,4);
  302. #endif
  303. if (type == SOCK_STREAM)
  304. s=socket(AF_INET,SOCK_STREAM,SOCKET_PROTOCOL);
  305. else /* type == SOCK_DGRAM */
  306. s=socket(AF_INET, SOCK_DGRAM,IPPROTO_UDP);
  307. if (s == INVALID_SOCKET) goto err;
  308. #if defined SOL_SOCKET && defined SO_REUSEADDR
  309. {
  310. int j = 1;
  311. setsockopt(s, SOL_SOCKET, SO_REUSEADDR,
  312. (void *) &j, sizeof j);
  313. }
  314. #endif
  315. if (bind(s,(struct sockaddr *)&server,sizeof(server)) == -1)
  316. {
  317. #ifndef OPENSSL_SYS_WINDOWS
  318. perror("bind");
  319. #endif
  320. goto err;
  321. }
  322. /* Make it 128 for linux */
  323. if (type==SOCK_STREAM && listen(s,128) == -1) goto err;
  324. i=0;
  325. *sock=s;
  326. ret=1;
  327. err:
  328. if ((ret == 0) && (s != -1))
  329. {
  330. SHUTDOWN(s);
  331. }
  332. return(ret);
  333. }
  334. static int init_server(int *sock, int port, int type)
  335. {
  336. return(init_server_long(sock, port, NULL, type));
  337. }
  338. static int do_accept(int acc_sock, int *sock, char **host)
  339. {
  340. int ret,i;
  341. struct hostent *h1,*h2;
  342. static struct sockaddr_in from;
  343. int len;
  344. /* struct linger ling; */
  345. if (!ssl_sock_init()) return(0);
  346. #ifndef OPENSSL_SYS_WINDOWS
  347. redoit:
  348. #endif
  349. memset((char *)&from,0,sizeof(from));
  350. len=sizeof(from);
  351. /* Note: under VMS with SOCKETSHR the fourth parameter is currently
  352. * of type (int *) whereas under other systems it is (void *) if
  353. * you don't have a cast it will choke the compiler: if you do
  354. * have a cast then you can either go for (int *) or (void *).
  355. */
  356. ret=accept(acc_sock,(struct sockaddr *)&from,(void *)&len);
  357. if (ret == INVALID_SOCKET)
  358. {
  359. #if defined(OPENSSL_SYS_WINDOWS) || (defined(OPENSSL_SYS_NETWARE) && !defined(NETWARE_BSDSOCK))
  360. i=WSAGetLastError();
  361. BIO_printf(bio_err,"accept error %d\n",i);
  362. #else
  363. if (errno == EINTR)
  364. {
  365. /*check_timeout(); */
  366. goto redoit;
  367. }
  368. fprintf(stderr,"errno=%d ",errno);
  369. perror("accept");
  370. #endif
  371. return(0);
  372. }
  373. /*
  374. ling.l_onoff=1;
  375. ling.l_linger=0;
  376. i=setsockopt(ret,SOL_SOCKET,SO_LINGER,(char *)&ling,sizeof(ling));
  377. if (i < 0) { perror("linger"); return(0); }
  378. i=0;
  379. i=setsockopt(ret,SOL_SOCKET,SO_KEEPALIVE,(char *)&i,sizeof(i));
  380. if (i < 0) { perror("keepalive"); return(0); }
  381. */
  382. if (host == NULL) goto end;
  383. #ifndef BIT_FIELD_LIMITS
  384. /* I should use WSAAsyncGetHostByName() under windows */
  385. h1=gethostbyaddr((char *)&from.sin_addr.s_addr,
  386. sizeof(from.sin_addr.s_addr),AF_INET);
  387. #else
  388. h1=gethostbyaddr((char *)&from.sin_addr,
  389. sizeof(struct in_addr),AF_INET);
  390. #endif
  391. if (h1 == NULL)
  392. {
  393. BIO_printf(bio_err,"bad gethostbyaddr\n");
  394. *host=NULL;
  395. /* return(0); */
  396. }
  397. else
  398. {
  399. if ((*host=(char *)OPENSSL_malloc(strlen(h1->h_name)+1)) == NULL)
  400. {
  401. perror("OPENSSL_malloc");
  402. return(0);
  403. }
  404. BUF_strlcpy(*host,h1->h_name,strlen(h1->h_name)+1);
  405. h2=GetHostByName(*host);
  406. if (h2 == NULL)
  407. {
  408. BIO_printf(bio_err,"gethostbyname failure\n");
  409. return(0);
  410. }
  411. i=0;
  412. if (h2->h_addrtype != AF_INET)
  413. {
  414. BIO_printf(bio_err,"gethostbyname addr is not AF_INET\n");
  415. return(0);
  416. }
  417. }
  418. end:
  419. *sock=ret;
  420. return(1);
  421. }
  422. int extract_host_port(char *str, char **host_ptr, unsigned char *ip,
  423. short *port_ptr)
  424. {
  425. char *h,*p;
  426. h=str;
  427. p=strchr(str,':');
  428. if (p == NULL)
  429. {
  430. BIO_printf(bio_err,"no port defined\n");
  431. return(0);
  432. }
  433. *(p++)='\0';
  434. if ((ip != NULL) && !host_ip(str,ip))
  435. goto err;
  436. if (host_ptr != NULL) *host_ptr=h;
  437. if (!extract_port(p,port_ptr))
  438. goto err;
  439. return(1);
  440. err:
  441. return(0);
  442. }
  443. static int host_ip(char *str, unsigned char ip[4])
  444. {
  445. unsigned int in[4];
  446. int i;
  447. if (sscanf(str,"%u.%u.%u.%u",&(in[0]),&(in[1]),&(in[2]),&(in[3])) == 4)
  448. {
  449. for (i=0; i<4; i++)
  450. if (in[i] > 255)
  451. {
  452. BIO_printf(bio_err,"invalid IP address\n");
  453. goto err;
  454. }
  455. ip[0]=in[0];
  456. ip[1]=in[1];
  457. ip[2]=in[2];
  458. ip[3]=in[3];
  459. }
  460. else
  461. { /* do a gethostbyname */
  462. struct hostent *he;
  463. if (!ssl_sock_init()) return(0);
  464. he=GetHostByName(str);
  465. if (he == NULL)
  466. {
  467. BIO_printf(bio_err,"gethostbyname failure\n");
  468. goto err;
  469. }
  470. /* cast to short because of win16 winsock definition */
  471. if ((short)he->h_addrtype != AF_INET)
  472. {
  473. BIO_printf(bio_err,"gethostbyname addr is not AF_INET\n");
  474. return(0);
  475. }
  476. ip[0]=he->h_addr_list[0][0];
  477. ip[1]=he->h_addr_list[0][1];
  478. ip[2]=he->h_addr_list[0][2];
  479. ip[3]=he->h_addr_list[0][3];
  480. }
  481. return(1);
  482. err:
  483. return(0);
  484. }
  485. int extract_port(char *str, short *port_ptr)
  486. {
  487. int i;
  488. struct servent *s;
  489. i=atoi(str);
  490. if (i != 0)
  491. *port_ptr=(unsigned short)i;
  492. else
  493. {
  494. s=getservbyname(str,"tcp");
  495. if (s == NULL)
  496. {
  497. BIO_printf(bio_err,"getservbyname failure for %s\n",str);
  498. return(0);
  499. }
  500. *port_ptr=ntohs((unsigned short)s->s_port);
  501. }
  502. return(1);
  503. }
  504. #define GHBN_NUM 4
  505. static struct ghbn_cache_st
  506. {
  507. char name[128];
  508. struct hostent ent;
  509. unsigned long order;
  510. } ghbn_cache[GHBN_NUM];
  511. static unsigned long ghbn_hits=0L;
  512. static unsigned long ghbn_miss=0L;
  513. static struct hostent *GetHostByName(char *name)
  514. {
  515. struct hostent *ret;
  516. int i,lowi=0;
  517. unsigned long low= (unsigned long)-1;
  518. for (i=0; i<GHBN_NUM; i++)
  519. {
  520. if (low > ghbn_cache[i].order)
  521. {
  522. low=ghbn_cache[i].order;
  523. lowi=i;
  524. }
  525. if (ghbn_cache[i].order > 0)
  526. {
  527. if (strncmp(name,ghbn_cache[i].name,128) == 0)
  528. break;
  529. }
  530. }
  531. if (i == GHBN_NUM) /* no hit*/
  532. {
  533. ghbn_miss++;
  534. ret=gethostbyname(name);
  535. if (ret == NULL) return(NULL);
  536. /* else add to cache */
  537. if(strlen(name) < sizeof ghbn_cache[0].name)
  538. {
  539. strcpy(ghbn_cache[lowi].name,name);
  540. memcpy((char *)&(ghbn_cache[lowi].ent),ret,sizeof(struct hostent));
  541. ghbn_cache[lowi].order=ghbn_miss+ghbn_hits;
  542. }
  543. return(ret);
  544. }
  545. else
  546. {
  547. ghbn_hits++;
  548. ret= &(ghbn_cache[i].ent);
  549. ghbn_cache[i].order=ghbn_miss+ghbn_hits;
  550. return(ret);
  551. }
  552. }
  553. #endif