Changeset 3841 for pjproject/trunk/pjsip-apps/src/samples/pcaputil.c

Timestamp:

Oct 24, 2011 9:28:13 AM (13 years ago)

Author:

ming

Message:

Re #1395: Backport of PJSIP 1.x branch into PJSIP 2.0 trunk

• Backport of r3557:r3832

TODO: ticket #1268 (Option for automatic/manual sending of RTCP SDES/BYE for the stream) for video stream.

Location:

pjproject/trunk

Files:

• . (modified) (1 prop)
• pjsip-apps/src/samples/pcaputil.c (modified) (17 diffs)

pjproject/trunk/pjsip-apps/src/samples/pcaputil.c

@@ -26 +26 @@
 "pcaputil [options] INPUT OUTPUT\n"
 "\n"
-"  Convert captured RTP packets in PCAP file to WAV or stream it\n"
-"  to remote destination.\n"
-"\n"
-"INPUT is the PCAP file name/path\n"
+"  Convert captured RTP packets in PCAP file to WAV file or play it\n"
+"  to audio device.\n"
+"\n"
+"  INPUT  is the PCAP file name/path.\n"
+"  OUTPUT is the WAV file name/path to store the output, or set to \"-\",\n"
+"         to play the output to audio device. The program will decode\n"
+"         the RTP contents using codec that is available in PJMEDIA,\n"
+"         and optionally decrypt the content using the SRTP crypto and\n"
+"         keys below.\n"
 "\n"
 "Options to filter packets from PCAP file:\n"
@@ -38 +43 @@
 "  --dst-port=port        Only include packets destined to this port number\n"
 "\n"
-"Options for saving to WAV file:\n"
+"Options for RTP packet processing:\n"
 ""
-"  OUTPUT is WAV file:    Set output to WAV file. The program will decode the\n"
-"                         RTP contents to the specified WAV file using codec\n"
-"                         that is available in PJMEDIA, and optionally decrypt\n"
-"                         the content using the SRTP crypto and keys below.\n"
+"  --codec=codec_id       The codec ID formatted \"name/clock-rate/channel-count\"\n"
+"                         must be specified for codec with dynamic PT,\n"
+"                         e.g: \"Speex/8000\"\n"
 "  --srtp-crypto=TAG, -c  Set crypto to be used to decrypt SRTP packets. Valid\n"
 "                         tags are: \n"
@@ -50 +54 @@
 "  --srtp-key=KEY, -k     Set the base64 key to decrypt SRTP packets.\n"
 "\n"
+"Options for playing to audio device:\n"
+""
+"  --play-dev-id=dev_id   Audio device ID for playback.\n"
+"\n"
 "  Example:\n"
 "    pcaputil file.pcap output.wav\n"
 "    pcaputil -c AES_CM_128_HMAC_SHA1_80 \\\n"
-"              -k VLDONbsbGl2Puqy+0PV7w/uGfpSPKFevDpxGsxN3 \\\n"
-"              file.pcap output.wav\n"
-"\n"
-"Remote streaming is not supported yet."
+"             -k VLDONbsbGl2Puqy+0PV7w/uGfpSPKFevDpxGsxN3 \\\n"
+"             file.pcap output.wav\n"
+"\n"
 ;
 
@@ -67 +74 @@
     pjmedia_port        *wav;
     pjmedia_codec       *codec;
+    pjmedia_aud_stream  *aud_strm;
     unsigned             pt;
     pjmedia_transport   *srtp;
@@ -73 +81 @@
 } app;
 
