← Back to branch summary

~ubuntu-branches/ubuntu/maverick/sflphone/maverick

« back to all changes in this revision

Viewing changes to sflphone-common/libs/pjproject/pjmedia/src/pjmedia/transport_loop.c

  • Committer: Bazaar Package Importer
  • Author(s): Francois Marier
  • Date: 2010-06-03 15:59:46 UTC
  • Revision ID: james.westby@ubuntu.com-20100603155946-ybe8d8o8zx8lp0m8
Tags: upstream-0.9.8.3
Import upstream version 0.9.8.3

Show diffs side-by-side

added added

removed removed

Lines of Context:
sflphone-common/libs/pjproject/pjmedia/src/pjmedia/transport_loop.c
 
1
/* $Id: transport_loop.c 2881 2009-08-15 09:53:58Z bennylp $ */
 
2
/* 
 
3
 * Copyright (C) 2008-2009 Teluu Inc. (http://www.teluu.com)
 
4
 * Copyright (C) 2003-2008 Benny Prijono <benny@prijono.org>
 
5
 *
 
6
 * This program is free software; you can redistribute it and/or modify
 
7
 * it under the terms of the GNU General Public License as published by
 
8
 * the Free Software Foundation; either version 2 of the License, or
 
9
 * (at your option) any later version.
 
10
 *
 
11
 * This program is distributed in the hope that it will be useful,
 
12
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 
13
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
14
 * GNU General Public License for more details.
 
15
 *
 
16
 * You should have received a copy of the GNU General Public License
 
17
 * along with this program; if not, write to the Free Software
 
18
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA 
 
19
 *
 
20
 *  Additional permission under GNU GPL version 3 section 7:
 
21
 *
 
22
 *  If you modify this program, or any covered work, by linking or
 
23
 *  combining it with the OpenSSL project's OpenSSL library (or a
 
24
 *  modified version of that library), containing parts covered by the
 
25
 *  terms of the OpenSSL or SSLeay licenses, Teluu Inc. (http://www.teluu.com)
 
26
 *  grants you additional permission to convey the resulting work.
 
27
 *  Corresponding Source for a non-source form of such a combination
 
28
 *  shall include the source code for the parts of OpenSSL used as well
 
29
 *  as that of the covered work.
 
30
 */
 
31
#include <pjmedia/transport_loop.h>
 
32
#include <pj/array.h>
 
33
#include <pj/assert.h>
 
34
#include <pj/errno.h>
 
35
#include <pj/ioqueue.h>
 
36
#include <pj/log.h>
 
37
#include <pj/pool.h>
 
38
#include <pj/rand.h>
 
39
#include <pj/string.h>
 
40
 
 
41
 
 
42
struct user
 
43
{
 
44
    pj_bool_t           rx_disabled;    /**< Doesn't want to receive pkt?   */
 
45
    void               *user_data;      /**< Only valid when attached       */
 
46
    void  (*rtp_cb)(    void*,          /**< To report incoming RTP.        */
 
47
                        void*,
 
48
                        pj_ssize_t);
 
49
    void  (*rtcp_cb)(   void*,          /**< To report incoming RTCP.       */
 
50
                        void*,
 
51
                        pj_ssize_t);
 
52
};
 
53
 
 
54
struct transport_loop
 
55
{
 
56
    pjmedia_transport   base;           /**< Base transport.                */
 
57
 
 
58
    pj_pool_t          *pool;           /**< Memory pool                    */
 
59
    unsigned            user_cnt;       /**< Number of attachments          */
 
60
    struct user         users[4];       /**< Array of users.                */
 
61
 
 
62
    unsigned            tx_drop_pct;    /**< Percent of tx pkts to drop.    */
 
63
    unsigned            rx_drop_pct;    /**< Percent of rx pkts to drop.    */
 
64
 
 
65
};
 
66
 
 
67
 
 
68
 
 
69
/*
 
70
 * These are media transport operations.
 
71
 */
 
