source: mainline/uspace/srv/net/tcp/tcp.c@ 2f19103

lfn serial ticket/834-toolchain-update topic/msim-upgrade topic/simplify-dev-export
Last change on this file since 2f19103 was 2f19103, checked in by Jiri Svoboda <jiri@…>, 10 years ago

TCP and UDP servers can make use of inet/endpoint.h types internally.

  • Property mode set to 100644
File size: 5.6 KB
RevLine 
[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]57static int tcp_inet_ev_recv(inet_dgram_t *dgram);
[1812a0d]58static void tcp_received_pdu(tcp_pdu_t *pdu);
[21580dd]59
[c76e926]60static inet_ev_ops_t tcp_inet_ev_ops = {
61 .recv = tcp_inet_ev_recv
62};
[7c8267b]63
[c76e926]64/** Received datagram callback */
65static 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. */
126void 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. */
159static 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]175static 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]204int 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 */
Note: See TracBrowser for help on using the repository browser.