mirror of
https://github.com/mpv-player/mpv
synced 2024-12-24 15:52:25 +00:00
8aa19f12a0
add missing closesocket calls. git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@15761 b3059339-0415-0410-9bf9-f77b7e298cf2
285 lines
7.7 KiB
C
285 lines
7.7 KiB
C
/* Imported from the dvbstream-0.2 project
|
|
*
|
|
* Modified for use with MPlayer, for details see the CVS changelog at
|
|
* http://www.mplayerhq.hu/cgi-bin/cvsweb.cgi/main/
|
|
* $Id$
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <sys/types.h>
|
|
#include <ctype.h>
|
|
#include "config.h"
|
|
#ifndef HAVE_WINSOCK2
|
|
#include <netinet/in.h>
|
|
#include <sys/socket.h>
|
|
#include <arpa/inet.h>
|
|
#define closesocket close
|
|
#else
|
|
#include <winsock2.h>
|
|
#include <ws2tcpip.h>
|
|
#endif
|
|
#include <errno.h>
|
|
#include "stream.h"
|
|
|
|
/* MPEG-2 TS RTP stack */
|
|
|
|
#define DEBUG 1
|
|
#include "rtp.h"
|
|
|
|
extern int network_bandwidth;
|
|
|
|
int read_rtp_from_server(int fd, char *buffer, int length) {
|
|
struct rtpheader rh;
|
|
char *data;
|
|
int len;
|
|
static int got_first = 0;
|
|
static unsigned short sequence;
|
|
|
|
if( buffer==NULL || length<0 ) return -1;
|
|
|
|
getrtp2(fd, &rh, &data, &len);
|
|
if( got_first && rh.b.sequence != (unsigned short)(sequence+1) )
|
|
mp_msg(MSGT_NETWORK,MSGL_ERR,"RTP packet sequence error! Expected: %d, received: %d\n",
|
|
sequence+1, rh.b.sequence);
|
|
got_first = 1;
|
|
sequence = rh.b.sequence;
|
|
memcpy(buffer, data, len);
|
|
return(len);
|
|
}
|
|
|
|
|
|
// Start listening on a UDP port. If multicast, join the group.
|
|
static int rtp_open_socket( URL_t *url ) {
|
|
int socket_server_fd, rxsockbufsz;
|
|
int err, err_len;
|
|
fd_set set;
|
|
struct sockaddr_in server_address;
|
|
struct ip_mreq mcast;
|
|
struct timeval tv;
|
|
struct hostent *hp;
|
|
|
|
mp_msg(MSGT_NETWORK,MSGL_V,"Listening for traffic on %s:%d ...\n", url->hostname, url->port );
|
|
|
|
socket_server_fd = socket(AF_INET, SOCK_DGRAM, 0);
|
|
// fcntl( socket_server_fd, F_SETFL, fcntl(socket_server_fd, F_GETFL) | O_NONBLOCK );
|
|
if( socket_server_fd==-1 ) {
|
|
mp_msg(MSGT_NETWORK,MSGL_ERR,"Failed to create socket\n");
|
|
return -1;
|
|
}
|
|
|
|
if( isalpha(url->hostname[0]) ) {
|
|
#ifndef HAVE_WINSOCK2
|
|
hp =(struct hostent*)gethostbyname( url->hostname );
|
|
if( hp==NULL ) {
|
|
mp_msg(MSGT_NETWORK,MSGL_ERR,"Counldn't resolve name: %s\n", url->hostname);
|
|
goto err_out;
|
|
}
|
|
memcpy( (void*)&server_address.sin_addr.s_addr, (void*)hp->h_addr, hp->h_length );
|
|
#else
|
|
server_address.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
#endif
|
|
} else {
|
|
#ifndef HAVE_WINSOCK2
|
|
#ifdef USE_ATON
|
|
inet_aton(url->hostname, &server_address.sin_addr);
|
|
#else
|
|
inet_pton(AF_INET, url->hostname, &server_address.sin_addr);
|
|
#endif
|
|
#else
|
|
server_address.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
#endif
|
|
}
|
|
server_address.sin_family=AF_INET;
|
|
server_address.sin_port=htons(url->port);
|
|
|
|
if( bind( socket_server_fd, (struct sockaddr*)&server_address, sizeof(server_address) )==-1 ) {
|
|
#ifndef HAVE_WINSOCK2
|
|
if( errno!=EINPROGRESS ) {
|
|
#else
|
|
if( WSAGetLastError() != WSAEINPROGRESS ) {
|
|
#endif
|
|
mp_msg(MSGT_NETWORK,MSGL_ERR,"Failed to connect to server\n");
|
|
goto err_out;
|
|
}
|
|
}
|
|
|
|
#ifdef HAVE_WINSOCK2
|
|
if (isalpha(url->hostname[0])) {
|
|
hp =(struct hostent*)gethostbyname( url->hostname );
|
|
if( hp==NULL ) {
|
|
mp_msg(MSGT_NETWORK,MSGL_ERR,"Counldn't resolve name: %s\n", url->hostname);
|
|
goto err_out;
|
|
}
|
|
memcpy( (void*)&server_address.sin_addr.s_addr, (void*)hp->h_addr, hp->h_length );
|
|
} else {
|
|
unsigned int addr = inet_addr(url->hostname);
|
|
memcpy( (void*)&server_address.sin_addr, (void*)&addr, sizeof(addr) );
|
|
}
|
|
#endif
|
|
|
|
// Increase the socket rx buffer size to maximum -- this is UDP
|
|
rxsockbufsz = 240 * 1024;
|
|
if( setsockopt( socket_server_fd, SOL_SOCKET, SO_RCVBUF, &rxsockbufsz, sizeof(rxsockbufsz))) {
|
|
mp_msg(MSGT_NETWORK,MSGL_ERR,"Couldn't set receive socket buffer size\n");
|
|
}
|
|
|
|
if((ntohl(server_address.sin_addr.s_addr) >> 28) == 0xe) {
|
|
mcast.imr_multiaddr.s_addr = server_address.sin_addr.s_addr;
|
|
//mcast.imr_interface.s_addr = inet_addr("10.1.1.2");
|
|
mcast.imr_interface.s_addr = 0;
|
|
if( setsockopt( socket_server_fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mcast, sizeof(mcast))) {
|
|
mp_msg(MSGT_NETWORK,MSGL_ERR,"IP_ADD_MEMBERSHIP failed (do you have multicasting enabled in your kernel?)\n");
|
|
goto err_out;
|
|
}
|
|
}
|
|
|
|
tv.tv_sec = 0;
|
|
tv.tv_usec = (1 * 1000000); // 1 second timeout
|
|
FD_ZERO( &set );
|
|
FD_SET( socket_server_fd, &set );
|
|
err = select(socket_server_fd+1, &set, NULL, NULL, &tv);
|
|
if (err < 0) {
|
|
mp_msg(MSGT_NETWORK, MSGL_FATAL, "Select failed: %s\n", strerror(errno));
|
|
goto err_out;
|
|
}
|
|
if (err == 0) {
|
|
mp_msg(MSGT_NETWORK,MSGL_ERR,"Timeout! No data from host %s\n", url->hostname );
|
|
goto err_out;
|
|
}
|
|
err_len = sizeof( err );
|
|
getsockopt( socket_server_fd, SOL_SOCKET, SO_ERROR, &err, &err_len );
|
|
if( err ) {
|
|
mp_msg(MSGT_NETWORK,MSGL_DBG2,"Socket error: %d\n", err );
|
|
goto err_out;
|
|
}
|
|
return socket_server_fd;
|
|
|
|
err_out:
|
|
closesocket(socket_server_fd);
|
|
return -1;
|
|
}
|
|
|
|
static int rtp_streaming_read( int fd, char *buffer, int size, streaming_ctrl_t *streaming_ctrl ) {
|
|
return read_rtp_from_server( fd, buffer, size );
|
|
}
|
|
|
|
static int rtp_streaming_start( stream_t *stream, int raw_udp ) {
|
|
streaming_ctrl_t *streaming_ctrl;
|
|
int fd;
|
|
|
|
if( stream==NULL ) return -1;
|
|
streaming_ctrl = stream->streaming_ctrl;
|
|
fd = stream->fd;
|
|
|
|
if( fd<0 ) {
|
|
fd = rtp_open_socket( (streaming_ctrl->url) );
|
|
if( fd<0 ) return -1;
|
|
stream->fd = fd;
|
|
}
|
|
|
|
if(raw_udp)
|
|
streaming_ctrl->streaming_read = nop_streaming_read;
|
|
else
|
|
streaming_ctrl->streaming_read = rtp_streaming_read;
|
|
streaming_ctrl->streaming_seek = nop_streaming_seek;
|
|
streaming_ctrl->prebuffer_size = 64*1024; // 64 KBytes
|
|
streaming_ctrl->buffering = 0;
|
|
streaming_ctrl->status = streaming_playing_e;
|
|
return 0;
|
|
}
|
|
|
|
|
|
int getrtp2(int fd, struct rtpheader *rh, char** data, int* lengthData) {
|
|
static char buf[1600];
|
|
unsigned int intP;
|
|
char* charP = (char*) &intP;
|
|
int headerSize;
|
|
int lengthPacket;
|
|
lengthPacket=recv(fd,buf,1590,0);
|
|
if (lengthPacket==0)
|
|
exit(1);
|
|
if (lengthPacket<0) {
|
|
fprintf(stderr,"socket read error\n");
|
|
exit(2);
|
|
}
|
|
if (lengthPacket<12) {
|
|
fprintf(stderr,"packet too small (%d) to be an rtp frame (>12bytes)\n", lengthPacket);
|
|
exit(3);
|
|
}
|
|
rh->b.v = (unsigned int) ((buf[0]>>6)&0x03);
|
|
rh->b.p = (unsigned int) ((buf[0]>>5)&0x01);
|
|
rh->b.x = (unsigned int) ((buf[0]>>4)&0x01);
|
|
rh->b.cc = (unsigned int) ((buf[0]>>0)&0x0f);
|
|
rh->b.m = (unsigned int) ((buf[1]>>7)&0x01);
|
|
rh->b.pt = (unsigned int) ((buf[1]>>0)&0x7f);
|
|
intP = 0;
|
|
memcpy(charP+2,&buf[2],2);
|
|
rh->b.sequence = ntohl(intP);
|
|
intP = 0;
|
|
memcpy(charP,&buf[4],4);
|
|
rh->timestamp = ntohl(intP);
|
|
|
|
headerSize = 12 + 4*rh->b.cc; /* in bytes */
|
|
|
|
*lengthData = lengthPacket - headerSize;
|
|
*data = (char*) buf + headerSize;
|
|
|
|
// fprintf(stderr,"Reading rtp: v=%x p=%x x=%x cc=%x m=%x pt=%x seq=%x ts=%x lgth=%d\n",rh->b.v,rh->b.p,rh->b.x,rh->b.cc,rh->b.m,rh->b.pt,rh->b.sequence,rh->timestamp,lengthPacket);
|
|
|
|
return(0);
|
|
}
|
|
|
|
|
|
static int open_s(stream_t *stream,int mode, void* opts, int* file_format) {
|
|
URL_t *url;
|
|
int udp = 0;
|
|
|
|
mp_msg(MSGT_OPEN, MSGL_INFO, "STREAM_RTP, URL: %s\n", stream->url);
|
|
stream->streaming_ctrl = streaming_ctrl_new();
|
|
if( stream->streaming_ctrl==NULL ) {
|
|
return STREAM_ERROR;
|
|
}
|
|
stream->streaming_ctrl->bandwidth = network_bandwidth;
|
|
url = url_new(stream->url);
|
|
stream->streaming_ctrl->url = check4proxies(url);
|
|
|
|
if( url->port==0 ) {
|
|
mp_msg(MSGT_NETWORK,MSGL_ERR,"You must enter a port number for RTP and UDP streams!\n");
|
|
goto fail;
|
|
}
|
|
if(!strncmp(stream->url, "udp", 3))
|
|
udp = 1;
|
|
|
|
if(rtp_streaming_start(stream, udp) < 0) {
|
|
mp_msg(MSGT_NETWORK,MSGL_ERR,"rtp_streaming_start(rtp) failed\n");
|
|
goto fail;
|
|
}
|
|
|
|
stream->type = STREAMTYPE_STREAM;
|
|
fixup_network_stream_cache(stream);
|
|
return STREAM_OK;
|
|
|
|
fail:
|
|
streaming_ctrl_free( stream->streaming_ctrl );
|
|
stream->streaming_ctrl = NULL;
|
|
return STREAM_UNSUPORTED;
|
|
}
|
|
|
|
|
|
stream_info_t stream_info_rtp_udp = {
|
|
"mpeg rtp and upd streaming",
|
|
"rtp and udp",
|
|
"Dave Chapman",
|
|
"native rtp support",
|
|
open_s,
|
|
{"rtp", "udp", NULL},
|
|
NULL,
|
|
0 // Urls are an option string
|
|
};
|
|
|
|
|