72
static pj_status_t transport_get_info (pjmedia_transport *tp,
 
73
                                       pjmedia_transport_info *info);
 
74
static pj_status_t transport_attach   (pjmedia_transport *tp,
 
75
                                       void *user_data,
 
76
                                       const pj_sockaddr_t *rem_addr,
 
77
                                       const pj_sockaddr_t *rem_rtcp,
 
78
                                       unsigned addr_len,
 
79
                                       void (*rtp_cb)(void*,
 
80
                                                      void*,
 
81
                                                      pj_ssize_t),
 
82
                                       void (*rtcp_cb)(void*,
 
83
                                                       void*,
 
84
                                                       pj_ssize_t));
 
85
static void        transport_detach   (pjmedia_transport *tp,
 
86
                                       void *strm);
 
87
static pj_status_t transport_send_rtp( pjmedia_transport *tp,
 
88
                                       const void *pkt,
 
89
                                       pj_size_t size);
 
90
static pj_status_t transport_send_rtcp(pjmedia_transport *tp,
 
91
                                       const void *pkt,
 
92
                                       pj_size_t size);
 
93
static pj_status_t transport_send_rtcp2(pjmedia_transport *tp,
 
94
                                       const pj_sockaddr_t *addr,
 
95
                                       unsigned addr_len,
 
96
                                       const void *pkt,
 
97
                                       pj_size_t size);
 
98
static pj_status_t transport_media_create(pjmedia_transport *tp,
 
99
                                       pj_pool_t *pool,
 
100
                                       unsigned options,
 
101
                                       const pjmedia_sdp_session *sdp_remote,
 
102
                                       unsigned media_index);
 
103
static pj_status_t transport_encode_sdp(pjmedia_transport *tp,
 
104
                                        pj_pool_t *pool,
 
105
                                        pjmedia_sdp_session *sdp_local,
 
106
                                        const pjmedia_sdp_session *rem_sdp,
 
107
                                        unsigned media_index);
 
108
static pj_status_t transport_media_start (pjmedia_transport *tp,
 
109
                                       pj_pool_t *pool,
 
110
                                       const pjmedia_sdp_session *sdp_local,
 
111
                                       const pjmedia_sdp_session *sdp_remote,
 
112
                                       unsigned media_index);
 
113
static pj_status_t transport_media_stop(pjmedia_transport *tp);
 
114
static pj_status_t transport_simulate_lost(pjmedia_transport *tp,
 
115
                                       pjmedia_dir dir,
 
116
                                       unsigned pct_lost);
 
117
static pj_status_t transport_destroy  (pjmedia_transport *tp);
 
118
 
 
119
 
 
120
static pjmedia_transport_op transport_udp_op = 
 
121
{
 
122
    &transport_get_info,
 
123
    &transport_attach,
 
124
    &transport_detach,
 
125
    &transport_send_rtp,
 
126
    &transport_send_rtcp,
 
127
    &transport_send_rtcp2,
 
128
    &transport_media_create,
 
129
    &transport_encode_sdp,
 
130
    &transport_media_start,
 
131
    &transport_media_stop,
 
132
    &transport_simulate_lost,
 
133
    &transport_destroy
 
134
};
 
135
 
 
136
 
 
137
/**
 
138
 * Create loopback transport.
 
139
 */
 
140
PJ_DEF(pj_status_t) pjmedia_transport_loop_create(pjmedia_endpt *endpt,
 
141
                                                  pjmedia_transport **p_tp)
 