-static void err_exit(const char *title, pj_status_t status)
-{
-    if (status != PJ_SUCCESS) {
-        char errmsg[PJ_ERR_MSG_SIZE];
-        pj_strerror(status, errmsg, sizeof(errmsg));
-        printf("Error: %s: %s\n", title, errmsg);
-    } else {
-        printf("Error: %s\n", title);
-    }
-
+
+static void cleanup()
+{
     if (app.srtp) pjmedia_transport_close(app.srtp);
     if (app.wav) {
@@ -103 +104 @@
         pjmedia_codec_mgr_dealloc_codec(cmgr, app.codec);
     }
+    if (app.aud_strm) {
+        pjmedia_aud_stream_stop(app.aud_strm);
+        pjmedia_aud_stream_destroy(app.aud_strm);
+    }
     if (app.mept) pjmedia_endpt_destroy(app.mept);
     if (app.pool) pj_pool_release(app.pool);
     pj_caching_pool_destroy(&app.cp);
     pj_shutdown();
-
+}
+
+static void err_exit(const char *title, pj_status_t status)
+{
+    if (status != PJ_SUCCESS) {
+        char errmsg[PJ_ERR_MSG_SIZE];
+        pj_strerror(status, errmsg, sizeof(errmsg));
+        printf("Error: %s: %s\n", title, errmsg);
+    } else {
+        printf("Error: %s\n", title);
+    }
+    cleanup();
     exit(1);
 }
@@ -213 +229 @@
 }
 
-static void pcap2wav(const char *wav_filename, const pj_str_t *srtp_crypto,
+pjmedia_frame play_frm;
+static pj_bool_t play_frm_copied, play_frm_ready;
+
+static pj_status_t wait_play(pjmedia_frame *f)
+{
+    play_frm_copied = PJ_FALSE;
+    play_frm = *f;
+    play_frm_ready = PJ_TRUE;
+    while (!play_frm_copied) {
+        pj_thread_sleep(1);
+    }
+    play_frm_ready = PJ_FALSE;
+
+    return PJ_SUCCESS;
+}
+
+static pj_status_t play_cb(void *user_data, pjmedia_frame *f)
+{
+    PJ_UNUSED_ARG(user_data);
+
+    if (!play_frm_ready) {
+        PJ_LOG(3, ("play_cb()", "Warning! Play frame not ready")); 
+        return PJ_SUCCESS;
+    }
+
+    pj_memcpy(f->buf, play_frm.buf, play_frm.size);
+    f->size = play_frm.size;
+
+    play_frm_copied = PJ_TRUE;
+    return PJ_SUCCESS;
+}
+
+static void pcap2wav(const pj_str_t *codec,
+                     const pj_str_t *wav_filename,
+                     pjmedia_aud_dev_index dev_id,
+                     const pj_str_t *srtp_crypto,
                      const pj_str_t *srtp_key)
 {
+    const pj_str_t WAV = {".wav", 4};
     struct pkt
     {
@@ -256 +308 @@
     /* Get codec info and param for the specified payload type */
     app.pt = pkt0.rtp->pt;
-    T( pjmedia_codec_mgr_get_codec_info(cmgr, pkt0.rtp->pt, &ci) );
+    if (app.pt >=0 && app.pt < 96) {
+        T( pjmedia_codec_mgr_get_codec_info(cmgr, pkt0.rtp->pt, &ci) );
+    } else {
+        unsigned cnt = 2;
+        const pjmedia_codec_info *info[2];
+        T( pjmedia_codec_mgr_find_codecs_by_id(cmgr, codec, &cnt, 
+                                               info, NULL) );
+        if (cnt != 1)
+            err_exit("Codec ID must be specified and unique!", 0);
+
+        ci = info[0];
+    }
     T( pjmedia_codec_mgr_get_default_param(cmgr, ci, &param) );
 
@@ -264 +327 @@
     T( pjmedia_codec_open(app.codec, &param) );
 
-    /* Open WAV file */
+    /* Init audio device or WAV file */
     samples_per_frame = ci->clock_rate * param.info.frm_ptime / 1000;
-    T( pjmedia_wav_writer_port_create(app.pool, wav_filename,
-                                      ci->clock_rate, ci->channel_cnt,
-                                      samples_per_frame,
-                                      param.info.pcm_bits_per_sample, 0, 0,
-                                      &app.wav) );
+    if (pj_strcmp2(wav_filename, "-") == 0) {
+        pjmedia_aud_param aud_param;
+
+        /* Open audio device */
+        T( pjmedia_aud_dev_default_param(dev_id, &aud_param) );
+        aud_param.dir = PJMEDIA_DIR_PLAYBACK;
+        aud_param.channel_count = ci->channel_cnt;
+        aud_param.clock_rate = ci->clock_rate;
+        aud_param.samples_per_frame = samples_per_frame;
+        T( pjmedia_aud_stream_create(&aud_param, NULL, &play_cb, 
+                                     NULL, &app.aud_strm) );
+        T( pjmedia_aud_stream_start(app.aud_strm) );
+    } else if (pj_stristr(wav_filename, &WAV)) {
+        /* Open WAV file */
+        T( pjmedia_wav_writer_port_create(app.pool, wav_filename->ptr,
+                                          ci->clock_rate, ci->channel_cnt,
+                                          samples_per_frame,
+                                          param.info.pcm_bits_per_sample, 0, 0,
+                                          &app.wav) );
+    } else {
+        err_exit("invalid output file", PJ_EINVAL);
+    }
 
     /* Loop reading PCAP and writing WAV file */
@@ -299 +379 @@
             T( pjmedia_codec_decode(app.codec, &frames[i], pcm_frame.size, 
                                      &pcm_frame) );
-            T( pjmedia_port_put_frame(app.wav, &pcm_frame) );
+            if (app.wav) {
+                T( pjmedia_port_put_frame(app.wav, &pcm_frame) );
+            }
+            if (app.aud_strm) {
+                T( wait_play(&pcm_frame) );
+            }
             samples_cnt += samples_per_frame;
         }
@@ -322 +407 @@
             }
 
-            T( pjmedia_port_put_frame(app.wav, &pcm_frame) );
+            if (app.wav) {
+                T( pjmedia_port_put_frame(app.wav, &pcm_frame) );
+            }
+            if (app.aud_strm) {
+                T( wait_play(&pcm_frame) );
+            }
             ts_gap -= samples_per_frame;
         }
