mirror of https://github.com/neutrinolabs/xrdp
230 lines
8.2 KiB
C
230 lines
8.2 KiB
C
/**
|
|
* xrdp: A Remote Desktop Protocol server.
|
|
*
|
|
* Copyright (C) Laxmikant Rashinkar 2012-2013 LK.Rashinkar@gmail.com
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
* a program that uses xrdpapi and ffmpeg to redirect media streams
|
|
* to a FreeRDP client where it is decompressed and played locally
|
|
*
|
|
*/
|
|
|
|
#ifndef __XRDPVR_INTERNAL_H__
|
|
#define __XRDPVR_INTERNAL_H__
|
|
|
|
#include <stdint.h>
|
|
#include <sys/types.h>
|
|
#include <fcntl.h>
|
|
#include <libavcodec/avcodec.h>
|
|
#include <libavformat/avformat.h>
|
|
|
|
#if LIBAVCODEC_VERSION_MAJOR == 52 && LIBAVCODEC_VERSION_MINOR == 20
|
|
#define DISTRO_DEBIAN6
|
|
#endif
|
|
|
|
#if LIBAVCODEC_VERSION_MAJOR == 52 && LIBAVCODEC_VERSION_MINOR == 72
|
|
#define DISTRO_UBUNTU1104
|
|
#endif
|
|
|
|
#if LIBAVCODEC_VERSION_MAJOR == 53 && LIBAVCODEC_VERSION_MINOR == 35
|
|
#define DISTRO_UBUNTU1204
|
|
#endif
|
|
|
|
#if !defined(DISTRO_DEBIAN6) && !defined(DISTRO_UBUNTU1104) && !defined(DISTRO_UBUNTU1204)
|
|
#warning unsupported distro
|
|
#endif
|
|
|
|
#ifdef DISTRO_UBUNTU1204
|
|
#define CODEC_TYPE_VIDEO AVMEDIA_TYPE_VIDEO
|
|
#define CODEC_TYPE_AUDIO AVMEDIA_TYPE_AUDIO
|
|
#define PKT_FLAG_KEY AV_PKT_FLAG_KEY
|
|
#endif
|
|
|
|
#define MAX_BUFSIZE (1024 * 1024 * 8)
|
|
|
|
#define CMD_SET_VIDEO_FORMAT 1
|
|
#define CMD_SET_AUDIO_FORMAT 2
|
|
#define CMD_SEND_VIDEO_DATA 3
|
|
#define CMD_SEND_AUDIO_DATA 4
|
|
#define CMD_CREATE_META_DATA_FILE 5
|
|
#define CMD_CLOSE_META_DATA_FILE 6
|
|
#define CMD_WRITE_META_DATA 7
|
|
#define CMD_DEINIT_XRDPVR 8
|
|
#define CMD_SET_GEOMETRY 9
|
|
#define CMD_SET_VOLUME 10
|
|
#define CMD_INIT_XRDPVR 11
|
|
|
|
/* max number of bytes we can send in one pkt */
|
|
#define MAX_PDU_SIZE 1600
|
|
|
|
typedef uint8_t u8;
|
|
typedef uint16_t u16;
|
|
typedef uint32_t u32;
|
|
|
|
typedef struct stream
|
|
{
|
|
u8 *data;
|
|
u8 *p;
|
|
u32 size;
|
|
int from_buf;
|
|
} STREAM;
|
|
|
|
/**
|
|
* create and init a new stream
|
|
*
|
|
* @param _s stream to create and init
|
|
* @param _len number of bytes to store in stream
|
|
******************************************************************************/
|
|
#define stream_new(_s, _len) \
|
|
do \
|
|
{ \
|
|
(_s) = (STREAM *) calloc(1, sizeof(STREAM)); \
|
|
(_s)->data = (u8 *) calloc(1, (_len)); \
|
|
(_s)->p = (_s)->data; \
|
|
(_s)->size = (_len); \
|
|
(_s)->from_buf = 0; \
|
|
} while (0)
|
|
|
|
/**
|
|
* create a stream from an existing buffer
|
|
******************************************************************************/
|
|
#define stream_from_buffer(_s, _buf, _buf_len) \
|
|
do \
|
|
{ \
|
|
(_s) = (STREAM *) calloc(1, sizeof(STREAM)); \
|
|
(_s)->data = (u8 *) (_buf); \
|
|
(_s)->p = (_s)->data; \
|
|
(_s)->size = (_buf_len); \
|
|
(_s)->from_buf = 1; \
|
|
} while (0)
|
|
|
|
/**
|
|
* release stream resources, including stream itself
|
|
* note: release _s->data only if we allocated it
|
|
*
|
|
* @param _s the stream whose resources are to be released
|
|
******************************************************************************/
|
|
#define stream_free(_s) \
|
|
do \
|
|
{ \
|
|
if (!(_s)->from_buf) \
|
|
{ \
|
|
free((_s)->data); \
|
|
} \
|
|
free((_s)); \
|
|
(_s) = NULL; \
|
|
} while (0)
|
|
|
|
/** return number of bytes in stream */
|
|
#define stream_length(_s) (int) ((_s)->p - (_s)->data)
|
|
|
|
/** insert a 8 bit value into stream */
|
|
#define stream_ins_u8(_s, _val) \
|
|
do \
|
|
{ \
|
|
*(_s)->p++ = (unsigned char) (_val); \
|
|
} while(0)
|
|
|
|
/** insert a 16 bit value into stream */
|
|
#define stream_ins_u16_le(_s, _val) \
|
|
do \
|
|
{ \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 0); \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 8); \
|
|
} while (0)
|
|
|
|
/** insert a 32 bit value into stream */
|
|
#define stream_ins_u32_le(_s, _val) \
|
|
do \
|
|
{ \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 0); \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 8); \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 16); \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 24); \
|
|
} while (0)
|
|
|
|
/** insert a 64 bit value into stream */
|
|
#define stream_ins_u64_le(_s, _val) \
|
|
do \
|
|
{ \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 0); \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 8); \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 16); \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 24); \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 32); \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 40); \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 48); \
|
|
*(_s)->p++ = (unsigned char) ((_val) >> 56); \
|
|
} while (0)
|
|
|
|
/** insert array of chars into stream */
|
|
#define stream_ins_byte_array(_s, _ba, _count) \
|
|
do \
|
|
{ \
|
|
memcpy((_s)->p, (_ba), (_count)); \
|
|
(_s)->p += (_count); \
|
|
} while (0)
|
|
|
|
/** extract a 8 bit value from stream */
|
|
#define stream_ext_u8(_s, _v) \
|
|
do \
|
|
{ \
|
|
(_v) = (u8) *(_s)->p++; \
|
|
} while (0)
|
|
|
|
/** extract a 16 bit value from stream */
|
|
#define stream_ext_u16_le(_s, _v) \
|
|
do \
|
|
{ \
|
|
(_v) = (u16) ((_s)->p[1] << 8 | (_s)->p[0]); \
|
|
(_s)->p += 2; \
|
|
} while (0)
|
|
|
|
/** extract a 32 bit value from stream */
|
|
#define stream_ext_u32_le(_s, _v) \
|
|
do \
|
|
{ \
|
|
(_v) = (u32) ((_s)->p[3] << 24 | \
|
|
(_s)->p[2] << 16 | \
|
|
(_s)->p[1] << 8 | \
|
|
(_s)->p[0]); \
|
|
(_s)->p += 4; \
|
|
} while (0)
|
|
|
|
typedef struct _player_state_info
|
|
{
|
|
AVFormatContext *p_format_ctx;
|
|
AVCodecContext *p_audio_codec_ctx;
|
|
AVCodecContext *p_video_codec_ctx;
|
|
AVCodec *p_audio_codec;
|
|
AVCodec *p_video_codec;
|
|
|
|
int audio_stream_index;
|
|
int video_stream_index;
|
|
double audioTimeout;
|
|
double videoTimeout;
|
|
|
|
/* LK_TODO delete this after we fix the problem */
|
|
AVFrame *frame;
|
|
AVPacket avpkt;
|
|
|
|
AVBitStreamFilterContext *bsfc;
|
|
|
|
} PLAYER_STATE_INFO;
|
|
|
|
static int xrdpvr_read_from_client(void *channel, STREAM *s, int bytes, int timeout);
|
|
static int xrdpvr_write_to_client(void *channel, STREAM *s);
|
|
|
|
#endif /* __XRDPVR_INTERNAL_H__ */
|