142
{
 
143
    struct transport_loop *tp;
 
144
    pj_pool_t *pool;
 
145
 
 
146
    /* Sanity check */
 
147
    PJ_ASSERT_RETURN(endpt && p_tp, PJ_EINVAL);
 
148
 
 
149
    /* Create transport structure */
 
150
    pool = pjmedia_endpt_create_pool(endpt, "tploop", 512, 512);
 
151
    if (!pool)
 
152
        return PJ_ENOMEM;
 
153
 
 
154
    tp = PJ_POOL_ZALLOC_T(pool, struct transport_loop);
 
155
    tp->pool = pool;
 
156
    pj_ansi_strncpy(tp->base.name, tp->pool->obj_name, PJ_MAX_OBJ_NAME-1);
 
157
    tp->base.op = &transport_udp_op;
 
158
    tp->base.type = PJMEDIA_TRANSPORT_TYPE_UDP;
 
159
 
 
160
    /* Done */
 
161
    *p_tp = &tp->base;
 
162
    return PJ_SUCCESS;
 
163
}
 
164
 
 
165
 
 
166
PJ_DEF(pj_status_t) pjmedia_transport_loop_disable_rx( pjmedia_transport *tp,
 
167
                                                       void *user,
 
168
                                                       pj_bool_t disabled)
 
169
{
 
170
    struct transport_loop *loop = (struct transport_loop*) tp;
 
171
    unsigned i;
 
172
 
 
173
    for (i=0; i<loop->user_cnt; ++i) {
 
174
        if (loop->users[i].user_data == user) {
 
175
            loop->users[i].rx_disabled = disabled;
 
176
            return PJ_SUCCESS;
 
177
        }
 
178
    }
 
179
    pj_assert(!"Invalid stream user");
 
180
    return PJ_ENOTFOUND;
 
181
}
 
182
 
 
183
/**
 
184
 * Close loopback transport.
 
185
 */
 
186
static pj_status_t transport_destroy(pjmedia_transport *tp)
 
187
{
 
188
    struct transport_loop *loop = (struct transport_loop*) tp;
 
189
 
 
190
    /* Sanity check */
 
191
    PJ_ASSERT_RETURN(tp, PJ_EINVAL);
 
192
 
 
193
    pj_pool_release(loop->pool);
 
194
 
 
195
    return PJ_SUCCESS;
 
196
}
 
197
 
 
198
 
 
199
/* Called to get the transport info */
 
200
static pj_status_t transport_get_info(pjmedia_transport *tp,
 
201
                                      pjmedia_transport_info *info)
 
202
{
 
203
    PJ_ASSERT_RETURN(tp && info, PJ_EINVAL);
 
204
 
 
205
    info->sock_info.rtp_sock = 1;
 
206
    pj_sockaddr_in_init(&info->sock_info.rtp_addr_name.ipv4, 0, 0);
 
207
    info->sock_info.rtcp_sock = 2;
 
208
    pj_sockaddr_in_init(&info->sock_info.rtcp_addr_name.ipv4, 0, 0);
 
209
 
 
210
    return PJ_SUCCESS;
 
211
}
 
212
 
 
213
 
 
214
/* Called by application to initialize the transport */
 
215
static pj_status_t transport_attach(   pjmedia_transport *tp,
 
216
                                       void *user_data,
 
217
                                       const pj_sockaddr_t *rem_addr,
 
218
                                       const pj_sockaddr_t *rem_rtcp,
 
219
                                       unsigned addr_len,
 
220
                                       void (*rtp_cb)(void*,
 
221
                                                      void*,
 
222
                                                      pj_ssize_t),
 
223
                                       void (*rtcp_cb)(void*,
 
224
                                                       void*,
 
225
                                                       pj_ssize_t))
 
