]>
Commit | Line | Data |
---|---|---|
9a1955a7 | 1 | /* |
406b689d | 2 | * Copyright (c) 2015-2020 Red Hat, Inc. |
9a1955a7 JF |
3 | * |
4 | * All rights reserved. | |
5 | * | |
6 | * Author: Jan Friesse (jfriesse@redhat.com) | |
7 | * | |
8 | * This software licensed under BSD license, the text of which follows: | |
9 | * | |
10 | * Redistribution and use in source and binary forms, with or without | |
11 | * modification, are permitted provided that the following conditions are met: | |
12 | * | |
13 | * - Redistributions of source code must retain the above copyright notice, | |
14 | * this list of conditions and the following disclaimer. | |
15 | * - Redistributions in binary form must reproduce the above copyright notice, | |
16 | * this list of conditions and the following disclaimer in the documentation | |
17 | * and/or other materials provided with the distribution. | |
18 | * - Neither the name of the Red Hat, Inc. nor the names of its | |
19 | * contributors may be used to endorse or promote products derived from this | |
20 | * software without specific prior written permission. | |
21 | * | |
22 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" | |
23 | * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
24 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |
25 | * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE | |
26 | * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR | |
27 | * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF | |
28 | * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS | |
29 | * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN | |
30 | * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) | |
31 | * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF | |
32 | * THE POSSIBILITY OF SUCH DAMAGE. | |
33 | */ | |
34 | ||
313d42d1 | 35 | #include "log.h" |
9a1955a7 JF |
36 | #include "msg.h" |
37 | #include "msgio.h" | |
38 | #include "qnet-config.h" | |
9a1955a7 JF |
39 | #include "qdevice-net-msg-received.h" |
40 | #include "qdevice-net-nss.h" | |
41 | #include "qdevice-net-send.h" | |
42 | #include "qdevice-net-socket.h" | |
43 | ||
44 | /* | |
45 | * -1 means end of connection (EOF) or some other unhandled error. 0 = success | |
46 | */ | |
47 | int | |
48 | qdevice_net_socket_read(struct qdevice_net_instance *instance) | |
49 | { | |
50 | int res; | |
51 | int ret_val; | |
52 | int orig_skipping_msg; | |
53 | ||
54 | orig_skipping_msg = instance->skipping_msg; | |
55 | ||
56 | res = msgio_read(instance->socket, &instance->receive_buffer, | |
57 | &instance->msg_already_received_bytes, &instance->skipping_msg); | |
58 | ||
59 | if (!orig_skipping_msg && instance->skipping_msg) { | |
c8d19612 | 60 | log(LOG_DEBUG, "msgio_read set skipping_msg"); |
9a1955a7 JF |
61 | } |
62 | ||
63 | ret_val = 0; | |
64 | ||
65 | switch (res) { | |
66 | case 0: | |
67 | /* | |
68 | * Partial read | |
69 | */ | |
70 | break; | |
71 | case -1: | |
c8d19612 | 72 | log(LOG_DEBUG, "Server closed connection"); |
9a1955a7 JF |
73 | instance->disconnect_reason = QDEVICE_NET_DISCONNECT_REASON_SERVER_CLOSED_CONNECTION; |
74 | ret_val = -1; | |
75 | break; | |
76 | case -2: | |
c8d19612 | 77 | log(LOG_ERR, "Unhandled error when reading from server. " |
9a1955a7 JF |
78 | "Disconnecting from server"); |
79 | instance->disconnect_reason = QDEVICE_NET_DISCONNECT_REASON_CANT_READ_MESSAGE; | |
80 | ret_val = -1; | |
81 | break; | |
82 | case -3: | |
c8d19612 | 83 | log(LOG_ERR, "Can't store message header from server. " |
9a1955a7 JF |
84 | "Disconnecting from server"); |
85 | instance->disconnect_reason = QDEVICE_NET_DISCONNECT_REASON_CANT_READ_MESSAGE; | |
86 | ret_val = -1; | |
87 | break; | |
88 | case -4: | |
c8d19612 | 89 | log(LOG_ERR, "Can't store message from server. " |
9a1955a7 JF |
90 | "Disconnecting from server"); |
91 | instance->disconnect_reason = QDEVICE_NET_DISCONNECT_REASON_CANT_READ_MESSAGE; | |
92 | ret_val = -1; | |
93 | break; | |
94 | case -5: | |
c8d19612 | 95 | log(LOG_WARNING, "Server sent unsupported msg type %u. " |
9a1955a7 JF |
96 | "Disconnecting from server", msg_get_type(&instance->receive_buffer)); |
97 | instance->disconnect_reason = QDEVICE_NET_DISCONNECT_REASON_UNSUPPORTED_MSG; | |
98 | ret_val = -1; | |
99 | break; | |
100 | case -6: | |
c8d19612 | 101 | log(LOG_WARNING, |
9a1955a7 JF |
102 | "Server wants to send too long message %u bytes. Disconnecting from server", |
103 | msg_get_len(&instance->receive_buffer)); | |
104 | instance->disconnect_reason = QDEVICE_NET_DISCONNECT_REASON_CANT_READ_MESSAGE; | |
105 | ret_val = -1; | |
106 | break; | |
107 | case 1: | |
108 | /* | |
109 | * Full message received / skipped | |
110 | */ | |
111 | if (!instance->skipping_msg) { | |
112 | if (qdevice_net_msg_received(instance) == -1) { | |
113 | ret_val = -1; | |
114 | } | |
115 | } else { | |
c8d19612 | 116 | log(LOG_CRIT, "net_socket_read in skipping msg state"); |
406b689d | 117 | exit(EXIT_FAILURE); |
9a1955a7 JF |
118 | } |
119 | ||
120 | instance->skipping_msg = 0; | |
121 | instance->msg_already_received_bytes = 0; | |
122 | dynar_clean(&instance->receive_buffer); | |
123 | break; | |
124 | default: | |
c8d19612 | 125 | log(LOG_CRIT, "qdevice_net_socket_read unhandled error %d", res); |
406b689d | 126 | exit(EXIT_FAILURE); |
9a1955a7 JF |
127 | break; |
128 | } | |
129 | ||
130 | return (ret_val); | |
131 | } | |
132 | ||
133 | static int | |
134 | qdevice_net_socket_write_finished(struct qdevice_net_instance *instance) | |
135 | { | |
136 | PRFileDesc *new_pr_fd; | |
137 | ||
138 | if (instance->state == QDEVICE_NET_INSTANCE_STATE_WAITING_STARTTLS_BEING_SENT) { | |
139 | /* | |
140 | * StartTLS sent to server. Begin with TLS handshake | |
141 | */ | |
142 | if ((new_pr_fd = nss_sock_start_ssl_as_client(instance->socket, | |
143 | instance->advanced_settings->net_nss_qnetd_cn, | |
144 | qdevice_net_nss_bad_cert_hook, | |
145 | qdevice_net_nss_get_client_auth_data, | |
146 | instance, 0, NULL)) == NULL) { | |
c8d19612 | 147 | log_nss(LOG_ERR, "Can't start TLS"); |
9a1955a7 JF |
148 | instance->disconnect_reason = QDEVICE_NET_DISCONNECT_REASON_CANT_START_TLS; |
149 | return (-1); | |
150 | } | |
151 | ||
152 | /* | |
153 | * And send init msg | |
154 | */ | |
155 | if (qdevice_net_send_init(instance) != 0) { | |
156 | instance->disconnect_reason = | |
157 | QDEVICE_NET_DISCONNECT_REASON_CANT_ALLOCATE_MSG_BUFFER; | |
158 | ||
159 | return (-1); | |
160 | } | |
161 | ||
162 | instance->socket = new_pr_fd; | |
163 | instance->using_tls = 1; | |
164 | } | |
165 | ||
166 | return (0); | |
167 | } | |
168 | ||
169 | int | |
170 | qdevice_net_socket_write(struct qdevice_net_instance *instance) | |
171 | { | |
172 | int res; | |
173 | struct send_buffer_list_entry *send_buffer; | |
174 | enum msg_type sent_msg_type; | |
175 | ||
176 | send_buffer = send_buffer_list_get_active(&instance->send_buffer_list); | |
177 | if (send_buffer == NULL) { | |
c8d19612 | 178 | log(LOG_CRIT, "send_buffer_list_get_active returned NULL"); |
9a1955a7 JF |
179 | instance->disconnect_reason = QDEVICE_NET_DISCONNECT_REASON_CANT_SEND_MESSAGE; |
180 | ||
181 | return (-1); | |
182 | } | |
183 | ||
184 | res = msgio_write(instance->socket, &send_buffer->buffer, | |
185 | &send_buffer->msg_already_sent_bytes); | |
186 | ||
187 | if (res == 1) { | |
188 | sent_msg_type = msg_get_type(&send_buffer->buffer); | |
189 | ||
190 | send_buffer_list_delete(&instance->send_buffer_list, send_buffer); | |
191 | ||
192 | if (sent_msg_type != MSG_TYPE_ECHO_REQUEST) { | |
193 | if (qdevice_net_socket_write_finished(instance) == -1) { | |
194 | return (-1); | |
195 | } | |
196 | } | |
197 | } | |
198 | ||
199 | if (res == -1) { | |
c8d19612 | 200 | log_nss(LOG_CRIT, "PR_Send returned 0"); |
9a1955a7 JF |
201 | instance->disconnect_reason = QDEVICE_NET_DISCONNECT_REASON_SERVER_CLOSED_CONNECTION; |
202 | return (-1); | |
203 | } | |
204 | ||
205 | if (res == -2) { | |
c8d19612 | 206 | log_nss(LOG_ERR, "Unhandled error when sending message to server"); |
9a1955a7 JF |
207 | instance->disconnect_reason = QDEVICE_NET_DISCONNECT_REASON_CANT_SEND_MESSAGE; |
208 | ||
209 | return (-1); | |
210 | } | |
211 | ||
212 | return (0); | |
213 | } |