1
/* $Id: audio_tool.c 3553 2011-05-05 06:14:19Z nanang $ */
3
* Copyright (C) 2008-2011 Teluu Inc. (http://www.teluu.com)
4
* Copyright (C) 2003-2008 Benny Prijono <benny@prijono.org>
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.
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.
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
24
#define THIS_FILE "audio_tool.c"
26
static pj_caching_pool caching_pool;
27
static pj_pool_factory *pf;
29
static pj_med_mgr_t *mm;
30
static pjmedia_codec *codec;
31
static pjmedia_codec_param cattr;
34
#define WRITE_ORIGINAL_PCM 0
35
#if WRITE_ORIGINAL_PCM
36
static FILE *fhnd_pcm;
39
static char talker_sdp[] =
41
"o=- 0 0 IN IP4 127.0.0.1\r\n"
43
"c=IN IP4 127.0.0.1\r\n"
45
"m=audio 4002 RTP/AVP 0\r\n"
46
"a=rtpmap:0 PCMU/8000\r\n"
48
static char listener_sdp[] =
50
"o=- 0 0 IN IP4 127.0.0.1\r\n"
52
"c=IN IP4 127.0.0.1\r\n"
54
"m=audio 4000 RTP/AVP 0\r\n"
55
"a=rtpmap:0 PCMU/8000\r\n"
58
static pj_status_t play_callback(/* in */ void *user_data,
59
/* in */ pj_uint32_t timestamp,
60
/* out */ void *frame,
61
/* out */ unsigned size)
64
struct pjmedia_frame in, out;
65
int frmsz = cattr.avg_bps / 8 * cattr.ptime / 1000;
67
if (fread(pkt, frmsz, 1, fhnd) != 1) {
71
in.type = PJMEDIA_FRAME_TYPE_AUDIO;
75
if (codec->op->decode (codec, &in, size, &out) != 0)
83
static pj_status_t rec_callback( /* in */ void *user_data,
84
/* in */ pj_uint32_t timestamp,
85
/* in */ const void *frame,
86
/* in*/ unsigned size)
89
struct pjmedia_frame in, out;
90
//int frmsz = cattr.avg_bps / 8 * cattr.ptime / 1000;
92
#if WRITE_ORIGINAL_PCM
93
fwrite(frame, size, 1, fhnd_pcm);
96
in.type = PJMEDIA_FRAME_TYPE_AUDIO;
97
in.buf = (void*)frame;
101
if (codec->op->encode(codec, &in, sizeof(pkt), &out) != 0)
104
if (fwrite(pkt, out.size, 1, fhnd) != 1)
109
static pj_status_t init()
111
pjmedia_codec_mgr *cm;
112
pjmedia_codec_info id;
115
pj_caching_pool_init(&caching_pool, &pj_pool_factory_default_policy, 0);
116
pf = &caching_pool.factory;
118
if (pj_snd_init(&caching_pool.factory))
121
PJ_LOG(3,(THIS_FILE, "Dumping audio devices:"));
122
for (i=0; i<pj_snd_get_dev_count(); ++i) {
123
const pj_snd_dev_info *info;
124
info = pj_snd_get_dev_info(i);
125
PJ_LOG(3,(THIS_FILE, " %d: %s\t(%d in, %d out",
127
info->input_count, info->output_count));
130
mm = pj_med_mgr_create (&caching_pool.factory);
131
cm = pj_med_mgr_get_codec_mgr (mm);
133
id.type = PJMEDIA_TYPE_AUDIO;
135
id.encoding_name = pj_str("PCMU");
136
id.sample_rate = 8000;
138
codec = pjmedia_codec_mgr_alloc_codec (cm, &id);
139
codec->op->default_attr(codec, &cattr);
140
codec->op->open(codec, &cattr);
144
static pj_status_t deinit()
146
pjmedia_codec_mgr *cm;
147
cm = pj_med_mgr_get_codec_mgr (mm);
148
codec->op->close(codec);
149
pjmedia_codec_mgr_dealloc_codec (cm, codec);
150
pj_med_mgr_destroy (mm);
151
pj_caching_pool_destroy(&caching_pool);
155
static pj_status_t record_file (const char *filename)
157
pj_snd_stream *stream;
158
pj_snd_stream_info info;
162
printf("Recording to file %s...\n", filename);
164
fhnd = fopen(filename, "wb");
168
#if WRITE_ORIGINAL_PCM
169
fhnd_pcm = fopen("ORIGINAL.PCM", "wb");
174
pj_bzero(&info, sizeof(info));
175
info.bits_per_sample = 16;
176
info.bytes_per_frame = 2;
177
info.frames_per_packet = 160;
178
info.samples_per_frame = 1;
179
info.samples_per_sec = 8000;
181
stream = pj_snd_open_recorder(-1, &info, &rec_callback, NULL);
185
status = pj_snd_stream_start(stream);
189
puts("Press <ENTER> to exit recording");
190
fgets(s, sizeof(s), stdin);
192
pj_snd_stream_stop(stream);
193
pj_snd_stream_close(stream);
195
#if WRITE_ORIGINAL_PCM
202
pj_snd_stream_stop(stream);
203
pj_snd_stream_close(stream);
208
static pj_status_t play_file (const char *filename)
210
pj_snd_stream *stream;
211
pj_snd_stream_info info;
215
printf("Playing file %s...\n", filename);
217
fhnd = fopen(filename, "rb");
221
pj_bzero(&info, sizeof(info));
222
info.bits_per_sample = 16;
223
info.bytes_per_frame = 2;
224
info.frames_per_packet = 160;
225
info.samples_per_frame = 1;
226
info.samples_per_sec = 8000;
228
stream = pj_snd_open_player(-1, &info, &play_callback, NULL);
232
status = pj_snd_stream_start(stream);
236
puts("Press <ENTER> to exit playing");
237
fgets(s, sizeof(s), stdin);
239
pj_snd_stream_stop(stream);
240
pj_snd_stream_close(stream);
246
pj_snd_stream_stop(stream);
247
pj_snd_stream_close(stream);
251
static int create_ses_by_remote_sdp(int local_port, char *sdp)
253
pj_media_session_t *ses = NULL;
254
pjsdp_session_desc *sdp_ses;
255
pj_media_sock_info skinfo;
258
const pj_media_stream_info *info[2];
261
pool = pj_pool_create(pf, "sdp", 1024, 0, NULL);
263
PJ_LOG(1,(THIS_FILE, "Unable to create pool"));
267
pj_bzero(&skinfo, sizeof(skinfo));
268
skinfo.rtp_sock = skinfo.rtcp_sock = pj_sock_socket(pj_AF_INET(), pj_SOCK_DGRAM(), 0, 0);
269
if (skinfo.rtp_sock == PJ_INVALID_SOCKET) {
270
PJ_LOG(1,(THIS_FILE, "Unable to create socket"));
274
pj_sockaddr_init2(&skinfo.rtp_addr_name, "0.0.0.0", local_port);
275
if (pj_sock_bind(skinfo.rtp_sock, (struct pj_sockaddr*)&skinfo.rtp_addr_name, sizeof(pj_sockaddr_in)) != 0) {
276
PJ_LOG(1,(THIS_FILE, "Unable to bind socket"));
280
sdp_ses = pjsdp_parse(sdp, strlen(sdp), pool);
282
PJ_LOG(1,(THIS_FILE, "Error parsing SDP"));
286
ses = pj_media_session_create_from_sdp(mm, sdp_ses, &skinfo);
288
PJ_LOG(1,(THIS_FILE, "Unable to create session from SDP"));
292
if (pj_media_session_activate(ses) != 0) {
293
PJ_LOG(1,(THIS_FILE, "Error activating session"));
297
count = pj_media_session_enum_streams(ses, 2, info);
298
printf("\nDumping streams: \n");
299
for (i=0; i<count; ++i) {
303
switch (info[i]->dir) {
304
case PJMEDIA_DIR_NONE:
305
dir = "- NONE -"; break;
306
case PJMEDIA_DIR_ENCODING:
307
dir = "SENDONLY"; break;
308
case PJMEDIA_DIR_DECODING:
309
dir = "RECVONLY"; break;
310
case PJMEDIA_DIR_ENCODING_DECODING:
311
dir = "SENDRECV"; break;
313
dir = "?UNKNOWN"; break;
316
local_ip = pj_sockaddr_get_str_addr(&info[i]->sock_info.rtp_addr_name);
318
printf(" Stream %d: %.*s %s local=%s:%d remote=%.*s:%d\n",
319
i, info[i]->type.slen, info[i]->type.ptr,
321
local_ip, pj_sockaddr_get_port(&info[i]->sock_info.rtp_addr_name),
322
info[i]->rem_addr.slen, info[i]->rem_addr.ptr, info[i]->rem_port);
325
puts("Press <ENTER> to quit");
326
fgets(s, sizeof(s), stdin);
328
pj_media_session_destroy(ses);
329
pj_sock_close(skinfo.rtp_sock);
330
pj_pool_release(pool);
336
pj_media_session_destroy(ses);
337
if (skinfo.rtp_sock != PJ_INVALID_SOCKET)
338
pj_sock_close(skinfo.rtp_sock);
340
pj_pool_release(pool);
344
#if WRITE_ORIGINAL_PCM
345
static pj_status_t convert(const char *src, const char *dst)
349
struct pjmedia_frame in, out;
351
fhnd_pcm = fopen(src, "rb");
354
fhnd = fopen(dst, "wb");
358
while (fread(pcm, 320, 1, fhnd_pcm) == 1) {
360
in.type = PJMEDIA_FRAME_TYPE_AUDIO;
365
if (codec->op->encode(codec, &in, 160, &out) != 0)
368
if (fwrite(frame, out.size, 1, fhnd) != 1)
379
static void usage(const char *exe)
381
printf("Usage: %s <command> <file>\n", exe);
383
puts(" <command> play|record|send|recv");
386
int main(int argc, char *argv[])
397
if (stricmp(argv[1], "record")==0) {
398
record_file("FILE.PCM");
399
} else if (stricmp(argv[1], "play")==0) {
400
play_file("FILE.PCM");
401
} else if (stricmp(argv[1], "send")==0) {
402
create_ses_by_remote_sdp(4002, listener_sdp);
403
} else if (stricmp(argv[1], "recv")==0) {
404
create_ses_by_remote_sdp(4000, talker_sdp);