1 /* Copyright (C) 1995-1998 Eric Young ([email protected])
2 * All rights reserved.
3 *
4 * This package is an SSL implementation written
5 * by Eric Young ([email protected]).
6 * The implementation was written so as to conform with Netscapes SSL.
7 *
8 * This library is free for commercial and non-commercial use as long as
9 * the following conditions are aheared to. The following conditions
10 * apply to all code found in this distribution, be it the RC4, RSA,
11 * lhash, DES, etc., code; not just the SSL code. The SSL documentation
12 * included with this distribution is covered by the same copyright terms
13 * except that the holder is Tim Hudson ([email protected]).
14 *
15 * Copyright remains Eric Young's, and as such any Copyright notices in
16 * the code are not to be removed.
17 * If this package is used in a product, Eric Young should be given attribution
18 * as the author of the parts of the library used.
19 * This can be in the form of a textual message at program startup or
20 * in documentation (online or textual) provided with the package.
21 *
22 * Redistribution and use in source and binary forms, with or without
23 * modification, are permitted provided that the following conditions
24 * are met:
25 * 1. Redistributions of source code must retain the copyright
26 * notice, this list of conditions and the following disclaimer.
27 * 2. Redistributions in binary form must reproduce the above copyright
28 * notice, this list of conditions and the following disclaimer in the
29 * documentation and/or other materials provided with the distribution.
30 * 3. All advertising materials mentioning features or use of this software
31 * must display the following acknowledgement:
32 * "This product includes cryptographic software written by
33 * Eric Young ([email protected])"
34 * The word 'cryptographic' can be left out if the rouines from the library
35 * being used are not cryptographic related :-).
36 * 4. If you include any Windows specific code (or a derivative thereof) from
37 * the apps directory (application code) you must include an acknowledgement:
38 * "This product includes software written by Tim Hudson ([email protected])"
39 *
40 * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
41 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
42 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
43 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
44 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
45 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
46 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
47 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
48 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
49 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
50 * SUCH DAMAGE.
51 *
52 * The licence and distribution terms for any publically available version or
53 * derivative of this code cannot be changed. i.e. this code cannot simply be
54 * copied and put under another distribution licence
55 * [including the GNU Public Licence.] */
56
57 #include <openssl/bio.h>
58
59 #if !defined(OPENSSL_NO_SOCK)
60
61 #include <fcntl.h>
62 #include <string.h>
63
64 #if !defined(OPENSSL_WINDOWS)
65 #include <unistd.h>
66 #else
67 OPENSSL_MSVC_PRAGMA(warning(push, 3))
68 #include <winsock2.h>
OPENSSL_MSVC_PRAGMA(warning (pop))69 OPENSSL_MSVC_PRAGMA(warning(pop))
70
71 OPENSSL_MSVC_PRAGMA(comment(lib, "Ws2_32.lib"))
72 #endif
73
74 #include "internal.h"
75
76
77 #if !defined(OPENSSL_WINDOWS)
78 static int closesocket(int sock) {
79 return close(sock);
80 }
81 #endif
82
sock_free(BIO * bio)83 static int sock_free(BIO *bio) {
84 if (bio->shutdown) {
85 if (bio->init) {
86 closesocket(bio->num);
87 }
88 bio->init = 0;
89 bio->flags = 0;
90 }
91 return 1;
92 }
93
sock_read(BIO * b,char * out,int outl)94 static int sock_read(BIO *b, char *out, int outl) {
95 if (out == NULL) {
96 return 0;
97 }
98
99 bio_clear_socket_error();
100 #if defined(OPENSSL_WINDOWS)
101 int ret = recv(b->num, out, outl, 0);
102 #else
103 int ret = (int)read(b->num, out, outl);
104 #endif
105 BIO_clear_retry_flags(b);
106 if (ret <= 0) {
107 if (bio_socket_should_retry(ret)) {
108 BIO_set_retry_read(b);
109 }
110 }
111 return ret;
112 }
113
sock_write(BIO * b,const char * in,int inl)114 static int sock_write(BIO *b, const char *in, int inl) {
115 bio_clear_socket_error();
116 #if defined(OPENSSL_WINDOWS)
117 int ret = send(b->num, in, inl, 0);
118 #else
119 int ret = (int)write(b->num, in, inl);
120 #endif
121 BIO_clear_retry_flags(b);
122 if (ret <= 0) {
123 if (bio_socket_should_retry(ret)) {
124 BIO_set_retry_write(b);
125 }
126 }
127 return ret;
128 }
129
sock_ctrl(BIO * b,int cmd,long num,void * ptr)130 static long sock_ctrl(BIO *b, int cmd, long num, void *ptr) {
131 long ret = 1;
132 int *ip;
133
134 switch (cmd) {
135 case BIO_C_SET_FD:
136 sock_free(b);
137 b->num = *((int *)ptr);
138 b->shutdown = (int)num;
139 b->init = 1;
140 break;
141 case BIO_C_GET_FD:
142 if (b->init) {
143 ip = (int *)ptr;
144 if (ip != NULL) {
145 *ip = b->num;
146 }
147 ret = b->num;
148 } else {
149 ret = -1;
150 }
151 break;
152 case BIO_CTRL_GET_CLOSE:
153 ret = b->shutdown;
154 break;
155 case BIO_CTRL_SET_CLOSE:
156 b->shutdown = (int)num;
157 break;
158 case BIO_CTRL_FLUSH:
159 ret = 1;
160 break;
161 default:
162 ret = 0;
163 break;
164 }
165 return ret;
166 }
167
168 static const BIO_METHOD methods_sockp = {
169 BIO_TYPE_SOCKET, "socket",
170 sock_write, sock_read,
171 NULL /* puts */, NULL /* gets, */,
172 sock_ctrl, NULL /* create */,
173 sock_free, NULL /* callback_ctrl */,
174 };
175
BIO_s_socket(void)176 const BIO_METHOD *BIO_s_socket(void) { return &methods_sockp; }
177
BIO_new_socket(int fd,int close_flag)178 BIO *BIO_new_socket(int fd, int close_flag) {
179 BIO *ret;
180
181 ret = BIO_new(BIO_s_socket());
182 if (ret == NULL) {
183 return NULL;
184 }
185 BIO_set_fd(ret, fd, close_flag);
186 return ret;
187 }
188
189 #endif // OPENSSL_NO_SOCK
190