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 <openssl/bio.h> |
59 | |
60 | #if !defined(OPENSSL_TRUSTY) |
61 | |
62 | #include <fcntl.h> |
63 | #include <string.h> |
64 | |
65 | #if !defined(OPENSSL_WINDOWS) |
66 | #include <unistd.h> |
67 | #else |
68 | OPENSSL_MSVC_PRAGMA(warning(push, 3)) |
69 | #include <winsock2.h> |
70 | OPENSSL_MSVC_PRAGMA(warning(pop)) |
71 | |
72 | OPENSSL_MSVC_PRAGMA(comment(lib, "Ws2_32.lib" )) |
73 | #endif |
74 | |
75 | #include "internal.h" |
76 | |
77 | |
78 | #if !defined(OPENSSL_WINDOWS) |
79 | static int closesocket(int sock) { |
80 | return close(sock); |
81 | } |
82 | #endif |
83 | |
84 | static int sock_new(BIO *bio) { |
85 | bio->init = 0; |
86 | bio->num = 0; |
87 | bio->ptr = NULL; |
88 | bio->flags = 0; |
89 | return 1; |
90 | } |
91 | |
92 | static int sock_free(BIO *bio) { |
93 | if (bio == NULL) { |
94 | return 0; |
95 | } |
96 | |
97 | if (bio->shutdown) { |
98 | if (bio->init) { |
99 | closesocket(bio->num); |
100 | } |
101 | bio->init = 0; |
102 | bio->flags = 0; |
103 | } |
104 | return 1; |
105 | } |
106 | |
107 | static int sock_read(BIO *b, char *out, int outl) { |
108 | int ret = 0; |
109 | |
110 | if (out == NULL) { |
111 | return 0; |
112 | } |
113 | |
114 | bio_clear_socket_error(); |
115 | #if defined(OPENSSL_WINDOWS) |
116 | ret = recv(b->num, out, outl, 0); |
117 | #else |
118 | ret = read(b->num, out, outl); |
119 | #endif |
120 | BIO_clear_retry_flags(b); |
121 | if (ret <= 0) { |
122 | if (bio_fd_should_retry(ret)) { |
123 | BIO_set_retry_read(b); |
124 | } |
125 | } |
126 | return ret; |
127 | } |
128 | |
129 | static int sock_write(BIO *b, const char *in, int inl) { |
130 | int ret; |
131 | |
132 | bio_clear_socket_error(); |
133 | #if defined(OPENSSL_WINDOWS) |
134 | ret = send(b->num, in, inl, 0); |
135 | #else |
136 | ret = write(b->num, in, inl); |
137 | #endif |
138 | BIO_clear_retry_flags(b); |
139 | if (ret <= 0) { |
140 | if (bio_fd_should_retry(ret)) { |
141 | BIO_set_retry_write(b); |
142 | } |
143 | } |
144 | return ret; |
145 | } |
146 | |
147 | static long sock_ctrl(BIO *b, int cmd, long num, void *ptr) { |
148 | long ret = 1; |
149 | int *ip; |
150 | |
151 | switch (cmd) { |
152 | case BIO_C_SET_FD: |
153 | sock_free(b); |
154 | b->num = *((int *)ptr); |
155 | b->shutdown = (int)num; |
156 | b->init = 1; |
157 | break; |
158 | case BIO_C_GET_FD: |
159 | if (b->init) { |
160 | ip = (int *)ptr; |
161 | if (ip != NULL) { |
162 | *ip = b->num; |
163 | } |
164 | ret = b->num; |
165 | } else { |
166 | ret = -1; |
167 | } |
168 | break; |
169 | case BIO_CTRL_GET_CLOSE: |
170 | ret = b->shutdown; |
171 | break; |
172 | case BIO_CTRL_SET_CLOSE: |
173 | b->shutdown = (int)num; |
174 | break; |
175 | case BIO_CTRL_FLUSH: |
176 | ret = 1; |
177 | break; |
178 | default: |
179 | ret = 0; |
180 | break; |
181 | } |
182 | return ret; |
183 | } |
184 | |
185 | static const BIO_METHOD methods_sockp = { |
186 | BIO_TYPE_SOCKET, "socket" , |
187 | sock_write, sock_read, |
188 | NULL /* puts */, NULL /* gets, */, |
189 | sock_ctrl, sock_new, |
190 | sock_free, NULL /* callback_ctrl */, |
191 | }; |
192 | |
193 | const BIO_METHOD *BIO_s_socket(void) { return &methods_sockp; } |
194 | |
195 | BIO *BIO_new_socket(int fd, int close_flag) { |
196 | BIO *ret; |
197 | |
198 | ret = BIO_new(BIO_s_socket()); |
199 | if (ret == NULL) { |
200 | return NULL; |
201 | } |
202 | BIO_set_fd(ret, fd, close_flag); |
203 | return ret; |
204 | } |
205 | |
206 | #endif // OPENSSL_TRUSTY |
207 | |