[21580dd] | 1 | /*
|
---|
[2f19103] | 2 | * Copyright (c) 2015 Jiri Svoboda
|
---|
[21580dd] | 3 | * All rights reserved.
|
---|
| 4 | *
|
---|
| 5 | * Redistribution and use in source and binary forms, with or without
|
---|
| 6 | * modification, are permitted provided that the following conditions
|
---|
| 7 | * are met:
|
---|
| 8 | *
|
---|
| 9 | * - Redistributions of source code must retain the above copyright
|
---|
| 10 | * notice, this list of conditions and the following disclaimer.
|
---|
| 11 | * - Redistributions in binary form must reproduce the above copyright
|
---|
| 12 | * notice, this list of conditions and the following disclaimer in the
|
---|
| 13 | * documentation and/or other materials provided with the distribution.
|
---|
| 14 | * - The name of the author may not be used to endorse or promote products
|
---|
| 15 | * derived from this software without specific prior written permission.
|
---|
| 16 | *
|
---|
| 17 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
|
---|
| 18 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
---|
| 19 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
---|
| 20 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
|
---|
| 21 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
---|
| 22 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
---|
| 23 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
---|
| 24 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
---|
| 25 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
---|
| 26 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
---|
| 27 | */
|
---|
| 28 |
|
---|
| 29 | /** @addtogroup tcp
|
---|
[89e57cee] | 30 | * @{
|
---|
[21580dd] | 31 | */
|
---|
| 32 |
|
---|
[c5808b41] | 33 | /**
|
---|
| 34 | * @file TCP (Transmission Control Protocol) network module
|
---|
[21580dd] | 35 | */
|
---|
| 36 |
|
---|
| 37 | #include <async.h>
|
---|
[fe5e9629] | 38 | #include <bitops.h>
|
---|
[1812a0d] | 39 | #include <byteorder.h>
|
---|
[e98b1d5] | 40 | #include <errno.h>
|
---|
[c76e926] | 41 | #include <inet/inet.h>
|
---|
[c5808b41] | 42 | #include <io/log.h>
|
---|
| 43 | #include <stdio.h>
|
---|
[fab2746] | 44 | #include <stdlib.h>
|
---|
[c5808b41] | 45 | #include <task.h>
|
---|
[21580dd] | 46 |
|
---|
[8218b6b] | 47 | #include "ncsim.h"
|
---|
[762b48a] | 48 | #include "pdu.h"
|
---|
[c5808b41] | 49 | #include "rqueue.h"
|
---|
[779541b] | 50 | #include "service.h"
|
---|
[1812a0d] | 51 | #include "std.h"
|
---|
[014dd57b] | 52 | #include "tcp.h"
|
---|
[c5808b41] | 53 | #include "test.h"
|
---|
[21580dd] | 54 |
|
---|
[c5808b41] | 55 | #define NAME "tcp"
|
---|
[21580dd] | 56 |
|
---|
[c76e926] | 57 | static int tcp_inet_ev_recv(inet_dgram_t *dgram);
|
---|
[1812a0d] | 58 | static void tcp_received_pdu(tcp_pdu_t *pdu);
|
---|
[21580dd] | 59 |
|
---|
[c76e926] | 60 | static inet_ev_ops_t tcp_inet_ev_ops = {
|
---|
| 61 | .recv = tcp_inet_ev_recv
|
---|
| 62 | };
|
---|
[7c8267b] | 63 |
|
---|
[c76e926] | 64 | /** Received datagram callback */
|
---|
| 65 | static int tcp_inet_ev_recv(inet_dgram_t *dgram)
|
---|
[7c8267b] | 66 | {
|
---|
[1812a0d] | 67 | uint8_t *pdu_raw;
|
---|
[c76e926] | 68 | size_t pdu_raw_size;
|
---|
| 69 |
|
---|
[a1a101d] | 70 | log_msg(LOG_DEFAULT, LVL_DEBUG, "tcp_inet_ev_recv()");
|
---|
[7c8267b] | 71 |
|
---|
[c76e926] | 72 | pdu_raw = dgram->data;
|
---|
| 73 | pdu_raw_size = dgram->size;
|
---|
[7c8267b] | 74 |
|
---|
[1812a0d] | 75 | /* Split into header and payload. */
|
---|
[21580dd] | 76 |
|
---|
[a1a101d] | 77 | log_msg(LOG_DEFAULT, LVL_DEBUG, "tcp_inet_ev_recv() - split header/payload");
|
---|
[7c8267b] | 78 |
|
---|
[1812a0d] | 79 | tcp_pdu_t *pdu;
|
---|
| 80 | size_t hdr_size;
|
---|
[fe5e9629] | 81 | tcp_header_t *hdr;
|
---|
| 82 | uint32_t data_offset;
|
---|
[7c8267b] | 83 |
|
---|
[fe5e9629] | 84 | if (pdu_raw_size < sizeof(tcp_header_t)) {
|
---|
[a1a101d] | 85 | log_msg(LOG_DEFAULT, LVL_WARN, "pdu_raw_size = %zu < sizeof(tcp_header_t) = %zu",
|
---|
[fe5e9629] | 86 | pdu_raw_size, sizeof(tcp_header_t));
|
---|
| 87 | return EINVAL;
|
---|
| 88 | }
|
---|
| 89 |
|
---|
| 90 | hdr = (tcp_header_t *)pdu_raw;
|
---|
| 91 | data_offset = BIT_RANGE_EXTRACT(uint32_t, DF_DATA_OFFSET_h, DF_DATA_OFFSET_l,
|
---|
| 92 | uint16_t_be2host(hdr->doff_flags));
|
---|
| 93 |
|
---|
| 94 | hdr_size = sizeof(uint32_t) * data_offset;
|
---|
[7c8267b] | 95 |
|
---|
[1812a0d] | 96 | if (pdu_raw_size < hdr_size) {
|
---|
[a1a101d] | 97 | log_msg(LOG_DEFAULT, LVL_WARN, "pdu_raw_size = %zu < hdr_size = %zu",
|
---|
[1812a0d] | 98 | pdu_raw_size, hdr_size);
|
---|
| 99 | return EINVAL;
|
---|
[7c8267b] | 100 | }
|
---|
| 101 |
|
---|
[fe5e9629] | 102 | if (hdr_size < sizeof(tcp_header_t)) {
|
---|
[a1a101d] | 103 | log_msg(LOG_DEFAULT, LVL_WARN, "hdr_size = %zu < sizeof(tcp_header_t) = %zu",
|
---|
[c76e926] | 104 | hdr_size, sizeof(tcp_header_t)); return EINVAL;
|
---|
[fe5e9629] | 105 | }
|
---|
| 106 |
|
---|
[a1a101d] | 107 | log_msg(LOG_DEFAULT, LVL_DEBUG, "pdu_raw_size=%zu, hdr_size=%zu",
|
---|
[1812a0d] | 108 | pdu_raw_size, hdr_size);
|
---|
| 109 | pdu = tcp_pdu_create(pdu_raw, hdr_size, pdu_raw + hdr_size,
|
---|
| 110 | pdu_raw_size - hdr_size);
|
---|
| 111 | if (pdu == NULL) {
|
---|
[a1a101d] | 112 | log_msg(LOG_DEFAULT, LVL_WARN, "Failed creating PDU. Dropped.");
|
---|
[1812a0d] | 113 | return ENOMEM;
|
---|
[21580dd] | 114 | }
|
---|
| 115 |
|
---|
[02a09ed] | 116 | pdu->src = dgram->src;
|
---|
| 117 | pdu->dest = dgram->dest;
|
---|
[21580dd] | 118 |
|
---|
[1812a0d] | 119 | tcp_received_pdu(pdu);
|
---|
| 120 | tcp_pdu_delete(pdu);
|
---|
[7c8267b] | 121 |
|
---|
[21580dd] | 122 | return EOK;
|
---|
| 123 | }
|
---|
| 124 |
|
---|
[1812a0d] | 125 | /** Transmit PDU over network layer. */
|
---|
| 126 | void tcp_transmit_pdu(tcp_pdu_t *pdu)
|
---|
[7c8267b] | 127 | {
|
---|
[0578271] | 128 | int rc;
|
---|
[c76e926] | 129 | uint8_t *pdu_raw;
|
---|
| 130 | size_t pdu_raw_size;
|
---|
| 131 | inet_dgram_t dgram;
|
---|
[7c8267b] | 132 |
|
---|
[c76e926] | 133 | pdu_raw_size = pdu->header_size + pdu->text_size;
|
---|
| 134 | pdu_raw = malloc(pdu_raw_size);
|
---|
| 135 | if (pdu_raw == NULL) {
|
---|
[a1a101d] | 136 | log_msg(LOG_DEFAULT, LVL_ERROR, "Failed to transmit PDU. Out of memory.");
|
---|
[1812a0d] | 137 | return;
|
---|
[7c8267b] | 138 | }
|
---|
[21580dd] | 139 |
|
---|
[c76e926] | 140 | memcpy(pdu_raw, pdu->header, pdu->header_size);
|
---|
| 141 | memcpy(pdu_raw + pdu->header_size, pdu->text,
|
---|
[1812a0d] | 142 | pdu->text_size);
|
---|
[7c8267b] | 143 |
|
---|
[695b6ff] | 144 | dgram.iplink = 0;
|
---|
[02a09ed] | 145 | dgram.src = pdu->src;
|
---|
| 146 | dgram.dest = pdu->dest;
|
---|
[c76e926] | 147 | dgram.tos = 0;
|
---|
| 148 | dgram.data = pdu_raw;
|
---|
| 149 | dgram.size = pdu_raw_size;
|
---|
| 150 |
|
---|
| 151 | rc = inet_send(&dgram, INET_TTL_MAX, 0);
|
---|
| 152 | if (rc != EOK)
|
---|
[a1a101d] | 153 | log_msg(LOG_DEFAULT, LVL_ERROR, "Failed to transmit PDU.");
|
---|
[9787a93] | 154 |
|
---|
| 155 | free(pdu_raw);
|
---|
[21580dd] | 156 | }
|
---|
| 157 |
|
---|
[1812a0d] | 158 | /** Process received PDU. */
|
---|
| 159 | static void tcp_received_pdu(tcp_pdu_t *pdu)
|
---|
[7c8267b] | 160 | {
|
---|
[1812a0d] | 161 | tcp_segment_t *dseg;
|
---|
[2f19103] | 162 | inet_ep2_t rident;
|
---|
[fb04cba8] | 163 |
|
---|
[a1a101d] | 164 | log_msg(LOG_DEFAULT, LVL_DEBUG, "tcp_received_pdu()");
|
---|
[fb04cba8] | 165 |
|
---|
[1812a0d] | 166 | if (tcp_pdu_decode(pdu, &rident, &dseg) != EOK) {
|
---|
[a1a101d] | 167 | log_msg(LOG_DEFAULT, LVL_WARN, "Not enough memory. PDU dropped.");
|
---|
[7c8267b] | 168 | return;
|
---|
| 169 | }
|
---|
[fb04cba8] | 170 |
|
---|
[1812a0d] | 171 | /* Insert decoded segment into rqueue */
|
---|
| 172 | tcp_rqueue_insert_seg(&rident, dseg);
|
---|
[21580dd] | 173 | }
|
---|
| 174 |
|
---|
[c76e926] | 175 | static int tcp_init(void)
|
---|
[7c8267b] | 176 | {
|
---|
[1812a0d] | 177 | int rc;
|
---|
[21580dd] | 178 |
|
---|
[a1a101d] | 179 | log_msg(LOG_DEFAULT, LVL_DEBUG, "tcp_init()");
|
---|
[1812a0d] | 180 |
|
---|
[c76e926] | 181 | tcp_rqueue_init();
|
---|
[88a6819] | 182 | tcp_rqueue_fibril_start();
|
---|
[1812a0d] | 183 |
|
---|
[c76e926] | 184 | tcp_ncsim_init();
|
---|
[88a6819] | 185 | tcp_ncsim_fibril_start();
|
---|
[a4ee3ab2] | 186 |
|
---|
[c76e926] | 187 | if (0) tcp_test();
|
---|
| 188 |
|
---|
[59157eb] | 189 | rc = inet_init(IP_PROTO_TCP, &tcp_inet_ev_ops);
|
---|
[ecff3d9] | 190 | if (rc != EOK) {
|
---|
[a1a101d] | 191 | log_msg(LOG_DEFAULT, LVL_ERROR, "Failed connecting to internet service.");
|
---|
[1812a0d] | 192 | return ENOENT;
|
---|
[ecff3d9] | 193 | }
|
---|
[21580dd] | 194 |
|
---|
[779541b] | 195 | rc = tcp_service_init();
|
---|
| 196 | if (rc != EOK) {
|
---|
| 197 | log_msg(LOG_DEFAULT, LVL_ERROR, "Failed initializing service.");
|
---|
| 198 | return ENOENT;
|
---|
| 199 | }
|
---|
[1812a0d] | 200 |
|
---|
| 201 | return EOK;
|
---|
[21580dd] | 202 | }
|
---|
| 203 |
|
---|
[c5808b41] | 204 | int main(int argc, char **argv)
|
---|
[7c8267b] | 205 | {
|
---|
[0578271] | 206 | int rc;
|
---|
[2e99277] | 207 |
|
---|
[c5808b41] | 208 | printf(NAME ": TCP (Transmission Control Protocol) network module\n");
|
---|
[7c8267b] | 209 |
|
---|
[267f235] | 210 | rc = log_init(NAME);
|
---|
[0578271] | 211 | if (rc != EOK) {
|
---|
[c5808b41] | 212 | printf(NAME ": Failed to initialize log.\n");
|
---|
| 213 | return 1;
|
---|
[21580dd] | 214 | }
|
---|
[7c8267b] | 215 |
|
---|
[ecff3d9] | 216 | rc = tcp_init();
|
---|
| 217 | if (rc != EOK)
|
---|
| 218 | return 1;
|
---|
| 219 |
|
---|
| 220 | printf(NAME ": Accepting connections.\n");
|
---|
[c76e926] | 221 | task_retval(0);
|
---|
[c5808b41] | 222 | async_manager();
|
---|
[849ed54] | 223 |
|
---|
[c5808b41] | 224 | /* Not reached */
|
---|
| 225 | return 0;
|
---|
[014dd57b] | 226 | }
|
---|
| 227 |
|
---|
[c5808b41] | 228 | /**
|
---|
| 229 | * @}
|
---|
[21580dd] | 230 | */
|
---|