226
{
 
227
    struct transport_loop *loop = (struct transport_loop*) tp;
 
228
    unsigned i;
 
229
    const pj_sockaddr *rtcp_addr;
 
230
 
 
231
    /* Validate arguments */
 
232
    PJ_ASSERT_RETURN(tp && rem_addr && addr_len, PJ_EINVAL);
 
233
 
 
234
    /* Must not be "attached" to same user */
 
235
    for (i=0; i<loop->user_cnt; ++i) {
 
236
        PJ_ASSERT_RETURN(loop->users[i].user_data != user_data,
 
237
                         PJ_EINVALIDOP);
 
238
    }
 
239
    PJ_ASSERT_RETURN(loop->user_cnt != PJ_ARRAY_SIZE(loop->users), 
 
240
                     PJ_ETOOMANY);
 
241
 
 
242
    PJ_UNUSED_ARG(rem_rtcp);
 
243
    PJ_UNUSED_ARG(rtcp_addr);
 
244
 
 
245
    /* "Attach" the application: */
 
246
 
 
247
    /* Save the new user */
 
248
    loop->users[loop->user_cnt].rtp_cb = rtp_cb;
 
249
    loop->users[loop->user_cnt].rtcp_cb = rtcp_cb;
 
250
    loop->users[loop->user_cnt].user_data = user_data;
 
251
    ++loop->user_cnt;
 
252
 
 
253
    return PJ_SUCCESS;
 
254
}
 
255
 
 
256
 
 
257
/* Called by application when it no longer needs the transport */
 
258
static void transport_detach( pjmedia_transport *tp,
 
259
                              void *user_data)
 
260
{
 
261
    struct transport_loop *loop = (struct transport_loop*) tp;
 
262
    unsigned i;
 
263
 
 
264
    pj_assert(tp);
 
265
 
 
266
    for (i=0; i<loop->user_cnt; ++i) {
 
267
        if (loop->users[i].user_data == user_data)
 
268
            break;
 
269
    }
 
270
 
 
271
    /* Remove this user */
 
272
    if (i != loop->user_cnt) {
 
273
        pj_array_erase(loop->users, sizeof(loop->users[0]),
 
274
                       loop->user_cnt, i);
 
275
        --loop->user_cnt;
 
276
    }
 
277
}
 
278
 
 
279
 
 
280
/* Called by application to send RTP packet */
 
281
static pj_status_t transport_send_rtp( pjmedia_transport *tp,
 
282
                                       const void *pkt,
 
283
                                       pj_size_t size)
 
284
{
 
285
    struct transport_loop *loop = (struct transport_loop*)tp;
 
286
    unsigned i;
 
287
 
 
288
    /* Simulate packet lost on TX direction */
 
289
    if (loop->tx_drop_pct) {
 
290
        if ((pj_rand() % 100) <= (int)loop->tx_drop_pct) {
 
291
            PJ_LOG(5,(loop->base.name, 
 
292
                      "TX RTP packet dropped because of pkt lost "
 
293
                      "simulation"));
 
294
            return PJ_SUCCESS;
 
295
        }
 
296
    }
 
297
 
 
298
    /* Simulate packet lost on RX direction */
 
299
    if (loop->rx_drop_pct) {
 
300
        if ((pj_rand() % 100) <= (int)loop->rx_drop_pct) {
 
301
            PJ_LOG(5,(loop->base.name, 
 
302
                      "RX RTP packet dropped because of pkt lost "
 
303
                      "simulation"));
 
304
            return PJ_SUCCESS;
 
305
        }
 
306
    }
 
307
 
 
308
    /* Distribute to users */
 
309
    for (i=0; i<loop->user_cnt; ++i) {
 
310
        if (!loop->users[i].rx_disabled && loop->users[i].rtp_cb)
 
311
            (*loop->users[i].rtp_cb)(loop->users[i].user_data, (void*)pkt, 
 
312
                                     size);
 
313
    }
 
314
 
 
315
    return PJ_SUCCESS;
 
316
}
 
317
 
 
318
/* Called by application to send RTCP packet */
 
319
static pj_status_t transport_send_rtcp(pjmedia_transport *tp,
 
320
                                       const void *pkt,
 
321
                                       pj_size_t size)
 
322
{
 
323
    return transport_send_rtcp2(tp, NULL, 0, pkt, size);
 
324
}
 
325
 
 
326
 
 
327
/* Called by application to send RTCP packet */
 
328
static pj_status_t transport_send_rtcp2(pjmedia_transport *tp,
 
329
                                        const pj_sockaddr_t *addr,
 
330
                                        unsigned addr_len,
 
331
                                        const void *pkt,
 
332
                                        pj_size_t size)
 
