source: mainline/uspace/srv/net/udp/udp_inet.c@ f1a8c23

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

Fix leak of inet dgram.data in TCP, should be freed by inet_ev_recv().

  • Property mode set to 100644
File size: 3.5 KB
Line 
1/*
2 * Copyright (c) 2012 Jiri Svoboda
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 udp
30 * @{
31 */
32
33/**
34 * @file
35 */
36
37#include <bitops.h>
38#include <byteorder.h>
39#include <errno.h>
40#include <inet/inet.h>
41#include <io/log.h>
42#include <stdio.h>
43#include <task.h>
44
45#include "assoc.h"
46#include "pdu.h"
47#include "std.h"
48#include "udp_inet.h"
49#include "udp_type.h"
50
51static int udp_inet_ev_recv(inet_dgram_t *dgram);
52static void udp_received_pdu(udp_pdu_t *pdu);
53
54static inet_ev_ops_t udp_inet_ev_ops = {
55 .recv = udp_inet_ev_recv
56};
57
58/** Received datagram callback */
59static int udp_inet_ev_recv(inet_dgram_t *dgram)
60{
61 udp_pdu_t *pdu;
62
63 log_msg(LOG_DEFAULT, LVL_DEBUG, "udp_inet_ev_recv()");
64
65 pdu = udp_pdu_new();
66 pdu->data = dgram->data;
67 pdu->data_size = dgram->size;
68
69 pdu->src = dgram->src;
70 pdu->dest = dgram->dest;
71
72 udp_received_pdu(pdu);
73
74 /* We don't want udp_pdu_delete() to free dgram->data */
75 pdu->data = NULL;
76 udp_pdu_delete(pdu);
77
78 return EOK;
79}
80
81/** Transmit PDU over network layer. */
82int udp_transmit_pdu(udp_pdu_t *pdu)
83{
84 int rc;
85 inet_dgram_t dgram;
86
87 log_msg(LOG_DEFAULT, LVL_DEBUG, "udp_transmit_pdu()");
88
89 dgram.iplink = pdu->iplink;
90 dgram.src = pdu->src;
91 dgram.dest = pdu->dest;
92 dgram.tos = 0;
93 dgram.data = pdu->data;
94 dgram.size = pdu->data_size;
95
96 rc = inet_send(&dgram, INET_TTL_MAX, 0);
97 if (rc != EOK)
98 log_msg(LOG_DEFAULT, LVL_ERROR, "Failed to transmit PDU.");
99
100 return rc;
101}
102
103/** Process received PDU. */
104static void udp_received_pdu(udp_pdu_t *pdu)
105{
106 udp_msg_t *dmsg;
107 udp_sockpair_t rident;
108
109 log_msg(LOG_DEFAULT, LVL_DEBUG, "udp_received_pdu()");
110
111 if (udp_pdu_decode(pdu, &rident, &dmsg) != EOK) {
112 log_msg(LOG_DEFAULT, LVL_WARN, "Not enough memory. PDU dropped.");
113 return;
114 }
115
116 /*
117 * Insert decoded message into appropriate receive queue.
118 * This transfers ownership of dmsg to the callee, we do not
119 * free it.
120 */
121 udp_assoc_received(&rident, dmsg);
122}
123
124int udp_inet_init(void)
125{
126 int rc;
127
128 log_msg(LOG_DEFAULT, LVL_DEBUG, "udp_inet_init()");
129
130 rc = inet_init(IP_PROTO_UDP, &udp_inet_ev_ops);
131 if (rc != EOK) {
132 log_msg(LOG_DEFAULT, LVL_ERROR, "Failed connecting to internet service.");
133 return ENOENT;
134 }
135
136 return EOK;
137}
138
139/**
140 * @}
141 */
Note: See TracBrowser for help on using the repository browser.