Skip to content
Snippets Groups Projects
rtp.c 41.5 KiB
Newer Older
  • Learn to ignore specific revisions
  • Mark Spencer's avatar
    Mark Spencer committed
    /*
     * Asterisk -- A telephony toolkit for Linux.
     *
     * Real-time Protocol Support
    
     * 	Supports RTP and RTCP with Symmetric RTP support for NAT
     * 	traversal
    
    Mark Spencer's avatar
    Mark Spencer committed
     * 
     * Copyright (C) 1999, Mark Spencer
     *
    
     * Mark Spencer <markster@digium.com>
    
    Mark Spencer's avatar
    Mark Spencer committed
     *
     * This program is free software, distributed under the terms of
     * the GNU General Public License
     */
    
    #include <stdio.h>
    #include <stdlib.h>
    #include <string.h>
    #include <sys/time.h>
    #include <signal.h>
    #include <errno.h>
    #include <unistd.h>
    #include <netinet/in.h>
    #include <sys/time.h>
    #include <sys/socket.h>
    #include <arpa/inet.h>
    #include <fcntl.h>
    
    #include <asterisk/rtp.h>
    
    Mark Spencer's avatar
    Mark Spencer committed
    #include <asterisk/frame.h>
    
    Mark Spencer's avatar
    Mark Spencer committed
    #include <asterisk/logger.h>
    #include <asterisk/options.h>
    #include <asterisk/channel.h>
    
    Mark Spencer's avatar
    Mark Spencer committed
    #include <asterisk/acl.h>
    
    #include <asterisk/channel.h>
    
    #include <asterisk/channel_pvt.h>
    
    #include <asterisk/config.h>
    
    #define MAX_TIMESTAMP_SKEW	640
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    #define TYPE_HIGH	 0x0
    #define TYPE_LOW	 0x1
    
    Mark Spencer's avatar
    Mark Spencer committed
    #define TYPE_SILENCE	 0x2
    #define TYPE_DONTSEND	 0x3
    
    Mark Spencer's avatar
    Mark Spencer committed
    #define TYPE_MASK	 0x3
    
    
    static int dtmftimeout = 3000;	/* 3000 samples */
    
    static int rtpstart = 0;
    static int rtpend = 0;
    
    
    /* The value of each payload format mapping: */
    
    struct rtpPayloadType {
    
      int isAstFormat; 	/* whether the following code is an AST_FORMAT */
    
      int code;
    };
    
    #define MAX_RTP_PT 256
    
    
    #define FLAG_3389_WARNING (1 << 0)
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    struct ast_rtp {
    	int s;
    	char resp;
    	struct ast_frame f;
    
    	unsigned char rawdata[8192 + AST_FRIENDLY_OFFSET];
    
    Mark Spencer's avatar
    Mark Spencer committed
    	unsigned int ssrc;
    	unsigned int lastts;
    	unsigned int lastrxts;
    
    	unsigned int lastividtimestamp;
    	unsigned int lastovidtimestamp;
    
    	unsigned int lasteventseqn;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	int lasttxformat;
    
    	int lastrxformat;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	int dtmfcount;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	unsigned int dtmfduration;
    
    	int nat;
    
    	int flags;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	struct sockaddr_in us;
    	struct sockaddr_in them;
    	struct timeval rxcore;
    	struct timeval txcore;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	struct timeval dtmfmute;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	struct ast_smoother *smoother;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	int *ioid;
    	unsigned short seqno;
    	struct sched_context *sched;
    	struct io_context *io;
    	void *data;
    	ast_rtp_callback callback;
    
        struct rtpPayloadType current_RTP_PT[MAX_RTP_PT];
    
        int rtp_lookup_code_cache_isAstFormat;	/* a cache for the result of rtp_lookup_code(): */
    
        int rtp_lookup_code_cache_code;
        int rtp_lookup_code_cache_result;
    
    	struct ast_rtcp *rtcp;
    };
    
    struct ast_rtcp {
    	int s;		/* Socket */
    	struct sockaddr_in us;
    	struct sockaddr_in them;
    
    static struct ast_rtp_protocol *protos = NULL;
    
    
    int ast_rtp_fd(struct ast_rtp *rtp)
    {
    	return rtp->s;
    }
    
    int ast_rtcp_fd(struct ast_rtp *rtp)
    {
    	if (rtp->rtcp)
    		return rtp->rtcp->s;
    	return -1;
    }
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    static int g723_len(unsigned char buf)
    {
    	switch(buf & TYPE_MASK) {
    
    Mark Spencer's avatar
    Mark Spencer committed
    	case TYPE_DONTSEND:
    		return 0;
    		break;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	case TYPE_SILENCE:
    		return 4;
    		break;
    	case TYPE_HIGH:
    		return 24;
    		break;
    	case TYPE_LOW:
    		return 20;
    		break;
    	default:
    		ast_log(LOG_WARNING, "Badly encoded frame (%d)\n", buf & TYPE_MASK);
    	}
    	return -1;
    }
    
    static int g723_samples(unsigned char *buf, int maxlen)
    {
    	int pos = 0;
    	int samples = 0;
    	int res;
    	while(pos < maxlen) {
    		res = g723_len(buf[pos]);
    
    		if (res <= 0)
    
    Mark Spencer's avatar
    Mark Spencer committed
    			break;
    		samples += 240;
    		pos += res;
    	}
    	return samples;
    }
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    void ast_rtp_set_data(struct ast_rtp *rtp, void *data)
    {
    	rtp->data = data;
    }
    
    void ast_rtp_set_callback(struct ast_rtp *rtp, ast_rtp_callback callback)
    {
    	rtp->callback = callback;
    }
    
    
    void ast_rtp_setnat(struct ast_rtp *rtp, int nat)
    {
    	rtp->nat = nat;
    }
    
    
    static struct ast_frame *send_dtmf(struct ast_rtp *rtp)
    
    Mark Spencer's avatar
    Mark Spencer committed
    {
    
    Mark Spencer's avatar
    Mark Spencer committed
    	struct timeval tv;
    	static struct ast_frame null_frame = { AST_FRAME_NULL, };
    
    	char iabuf[INET_ADDRSTRLEN];
    
    Mark Spencer's avatar
    Mark Spencer committed
    	gettimeofday(&tv, NULL);
    	if ((tv.tv_sec < rtp->dtmfmute.tv_sec) ||
    	    ((tv.tv_sec == rtp->dtmfmute.tv_sec) && (tv.tv_usec < rtp->dtmfmute.tv_usec))) {
    
    		ast_log(LOG_DEBUG, "Ignore potential DTMF echo from '%s'\n", ast_inet_ntoa(iabuf, sizeof(iabuf), rtp->them.sin_addr));
    
    Mark Spencer's avatar
    Mark Spencer committed
    		rtp->resp = 0;
    		rtp->dtmfduration = 0;
    		return &null_frame;
    	}
    
    	ast_log(LOG_DEBUG, "Sending dtmf: %d (%c), at %s\n", rtp->resp, rtp->resp, ast_inet_ntoa(iabuf, sizeof(iabuf), rtp->them.sin_addr));
    
    Mark Spencer's avatar
    Mark Spencer committed
    	rtp->f.frametype = AST_FRAME_DTMF;
    	rtp->f.subclass = rtp->resp;
    	rtp->f.datalen = 0;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	rtp->f.samples = 0;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	rtp->f.mallocd = 0;
    	rtp->f.src = "RTP";
    	rtp->resp = 0;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	rtp->dtmfduration = 0;
    
    	return &rtp->f;
    
    Mark Spencer's avatar
    Mark Spencer committed
    static struct ast_frame *process_cisco_dtmf(struct ast_rtp *rtp, unsigned char *data, int len)
    {
    	unsigned int event;
    	char resp = 0;
    	struct ast_frame *f = NULL;
    	event = ntohl(*((unsigned int *)(data)));
    	event &= 0x001F;
    #if 0
    	printf("Cisco Digit: %08x (len = %d)\n", event, len);
    #endif	
    	if (event < 10) {
    		resp = '0' + event;
    	} else if (event < 11) {
    		resp = '*';
    	} else if (event < 12) {
    		resp = '#';
    	} else if (event < 16) {
    		resp = 'A' + (event - 12);
    	}
    	if (rtp->resp && (rtp->resp != resp)) {
    		f = send_dtmf(rtp);
    	}
    	rtp->resp = resp;
    	rtp->dtmfcount = dtmftimeout;
    	return f;
    }
    
    
    static struct ast_frame *process_rfc2833(struct ast_rtp *rtp, unsigned char *data, int len)
    
    Mark Spencer's avatar
    Mark Spencer committed
    {
    	unsigned int event;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	unsigned int event_end;
    	unsigned int duration;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	char resp = 0;
    
    	struct ast_frame *f = NULL;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	event = ntohl(*((unsigned int *)(data)));
    	event >>= 24;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	event_end = ntohl(*((unsigned int *)(data)));
    	event_end <<= 8;
    	event_end >>= 24;
    	duration = ntohl(*((unsigned int *)(data)));
    	duration &= 0xFFFF;
    
    Mark Spencer's avatar
    Mark Spencer committed
    #if 0
    	printf("Event: %08x (len = %d)\n", event, len);
    #endif	
    	if (event < 10) {
    		resp = '0' + event;
    	} else if (event < 11) {
    		resp = '*';
    	} else if (event < 12) {
    		resp = '#';
    	} else if (event < 16) {
    		resp = 'A' + (event - 12);
    	}
    	if (rtp->resp && (rtp->resp != resp)) {
    
    		f = send_dtmf(rtp);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	}
    
    Mark Spencer's avatar
    Mark Spencer committed
    	else if(event_end & 0x80)
    	{
    
    Mark Spencer's avatar
    Mark Spencer committed
    		if (rtp->resp) {
    			f = send_dtmf(rtp);
    			rtp->resp = 0;
    		}
    
    Mark Spencer's avatar
    Mark Spencer committed
    		resp = 0;
    		duration = 0;
    	}
    	else if(rtp->dtmfduration && (duration < rtp->dtmfduration))
    	{
    		f = send_dtmf(rtp);
    	}
    
    Mark Spencer's avatar
    Mark Spencer committed
    	if (!(event_end & 0x80))
    		rtp->resp = resp;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	rtp->dtmfcount = dtmftimeout;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	rtp->dtmfduration = duration;
    
    	return f;
    
    static struct ast_frame *process_rfc3389(struct ast_rtp *rtp, unsigned char *data, int len)
    {
    	struct ast_frame *f = NULL;
    	/* Convert comfort noise into audio with various codecs.  Unfortunately this doesn't
    	   totally help us out becuase we don't have an engine to keep it going and we are not
    	   guaranteed to have it every 20ms or anything */
    #if 0
    	printf("RFC3389: %d bytes, format is %d\n", len, rtp->lastrxformat);
    #endif	
    
    	if (!(rtp->flags & FLAG_3389_WARNING)) {
    		ast_log(LOG_NOTICE, "RFC3389 support incomplete.  Turn off on client if possible\n");
    		rtp->flags |= FLAG_3389_WARNING;
    	}
    
    	if (!rtp->lastrxformat)
    		return 	NULL;
    	switch(rtp->lastrxformat) {
    	case AST_FORMAT_ULAW:
    		rtp->f.frametype = AST_FRAME_VOICE;
    		rtp->f.subclass = AST_FORMAT_ULAW;
    		rtp->f.datalen = 160;
    		rtp->f.samples = 160;
    		memset(rtp->f.data, 0x7f, rtp->f.datalen);
    		f = &rtp->f;
    		break;
    	case AST_FORMAT_ALAW:
    		rtp->f.frametype = AST_FRAME_VOICE;
    		rtp->f.subclass = AST_FORMAT_ALAW;
    		rtp->f.datalen = 160;
    		rtp->f.samples = 160;
    		memset(rtp->f.data, 0x7e, rtp->f.datalen); /* XXX Is this right? XXX */
    		f = &rtp->f;
    		break;
    	case AST_FORMAT_SLINEAR:
    		rtp->f.frametype = AST_FRAME_VOICE;
    		rtp->f.subclass = AST_FORMAT_SLINEAR;
    		rtp->f.datalen = 320;
    		rtp->f.samples = 160;
    		memset(rtp->f.data, 0x00, rtp->f.datalen);
    		f = &rtp->f;
    		break;
    	default:
    		ast_log(LOG_NOTICE, "Don't know how to handle RFC3389 for receive codec %d\n", rtp->lastrxformat);
    	}
    	return f;
    }
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    static int rtpread(int *id, int fd, short events, void *cbdata)
    {
    	struct ast_rtp *rtp = cbdata;
    
    	struct ast_frame *f;
    	f = ast_rtp_read(rtp);
    	if (f) {
    		if (rtp->callback)
    			rtp->callback(rtp, f, rtp->data);
    	}
    	return 1;
    }
    
    
    struct ast_frame *ast_rtcp_read(struct ast_rtp *rtp)
    {
    	static struct ast_frame null_frame = { AST_FRAME_NULL, };
    	int len;
    	int hdrlen = 8;
    	int res;
    	struct sockaddr_in sin;
    	unsigned int rtcpdata[1024];
    
    	char iabuf[INET_ADDRSTRLEN];
    
    	
    	if (!rtp->rtcp)
    		return &null_frame;
    
    	len = sizeof(sin);
    	
    	res = recvfrom(rtp->rtcp->s, rtcpdata, sizeof(rtcpdata),
    					0, (struct sockaddr *)&sin, &len);
    	
    	if (res < 0) {
    
    		if (errno == EAGAIN)
    			ast_log(LOG_NOTICE, "RTP: Received packet with bad UDP checksum\n");
    		else
    			ast_log(LOG_WARNING, "RTP Read error: %s\n", strerror(errno));
    
    		if (errno == EBADF)
    			CRASH;
    		return &null_frame;
    	}
    
    	if (res < hdrlen) {
    		ast_log(LOG_WARNING, "RTP Read too short\n");
    		return &null_frame;
    	}
    
    	if (rtp->nat) {
    		/* Send to whoever sent to us */
    		if ((rtp->rtcp->them.sin_addr.s_addr != sin.sin_addr.s_addr) ||
    		    (rtp->rtcp->them.sin_port != sin.sin_port)) {
    			memcpy(&rtp->them, &sin, sizeof(rtp->them));
    
    			ast_log(LOG_DEBUG, "RTP NAT: Using address %s:%d\n", ast_inet_ntoa(iabuf, sizeof(iabuf), rtp->rtcp->them.sin_addr), ntohs(rtp->rtcp->them.sin_port));
    
    Mark Spencer's avatar
    Mark Spencer committed
    	if (option_debug)
    		ast_log(LOG_DEBUG, "Got RTCP report of %d bytes\n", res);
    
    static void calc_rxstamp(struct timeval *tv, struct ast_rtp *rtp, unsigned int timestamp, int mark)
    
    	if ((!rtp->rxcore.tv_sec && !rtp->rxcore.tv_usec) || mark) {
    
    		gettimeofday(&rtp->rxcore, NULL);
    
    		rtp->rxcore.tv_sec -= timestamp / 8000;
    
    		rtp->rxcore.tv_usec -= (timestamp % 8000) * 125;
    
    		/* Round to 20ms for nice, pretty timestamps */
    
    		rtp->rxcore.tv_usec -= rtp->rxcore.tv_usec % 20000;
    
    		if (rtp->rxcore.tv_usec < 0) {
    			/* Adjust appropriately if necessary */
    			rtp->rxcore.tv_usec += 1000000;
    			rtp->rxcore.tv_sec -= 1;
    		}
    	}
    	tv->tv_sec = rtp->rxcore.tv_sec + timestamp / 8000;
    	tv->tv_usec = rtp->rxcore.tv_usec + (timestamp % 8000) * 125;
    	if (tv->tv_usec >= 1000000) {
    		tv->tv_usec -= 1000000;
    		tv->tv_sec += 1;
    	}
    }
    
    
    struct ast_frame *ast_rtp_read(struct ast_rtp *rtp)
    {
    
    Mark Spencer's avatar
    Mark Spencer committed
    	int res;
    	struct sockaddr_in sin;
    	int len;
    	unsigned int seqno;
    	int payloadtype;
    	int hdrlen = 12;
    
    	char iabuf[INET_ADDRSTRLEN];
    
    Mark Spencer's avatar
    Mark Spencer committed
    	unsigned int timestamp;
    	unsigned int *rtpheader;
    
    	static struct ast_frame *f, null_frame = { AST_FRAME_NULL, };
    
    	struct rtpPayloadType rtpPT;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	
    	len = sizeof(sin);
    	
    
    	/* Cache where the header will go */
    
    	res = recvfrom(rtp->s, rtp->rawdata + AST_FRIENDLY_OFFSET, sizeof(rtp->rawdata) - AST_FRIENDLY_OFFSET,
    
    Mark Spencer's avatar
    Mark Spencer committed
    					0, (struct sockaddr *)&sin, &len);
    
    
    	rtpheader = (unsigned int *)(rtp->rawdata + AST_FRIENDLY_OFFSET);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	if (res < 0) {
    
    		if (errno == EAGAIN)
    			ast_log(LOG_NOTICE, "RTP: Received packet with bad UDP checksum\n");
    		else
    			ast_log(LOG_WARNING, "RTP Read error: %s\n", strerror(errno));
    
    Mark Spencer's avatar
    Mark Spencer committed
    		if (errno == EBADF)
    			CRASH;
    
    		return &null_frame;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	}
    	if (res < hdrlen) {
    		ast_log(LOG_WARNING, "RTP Read too short\n");
    
    		return &null_frame;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	}
    
    
    	/* Ignore if the other side hasn't been given an address
    	   yet.  */
    	if (!rtp->them.sin_addr.s_addr || !rtp->them.sin_port)
    		return &null_frame;
    
    
    	if (rtp->nat) {
    		/* Send to whoever sent to us */
    
    		if ((rtp->them.sin_addr.s_addr != sin.sin_addr.s_addr) ||
    		    (rtp->them.sin_port != sin.sin_port)) {
    			memcpy(&rtp->them, &sin, sizeof(rtp->them));
    
    			ast_log(LOG_DEBUG, "RTP NAT: Using address %s:%d\n", ast_inet_ntoa(iabuf, sizeof(iabuf), rtp->them.sin_addr), ntohs(rtp->them.sin_port));
    
    Mark Spencer's avatar
    Mark Spencer committed
    	/* Get fields */
    	seqno = ntohl(rtpheader[0]);
    	payloadtype = (seqno & 0x7f0000) >> 16;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	seqno &= 0xffff;
    	timestamp = ntohl(rtpheader[1]);
    
    Mark Spencer's avatar
    Mark Spencer committed
    #if 0
    
    	printf("Got RTP packet from %s:%d (type %d, seq %d, ts %d, len = %d)\n", ast_inet_ntoa(iabuf, sizeof(iabuf), sin.sin_addr), ntohs(sin.sin_port), payloadtype, seqno, timestamp,res - hdrlen);
    
    Mark Spencer's avatar
    Mark Spencer committed
    #endif	
    
    	rtpPT = ast_rtp_lookup_pt(rtp, payloadtype);
    
    	if (!rtpPT.isAstFormat) {
    
    	  /* This is special in-band data that's not one of our codecs */
    
    	  if (rtpPT.code == AST_RTP_DTMF) {
    	    /* It's special -- rfc2833 process it */
    
    	    if (rtp->lasteventseqn <= seqno) {
    	      f = process_rfc2833(rtp, rtp->rawdata + AST_FRIENDLY_OFFSET + hdrlen, res - hdrlen);
    	      rtp->lasteventseqn = seqno;
    	    }
    
    	    if (f) return f; else return &null_frame;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	  } else if (rtpPT.code == AST_RTP_CISCO_DTMF) {
    	    /* It's really special -- process it the Cisco way */
    
    	    if (rtp->lasteventseqn <= seqno) {
    	      f = process_cisco_dtmf(rtp, rtp->rawdata + AST_FRIENDLY_OFFSET + hdrlen, res - hdrlen);
    	      rtp->lasteventseqn = seqno;
    	    }
    
    Mark Spencer's avatar
    Mark Spencer committed
    	    if (f) return f; else return &null_frame;
    
    	  } else if (rtpPT.code == AST_RTP_CN) {
    	    /* Comfort Noise */
    	    f = process_rfc3389(rtp, rtp->rawdata + AST_FRIENDLY_OFFSET + hdrlen, res - hdrlen);
    	    if (f) return f; else return &null_frame;
    	  } else {
    	    ast_log(LOG_NOTICE, "Unknown RTP codec %d received\n", payloadtype);
    	    return &null_frame;
    	  }
    	}
    	rtp->f.subclass = rtpPT.code;
    
    	if (rtp->f.subclass < AST_FORMAT_MAX_AUDIO)
    		rtp->f.frametype = AST_FRAME_VOICE;
    	else
    		rtp->f.frametype = AST_FRAME_VIDEO;
    
    	rtp->lastrxformat = rtp->f.subclass;
    
    Mark Spencer's avatar
    Mark Spencer committed
    
    	if (!rtp->lastrxts)
    		rtp->lastrxts = timestamp;
    
    	if (rtp->dtmfcount) {
    #if 0
    		printf("dtmfcount was %d\n", rtp->dtmfcount);
    #endif		
    		rtp->dtmfcount -= (timestamp - rtp->lastrxts);
    		if (rtp->dtmfcount < 0)
    			rtp->dtmfcount = 0;
    #if 0
    		if (dtmftimeout != rtp->dtmfcount)
    			printf("dtmfcount is %d\n", rtp->dtmfcount);
    #endif
    	}
    	rtp->lastrxts = timestamp;
    
    	/* Send any pending DTMF */
    	if (rtp->resp && !rtp->dtmfcount) {
    
    		ast_log(LOG_DEBUG, "Sending pending DTMF\n");
    
    		return send_dtmf(rtp);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	}
    	rtp->f.mallocd = 0;
    
    	rtp->f.datalen = res - hdrlen;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	rtp->f.data = rtp->rawdata + hdrlen + AST_FRIENDLY_OFFSET;
    	rtp->f.offset = hdrlen + AST_FRIENDLY_OFFSET;
    
    	if (rtp->f.subclass < AST_FORMAT_MAX_AUDIO) {
    		switch(rtp->f.subclass) {
    		case AST_FORMAT_ULAW:
    		case AST_FORMAT_ALAW:
    			rtp->f.samples = rtp->f.datalen;
    			break;
    		case AST_FORMAT_SLINEAR:
    			rtp->f.samples = rtp->f.datalen / 2;
    			break;
    		case AST_FORMAT_GSM:
    			rtp->f.samples = 160 * (rtp->f.datalen / 33);
    			break;
    		case AST_FORMAT_ILBC:
    			rtp->f.samples = 240 * (rtp->f.datalen / 50);
    			break;
    		case AST_FORMAT_ADPCM:
    
    			rtp->f.samples = rtp->f.datalen * 2;
    			break;
    		case AST_FORMAT_G729A:
    			rtp->f.samples = rtp->f.datalen * 8;
    			break;
    		case AST_FORMAT_G723_1:
    			rtp->f.samples = g723_samples(rtp->f.data, rtp->f.datalen);
    			break;
    		case AST_FORMAT_SPEEX:
    		        rtp->f.samples = 160;
    
    			/* assumes that the RTP packet contained one Speex frame */
    
    			ast_log(LOG_NOTICE, "Unable to calculate samples for format %s\n", ast_getformatname(rtp->f.subclass));
    
    		calc_rxstamp(&rtp->f.delivery, rtp, timestamp, mark);
    
    	} else {
    		/* Video -- samples is # of samples vs. 90000 */
    
    Mark Spencer's avatar
    Mark Spencer committed
    		if (!rtp->lastividtimestamp)
    			rtp->lastividtimestamp = timestamp;
    
    		rtp->f.samples = timestamp - rtp->lastividtimestamp;
    		rtp->lastividtimestamp = timestamp;
    
    		rtp->f.delivery.tv_sec = 0;
    		rtp->f.delivery.tv_usec = 0;
    
    		if (mark)
    			rtp->f.subclass |= 0x1;
    		
    
    Mark Spencer's avatar
    Mark Spencer committed
    	}
    	rtp->f.src = "RTP";
    
    	return &rtp->f;
    
    /* The following array defines the MIME Media type (and subtype) for each
       of our codecs, or RTP-specific data type. */
    
    Mark Spencer's avatar
    Mark Spencer committed
    static struct {
    
      struct rtpPayloadType payloadType;
      char* type;
      char* subtype;
    } mimeTypes[] = {
      {{1, AST_FORMAT_G723_1}, "audio", "G723"},
      {{1, AST_FORMAT_GSM}, "audio", "GSM"},
      {{1, AST_FORMAT_ULAW}, "audio", "PCMU"},
      {{1, AST_FORMAT_ALAW}, "audio", "PCMA"},
    
      {{1, AST_FORMAT_G726}, "audio", "G726-32"},
    
      {{1, AST_FORMAT_ADPCM}, "audio", "DVI4"},
      {{1, AST_FORMAT_SLINEAR}, "audio", "L16"},
      {{1, AST_FORMAT_LPC10}, "audio", "LPC"},
      {{1, AST_FORMAT_G729A}, "audio", "G729"},
      {{1, AST_FORMAT_SPEEX}, "audio", "SPEEX"},
    
    Mark Spencer's avatar
    Mark Spencer committed
      {{1, AST_FORMAT_ILBC}, "audio", "iLBC"},
    
      {{0, AST_RTP_DTMF}, "audio", "telephone-event"},
    
      {{0, AST_RTP_CISCO_DTMF}, "audio", "cisco-telephone-event"},
    
      {{0, AST_RTP_CN}, "audio", "CN"},
      {{1, AST_FORMAT_JPEG}, "video", "JPEG"},
      {{1, AST_FORMAT_PNG}, "video", "PNG"},
      {{1, AST_FORMAT_H261}, "video", "H261"},
      {{1, AST_FORMAT_H263}, "video", "H263"},
    
    /* Static (i.e., well-known) RTP payload types for our "AST_FORMAT..."s:
       also, our own choices for dynamic payload types.  This is our master
       table for transmission */
    
    static struct rtpPayloadType static_RTP_PT[MAX_RTP_PT] = {
      [0] = {1, AST_FORMAT_ULAW},
    
      [2] = {1, AST_FORMAT_G726}, /* Technically this is G.721, but if Cisco can do it, so can we... */
    
      [3] = {1, AST_FORMAT_GSM},
      [4] = {1, AST_FORMAT_G723_1},
    
      [5] = {1, AST_FORMAT_ADPCM}, /* 8 kHz */
      [6] = {1, AST_FORMAT_ADPCM}, /* 16 kHz */
    
      [7] = {1, AST_FORMAT_LPC10},
      [8] = {1, AST_FORMAT_ALAW},
    
      [10] = {1, AST_FORMAT_SLINEAR}, /* 2 channels */
      [11] = {1, AST_FORMAT_SLINEAR}, /* 1 channel */
    
      [13] = {0, AST_RTP_CN},
    
      [16] = {1, AST_FORMAT_ADPCM}, /* 11.025 kHz */
      [17] = {1, AST_FORMAT_ADPCM}, /* 22.050 kHz */
    
      [18] = {1, AST_FORMAT_G729A},
      [26] = {1, AST_FORMAT_JPEG},
      [31] = {1, AST_FORMAT_H261},
      [34] = {1, AST_FORMAT_H263},
    
    Mark Spencer's avatar
    Mark Spencer committed
      [97] = {1, AST_FORMAT_ILBC},
    
      [101] = {0, AST_RTP_DTMF},
    
      [121] = {0, AST_RTP_CISCO_DTMF}, /* Must be type 121 */
    
    void ast_rtp_pt_clear(struct ast_rtp* rtp) 
    {
    
      int i;
    
      for (i = 0; i < MAX_RTP_PT; ++i) {
        rtp->current_RTP_PT[i].isAstFormat = 0;
        rtp->current_RTP_PT[i].code = 0;
      }
    
      rtp->rtp_lookup_code_cache_isAstFormat = 0;
      rtp->rtp_lookup_code_cache_code = 0;
      rtp->rtp_lookup_code_cache_result = 0;
    
    void ast_rtp_pt_default(struct ast_rtp* rtp) 
    {
      int i;
      /* Initialize to default payload types */
      for (i = 0; i < MAX_RTP_PT; ++i) {
        rtp->current_RTP_PT[i].isAstFormat = static_RTP_PT[i].isAstFormat;
        rtp->current_RTP_PT[i].code = static_RTP_PT[i].code;
      }
    
      rtp->rtp_lookup_code_cache_isAstFormat = 0;
      rtp->rtp_lookup_code_cache_code = 0;
      rtp->rtp_lookup_code_cache_result = 0;
    }
    
    
    /* Make a note of a RTP payload type that was seen in a SDP "m=" line. */
    /* By default, use the well-known value for this type (although it may */
    /* still be set to a different value by a subsequent "a=rtpmap:" line): */
    
    void ast_rtp_set_m_type(struct ast_rtp* rtp, int pt) {
    
      if (pt < 0 || pt > MAX_RTP_PT) return; /* bogus payload type */
    
    
      if (static_RTP_PT[pt].code != 0) {
        rtp->current_RTP_PT[pt] = static_RTP_PT[pt];
      }
    } 
    
    
    /* Make a note of a RTP payload type (with MIME type) that was seen in */
    /* a SDP "a=rtpmap:" line. */
    
    void ast_rtp_set_rtpmap_type(struct ast_rtp* rtp, int pt,
    
    			 char* mimeType, char* mimeSubtype) {
      int i;
    
    
      if (pt < 0 || pt > MAX_RTP_PT) return; /* bogus payload type */
    
    
      for (i = 0; i < sizeof mimeTypes/sizeof mimeTypes[0]; ++i) {
    
        if (strcasecmp(mimeSubtype, mimeTypes[i].subtype) == 0 &&
    	strcasecmp(mimeType, mimeTypes[i].type) == 0) {
    
          rtp->current_RTP_PT[pt] = mimeTypes[i].payloadType;
          return;
        }
      }
    } 
    
    
    /* Return the union of all of the codecs that were set by rtp_set...() calls */
    /* They're returned as two distinct sets: AST_FORMATs, and AST_RTPs */
    
    void ast_rtp_get_current_formats(struct ast_rtp* rtp,
    
    			     int* astFormats, int* nonAstFormats) {
      int pt;
    
      *astFormats = *nonAstFormats = 0;
      for (pt = 0; pt < MAX_RTP_PT; ++pt) {
        if (rtp->current_RTP_PT[pt].isAstFormat) {
          *astFormats |= rtp->current_RTP_PT[pt].code;
        } else {
          *nonAstFormats |= rtp->current_RTP_PT[pt].code;
        }
      }
    
    struct rtpPayloadType ast_rtp_lookup_pt(struct ast_rtp* rtp, int pt) 
    {
      struct rtpPayloadType result;
    
      if (pt < 0 || pt > MAX_RTP_PT) {
        result.isAstFormat = result.code = 0;
    
        return result; /* bogus payload type */
    
      /* Start with the negotiated codecs */
      result = rtp->current_RTP_PT[pt];
      /* If it doesn't exist, check our static RTP type list, just in case */
      if (!result.code) 
      	result = static_RTP_PT[pt];
      return result;
    
    /* Looks up an RTP code out of our *static* outbound list */
    
    int ast_rtp_lookup_code(struct ast_rtp* rtp, int isAstFormat, int code) {
    
      if (isAstFormat == rtp->rtp_lookup_code_cache_isAstFormat &&
          code == rtp->rtp_lookup_code_cache_code) {
    
        /* Use our cached mapping, to avoid the overhead of the loop below */
    
        return rtp->rtp_lookup_code_cache_result;
      }
    
    
    	/* Check the dynamic list first */
      for (pt = 0; pt < MAX_RTP_PT; ++pt) {
        if (rtp->current_RTP_PT[pt].code == code &&
    		rtp->current_RTP_PT[pt].isAstFormat == isAstFormat) {
          rtp->rtp_lookup_code_cache_isAstFormat = isAstFormat;
          rtp->rtp_lookup_code_cache_code = code;
          rtp->rtp_lookup_code_cache_result = pt;
          return pt;
        }
      }
    
    	/* Then the static list */
    
      for (pt = 0; pt < MAX_RTP_PT; ++pt) {
    
        if (static_RTP_PT[pt].code == code &&
    		static_RTP_PT[pt].isAstFormat == isAstFormat) {
    
          rtp->rtp_lookup_code_cache_isAstFormat = isAstFormat;
          rtp->rtp_lookup_code_cache_code = code;
          rtp->rtp_lookup_code_cache_result = pt;
          return pt;
        }
      }
      return -1;
    
    Mark Spencer's avatar
    Mark Spencer committed
    }
    
    char* ast_rtp_lookup_mime_subtype(int isAstFormat, int code) {
    
      int i;
    
      for (i = 0; i < sizeof mimeTypes/sizeof mimeTypes[0]; ++i) {
        if (mimeTypes[i].payloadType.code == code &&
    	mimeTypes[i].payloadType.isAstFormat == isAstFormat) {
          return mimeTypes[i].subtype;
        }
      }
      return "";
    }
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    static int rtp_socket(void)
    {
    	int s;
    	long flags;
    	s = socket(AF_INET, SOCK_DGRAM, 0);
    	if (s > -1) {
    		flags = fcntl(s, F_GETFL);
    		fcntl(s, F_SETFL, flags | O_NONBLOCK);
    	}
    	return s;
    }
    
    
    static struct ast_rtcp *ast_rtcp_new(void)
    {
    	struct ast_rtcp *rtcp;
    	rtcp = malloc(sizeof(struct ast_rtcp));
    	if (!rtcp)
    		return NULL;
    	memset(rtcp, 0, sizeof(struct ast_rtcp));
    
    Mark Spencer's avatar
    Mark Spencer committed
    	rtcp->s = rtp_socket();
    
    	rtcp->us.sin_family = AF_INET;
    	if (rtcp->s < 0) {
    		free(rtcp);
    		ast_log(LOG_WARNING, "Unable to allocate socket: %s\n", strerror(errno));
    		return NULL;
    	}
    	return rtcp;
    }
    
    struct ast_rtp *ast_rtp_new(struct sched_context *sched, struct io_context *io, int rtcpenable, int callbackmode)
    
    Mark Spencer's avatar
    Mark Spencer committed
    {
    	struct ast_rtp *rtp;
    	int x;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	int first;
    
    	int startplace;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	rtp = malloc(sizeof(struct ast_rtp));
    	if (!rtp)
    		return NULL;
    	memset(rtp, 0, sizeof(struct ast_rtp));
    	rtp->them.sin_family = AF_INET;
    	rtp->us.sin_family = AF_INET;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	rtp->s = rtp_socket();
    
    Mark Spencer's avatar
    Mark Spencer committed
    	rtp->ssrc = rand();
    	rtp->seqno = rand() & 0xffff;
    	if (rtp->s < 0) {
    		free(rtp);
    		ast_log(LOG_WARNING, "Unable to allocate socket: %s\n", strerror(errno));
    		return NULL;
    	}
    
    	if (sched && rtcpenable) {
    		rtp->sched = sched;
    		rtp->rtcp = ast_rtcp_new();
    	}
    
    	/* Find us a place */
    	x = (rand() % (rtpend-rtpstart)) + rtpstart;
    	x = x & ~1;
    	startplace = x;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	for (;;) {
    		/* Must be an even port number by RTP spec */
    		rtp->us.sin_port = htons(x);
    
    		if (rtp->rtcp)
    			rtp->rtcp->us.sin_port = htons(x + 1);
    
    Mark Spencer's avatar
    Mark Spencer committed
    		if (!(first = bind(rtp->s, (struct sockaddr *)&rtp->us, sizeof(rtp->us))) &&
    
    			(!rtp->rtcp || !bind(rtp->rtcp->s, (struct sockaddr *)&rtp->rtcp->us, sizeof(rtp->rtcp->us))))
    
    Mark Spencer's avatar
    Mark Spencer committed
    			break;
    
    Mark Spencer's avatar
    Mark Spencer committed
    		if (!first) {
    			/* Primary bind succeeded! Gotta recreate it */
    			close(rtp->s);
    			rtp->s = rtp_socket();
    		}
    
    Mark Spencer's avatar
    Mark Spencer committed
    		if (errno != EADDRINUSE) {
    			ast_log(LOG_WARNING, "Unexpected bind error: %s\n", strerror(errno));
    			close(rtp->s);
    
    			if (rtp->rtcp) {
    				close(rtp->rtcp->s);
    				free(rtp->rtcp);
    			}
    
    Mark Spencer's avatar
    Mark Spencer committed
    			free(rtp);
    			return NULL;
    		}
    
    		x += 2;
    		if (x > rtpend)
    			x = (rtpstart + 1) & ~1;
    		if (x == startplace) {
    			ast_log(LOG_WARNING, "No RTP ports remaining\n");
    			close(rtp->s);
    
    			if (rtp->rtcp) {
    				close(rtp->rtcp->s);
    				free(rtp->rtcp);
    			}
    
    			free(rtp);
    			return NULL;
    		}
    
    Mark Spencer's avatar
    Mark Spencer committed
    	}
    
    	if (io && sched && callbackmode) {
    
    		/* Operate this one in a callback mode */
    		rtp->sched = sched;
    		rtp->io = io;
    		rtp->ioid = ast_io_add(rtp->io, rtp->s, rtpread, AST_IO_IN, rtp);
    	}
    
    	ast_rtp_pt_default(rtp);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	return rtp;
    }
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    int ast_rtp_settos(struct ast_rtp *rtp, int tos)
    {
    	int res;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	if ((res = setsockopt(rtp->s, IPPROTO_IP, IP_TOS, &tos, sizeof(tos)))) 
    
    Mark Spencer's avatar
    Mark Spencer committed
    		ast_log(LOG_WARNING, "Unable to set TOS to %d\n", tos);
    	return res;
    }
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    void ast_rtp_set_peer(struct ast_rtp *rtp, struct sockaddr_in *them)
    {
    	rtp->them.sin_port = them->sin_port;
    	rtp->them.sin_addr = them->sin_addr;
    
    	if (rtp->rtcp) {
    		rtp->rtcp->them.sin_port = htons(ntohs(them->sin_port) + 1);
    		rtp->rtcp->them.sin_addr = them->sin_addr;
    	}
    
    Mark Spencer's avatar
    Mark Spencer committed
    void ast_rtp_get_peer(struct ast_rtp *rtp, struct sockaddr_in *them)
    {
    	them->sin_family = AF_INET;
    	them->sin_port = rtp->them.sin_port;
    	them->sin_addr = rtp->them.sin_addr;
    }
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    void ast_rtp_get_us(struct ast_rtp *rtp, struct sockaddr_in *us)
    {
    	memcpy(us, &rtp->us, sizeof(rtp->us));
    }
    
    
    void ast_rtp_stop(struct ast_rtp *rtp)
    {
    	memset(&rtp->them.sin_addr, 0, sizeof(rtp->them.sin_addr));
    	memset(&rtp->them.sin_port, 0, sizeof(rtp->them.sin_port));
    
    	if (rtp->rtcp) {
    		memset(&rtp->rtcp->them.sin_addr, 0, sizeof(rtp->them.sin_addr));
    		memset(&rtp->rtcp->them.sin_port, 0, sizeof(rtp->them.sin_port));
    	}
    
    Mark Spencer's avatar
    Mark Spencer committed
    void ast_rtp_destroy(struct ast_rtp *rtp)
    {
    
    Mark Spencer's avatar
    Mark Spencer committed
    	if (rtp->smoother)
    		ast_smoother_free(rtp->smoother);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	if (rtp->ioid)
    		ast_io_remove(rtp->io, rtp->ioid);
    	if (rtp->s > -1)
    		close(rtp->s);
    
    	if (rtp->rtcp) {
    		close(rtp->rtcp->s);
    		free(rtp->rtcp);
    	}
    
    Mark Spencer's avatar
    Mark Spencer committed
    	free(rtp);
    }
    
    
    static unsigned int calc_txstamp(struct ast_rtp *rtp, struct timeval *delivery)
    
    Mark Spencer's avatar
    Mark Spencer committed
    {
    	struct timeval now;
    	unsigned int ms;
    	if (!rtp->txcore.tv_sec && !rtp->txcore.tv_usec) {
    		gettimeofday(&rtp->txcore, NULL);
    
    		/* Round to 20ms for nice, pretty timestamps */
    
    Mark Spencer's avatar
    Mark Spencer committed
    		rtp->txcore.tv_usec -= rtp->txcore.tv_usec % 20000;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	}
    
    	if (delivery && (delivery->tv_sec || delivery->tv_usec)) {
    		/* Use previous txcore */
    
    		ms = (delivery->tv_sec - rtp->txcore.tv_sec) * 1000;
    
    		ms += (1000000 + delivery->tv_usec - rtp->txcore.tv_usec) / 1000 - 1000;
    
    		rtp->txcore.tv_sec = delivery->tv_sec;
    		rtp->txcore.tv_usec = delivery->tv_usec;
    	} else {
    		gettimeofday(&now, NULL);
    		ms = (now.tv_sec - rtp->txcore.tv_sec) * 1000;
    
    		ms += (1000000 + now.tv_usec - rtp->txcore.tv_usec) / 1000 - 1000;
    
    		/* Use what we just got for next time */
    		rtp->txcore.tv_sec = now.tv_sec;
    		rtp->txcore.tv_usec = now.tv_usec;
    	}
    
    Mark Spencer's avatar
    Mark Spencer committed
    	return ms;
    }
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    int ast_rtp_senddigit(struct ast_rtp *rtp, char digit)
    {
    	unsigned int *rtpheader;
    	int hdrlen = 12;
    	int res;
    	int ms;
    	int x;
    	char data[256];
    
    	char iabuf[INET_ADDRSTRLEN];
    
    Mark Spencer's avatar
    Mark Spencer committed
    
    	if ((digit <= '9') && (digit >= '0'))
    		digit -= '0';
    	else if (digit == '*')
    		digit = 10;
    	else if (digit == '#')
    		digit = 11;
    	else if ((digit >= 'A') && (digit <= 'D')) 
    		digit = digit - 'A' + 12;
    	else if ((digit >= 'a') && (digit <= 'd')) 
    		digit = digit - 'a' + 12;
    	else {
    		ast_log(LOG_WARNING, "Don't know how to represent '%c'\n", digit);
    		return -1;
    	}
    	
    
    	/* If we have no peer, return immediately */	
    	if (!rtp->them.sin_addr.s_addr)
    		return 0;
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    	gettimeofday(&rtp->dtmfmute, NULL);
    	rtp->dtmfmute.tv_usec += (500 * 1000);
    	if (rtp->dtmfmute.tv_usec > 1000000) {
    		rtp->dtmfmute.tv_usec -= 1000000;
    		rtp->dtmfmute.tv_sec += 1;
    	}
    
    
    	ms = calc_txstamp(rtp, NULL);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	/* Default prediction */
    
    	rtp->lastts = rtp->lastts + ms * 8;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	
    	/* Get a pointer to the header */
    	rtpheader = (unsigned int *)data;
    	rtpheader[0] = htonl((2 << 30) | (1 << 23) | (101 << 16) | (rtp->seqno++));
    	rtpheader[1] = htonl(rtp->lastts);
    	rtpheader[2] = htonl(rtp->ssrc); 
    	rtpheader[3] = htonl((digit << 24) | (0xa << 16) | (0));
    	for (x=0;x<4;x++) {
    		if (rtp->them.sin_port && rtp->them.sin_addr.s_addr) {
    
    			res = sendto(rtp->s, (void *)rtpheader, hdrlen + 4, 0, (struct sockaddr *)&rtp->them, sizeof(rtp->them));
    
    Mark Spencer's avatar
    Mark Spencer committed
    			if (res <0) 
    
    				ast_log(LOG_NOTICE, "RTP Transmission error to %s:%d: %s\n", ast_inet_ntoa(iabuf, sizeof(iabuf), rtp->them.sin_addr), ntohs(rtp->them.sin_port), strerror(errno));
    
    Mark Spencer's avatar
    Mark Spencer committed
    	#if 0
    
    		printf("Sent %d bytes of RTP data to %s:%d\n", res, ast_inet_ntoa(iabuf, sizeof(iabuf), rtp->them.sin_addr), ntohs(rtp->them.sin_port));
    
    Mark Spencer's avatar
    Mark Spencer committed
    	#endif		
    		}