333
{
 
334
    struct transport_loop *loop = (struct transport_loop*)tp;
 
335
    unsigned i;
 
336
 
 
337
    PJ_UNUSED_ARG(addr_len);
 
338
    PJ_UNUSED_ARG(addr);
 
339
 
 
340
    /* Distribute to users */
 
341
    for (i=0; i<loop->user_cnt; ++i) {
 
342
        if (!loop->users[i].rx_disabled && loop->users[i].rtcp_cb)
 
343
            (*loop->users[i].rtcp_cb)(loop->users[i].user_data, (void*)pkt,
 
344
                                      size);
 
345
    }
 
346
 
 
347
    return PJ_SUCCESS;
 
348
}
 
349
 
 
350
 
 
351
static pj_status_t transport_media_create(pjmedia_transport *tp,
 
352
                                  pj_pool_t *pool,
 
353
                                  unsigned options,
 
354
                                  const pjmedia_sdp_session *sdp_remote,
 
355
                                  unsigned media_index)
 
356
{
 
357
    PJ_UNUSED_ARG(tp);
 
358
    PJ_UNUSED_ARG(pool);
 
359
    PJ_UNUSED_ARG(options);
 
360
    PJ_UNUSED_ARG(sdp_remote);
 
361
    PJ_UNUSED_ARG(media_index);
 
362
    return PJ_SUCCESS;
 
363
}
 
364
 
 
365
static pj_status_t transport_encode_sdp(pjmedia_transport *tp,
 
366
                                        pj_pool_t *pool,
 
367
                                        pjmedia_sdp_session *sdp_local,
 
368
                                        const pjmedia_sdp_session *rem_sdp,
 
369
                                        unsigned media_index)
 
370
{
 
371
    PJ_UNUSED_ARG(tp);
 
372
    PJ_UNUSED_ARG(pool);
 
373
    PJ_UNUSED_ARG(sdp_local);
 
374
    PJ_UNUSED_ARG(rem_sdp);
 
375
    PJ_UNUSED_ARG(media_index);
 
376
    return PJ_SUCCESS;
 
377
}
 
378
 
 
379
static pj_status_t transport_media_start(pjmedia_transport *tp,
 
380
                                  pj_pool_t *pool,
 
381
                                  const pjmedia_sdp_session *sdp_local,
 
382
                                  const pjmedia_sdp_session *sdp_remote,
 
383
                                  unsigned media_index)
 
384
{
 
385
    PJ_UNUSED_ARG(tp);
 
386
    PJ_UNUSED_ARG(pool);
 
387
    PJ_UNUSED_ARG(sdp_local);
 
388
    PJ_UNUSED_ARG(sdp_remote);
 
389
    PJ_UNUSED_ARG(media_index);
 
390
    return PJ_SUCCESS;
 
391
}
 
392
 
 
393
static pj_status_t transport_media_stop(pjmedia_transport *tp)
 
394
{
 
395
    PJ_UNUSED_ARG(tp);
 
396
    return PJ_SUCCESS;
 
397
}
 
398
 
 
399
static pj_status_t transport_simulate_lost(pjmedia_transport *tp,
 
400
                                           pjmedia_dir dir,
 
401
                                           unsigned pct_lost)
 
402
{
 
403
    struct transport_loop *loop = (struct transport_loop*)tp;
 
404
 
 
405
    PJ_ASSERT_RETURN(tp && pct_lost <= 100, PJ_EINVAL);
 
406
 
 
407
    if (dir & PJMEDIA_DIR_ENCODING)
 
408
        loop->tx_drop_pct = pct_lost;
 
409
    
 
410
    if (dir & PJMEDIA_DIR_DECODING)
 
411
        loop->rx_drop_pct = pct_lost;
 
412
 
 
413
    return PJ_SUCCESS;
 
414
}
 
415