@@ -336 +426 @@
 int main(int argc, char *argv[])
 {
-    pj_str_t input, output, wav, srtp_crypto, srtp_key;
+    pj_str_t input, output, srtp_crypto, srtp_key, codec;
+    pjmedia_aud_dev_index dev_id = PJMEDIA_AUD_DEFAULT_PLAYBACK_DEV;
     pj_pcap_filter filter;
     pj_status_t status;
 
-    enum { OPT_SRC_IP = 1, OPT_DST_IP, OPT_SRC_PORT, OPT_DST_PORT };
+    enum { 
+        OPT_SRC_IP = 1, OPT_DST_IP, OPT_SRC_PORT, OPT_DST_PORT,
+        OPT_CODEC, OPT_PLAY_DEV_ID
+    };
     struct pj_getopt_option long_options[] = {
         { "srtp-crypto",    1, 0, 'c' },
@@ -348 +442 @@
         { "src-port",       1, 0, OPT_SRC_PORT },
         { "dst-port",       1, 0, OPT_DST_PORT },
+        { "codec",          1, 0, OPT_CODEC },
+        { "play-dev-id",    1, 0, OPT_PLAY_DEV_ID },
         { NULL, 0, 0, 0}
     };
@@ -355 +451 @@
 
     srtp_crypto.slen = srtp_key.slen = 0;
+    codec.slen = 0;
 
     pj_pcap_filter_default(&filter);
@@ -399 +496 @@
             filter.dst_port = pj_htons((pj_uint16_t)atoi(pj_optarg));
             break;
+        case OPT_CODEC:
+            codec = pj_str(pj_optarg);
+            break;
+        case OPT_PLAY_DEV_ID:
+            dev_id = atoi(pj_optarg);
+            break;
         default:
             puts("Error: invalid option");
@@ -418 +521 @@
     input = pj_str(argv[pj_optind]);
     output = pj_str(argv[pj_optind+1]);
-    wav = pj_str(".wav");
     
     T( pj_init() );
@@ -431 +533 @@
     T( pj_pcap_set_filter(app.pcap, &filter) );
 
-    if (pj_stristr(&output, &wav)) {
-        pcap2wav(output.ptr, &srtp_crypto, &srtp_key);
-    } else {
-        err_exit("invalid output file", PJ_EINVAL);
-    }
-
-    pjmedia_endpt_destroy(app.mept);
-    pj_pool_release(app.pool);
-    pj_caching_pool_destroy(&app.cp);
-    pj_shutdown();
+    pcap2wav(&codec, &output, dev_id, &srtp_crypto, &srtp_key);
+
+    cleanup();
     return 0;
 }