bss_sock.c 7.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288
  1. /* crypto/bio/bss_sock.c */
  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 <errno.h>
  60. #define USE_SOCKETS
  61. #include "cryptlib.h"
  62. #ifndef OPENSSL_NO_SOCK
  63. # include <openssl/bio.h>
  64. # ifdef WATT32
  65. # define sock_write SockWrite /* Watt-32 uses same names */
  66. # define sock_read SockRead
  67. # define sock_puts SockPuts
  68. # endif
  69. static int sock_write(BIO *h, const char *buf, int num);
  70. static int sock_read(BIO *h, char *buf, int size);
  71. static int sock_puts(BIO *h, const char *str);
  72. static long sock_ctrl(BIO *h, int cmd, long arg1, void *arg2);
  73. static int sock_new(BIO *h);
  74. static int sock_free(BIO *data);
  75. int BIO_sock_should_retry(int s);
  76. static BIO_METHOD methods_sockp = {
  77. BIO_TYPE_SOCKET,
  78. "socket",
  79. sock_write,
  80. sock_read,
  81. sock_puts,
  82. NULL, /* sock_gets, */
  83. sock_ctrl,
  84. sock_new,
  85. sock_free,
  86. NULL,
  87. };
  88. BIO_METHOD *BIO_s_socket(void)
  89. {
  90. return (&methods_sockp);
  91. }
  92. BIO *BIO_new_socket(int fd, int close_flag)
  93. {
  94. BIO *ret;
  95. ret = BIO_new(BIO_s_socket());
  96. if (ret == NULL)
  97. return (NULL);
  98. BIO_set_fd(ret, fd, close_flag);
  99. return (ret);
  100. }
  101. static int sock_new(BIO *bi)
  102. {
  103. bi->init = 0;
  104. bi->num = 0;
  105. bi->ptr = NULL;
  106. bi->flags = 0;
  107. return (1);
  108. }
  109. static int sock_free(BIO *a)
  110. {
  111. if (a == NULL)
  112. return (0);
  113. if (a->shutdown) {
  114. if (a->init) {
  115. SHUTDOWN2(a->num);
  116. }
  117. a->init = 0;
  118. a->flags = 0;
  119. }
  120. return (1);
  121. }
  122. static int sock_read(BIO *b, char *out, int outl)
  123. {
  124. int ret = 0;
  125. if (out != NULL) {
  126. clear_socket_error();
  127. ret = readsocket(b->num, out, outl);
  128. BIO_clear_retry_flags(b);
  129. if (ret <= 0) {
  130. if (BIO_sock_should_retry(ret))
  131. BIO_set_retry_read(b);
  132. }
  133. }
  134. return (ret);
  135. }
  136. static int sock_write(BIO *b, const char *in, int inl)
  137. {
  138. int ret;
  139. clear_socket_error();
  140. ret = writesocket(b->num, in, inl);
  141. BIO_clear_retry_flags(b);
  142. if (ret <= 0) {
  143. if (BIO_sock_should_retry(ret))
  144. BIO_set_retry_write(b);
  145. }
  146. return (ret);
  147. }
  148. static long sock_ctrl(BIO *b, int cmd, long num, void *ptr)
  149. {
  150. long ret = 1;
  151. int *ip;
  152. switch (cmd) {
  153. case BIO_C_SET_FD:
  154. sock_free(b);
  155. b->num = *((int *)ptr);
  156. b->shutdown = (int)num;
  157. b->init = 1;
  158. break;
  159. case BIO_C_GET_FD:
  160. if (b->init) {
  161. ip = (int *)ptr;
  162. if (ip != NULL)
  163. *ip = b->num;
  164. ret = b->num;
  165. } else
  166. ret = -1;
  167. break;
  168. case BIO_CTRL_GET_CLOSE:
  169. ret = b->shutdown;
  170. break;
  171. case BIO_CTRL_SET_CLOSE:
  172. b->shutdown = (int)num;
  173. break;
  174. case BIO_CTRL_DUP:
  175. case BIO_CTRL_FLUSH:
  176. ret = 1;
  177. break;
  178. default:
  179. ret = 0;
  180. break;
  181. }
  182. return (ret);
  183. }
  184. static int sock_puts(BIO *bp, const char *str)
  185. {
  186. int n, ret;
  187. n = strlen(str);
  188. ret = sock_write(bp, str, n);
  189. return (ret);
  190. }
  191. int BIO_sock_should_retry(int i)
  192. {
  193. int err;
  194. if ((i == 0) || (i == -1)) {
  195. err = get_last_socket_error();
  196. # if defined(OPENSSL_SYS_WINDOWS) && 0/* more microsoft stupidity? perhaps
  197. * not? Ben 4/1/99 */
  198. if ((i == -1) && (err == 0))
  199. return (1);
  200. # endif
  201. return (BIO_sock_non_fatal_error(err));
  202. }
  203. return (0);
  204. }
  205. int BIO_sock_non_fatal_error(int err)
  206. {
  207. switch (err) {
  208. # if defined(OPENSSL_SYS_WINDOWS) || defined(OPENSSL_SYS_NETWARE)
  209. # if defined(WSAEWOULDBLOCK)
  210. case WSAEWOULDBLOCK:
  211. # endif
  212. # if 0 /* This appears to always be an error */
  213. # if defined(WSAENOTCONN)
  214. case WSAENOTCONN:
  215. # endif
  216. # endif
  217. # endif
  218. # ifdef EWOULDBLOCK
  219. # ifdef WSAEWOULDBLOCK
  220. # if WSAEWOULDBLOCK != EWOULDBLOCK
  221. case EWOULDBLOCK:
  222. # endif
  223. # else
  224. case EWOULDBLOCK:
  225. # endif
  226. # endif
  227. # if defined(ENOTCONN)
  228. case ENOTCONN:
  229. # endif
  230. # ifdef EINTR
  231. case EINTR:
  232. # endif
  233. # ifdef EAGAIN
  234. # if EWOULDBLOCK != EAGAIN
  235. case EAGAIN:
  236. # endif
  237. # endif
  238. # ifdef EPROTO
  239. case EPROTO:
  240. # endif
  241. # ifdef EINPROGRESS
  242. case EINPROGRESS:
  243. # endif
  244. # ifdef EALREADY
  245. case EALREADY:
  246. # endif
  247. return (1);
  248. /* break; */
  249. default:
  250. break;
  251. }
  252. return (0);
  253. }
  254. #endif /* #ifndef OPENSSL_NO_SOCK */