kopia lustrzana https://github.com/F5OEO/tstools
427 wiersze
10 KiB
C
427 wiersze
10 KiB
C
/*
|
||
* ***** BEGIN LICENSE BLOCK *****
|
||
* Version: MPL 1.1
|
||
*
|
||
* The contents of this file are subject to the Mozilla Public License Version
|
||
* 1.1 (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.mozilla.org/MPL/
|
||
*
|
||
* Software distributed under the License is distributed on an "AS IS" basis,
|
||
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
||
* for the specific language governing rights and limitations under the
|
||
* License.
|
||
*
|
||
* The Original Code is the MPEG TS, PS and ES tools.
|
||
*
|
||
* The Initial Developer of the Original Code is Amino Communications Ltd.
|
||
* Portions created by the Initial Developer are Copyright (C) 2008
|
||
* the Initial Developer. All Rights Reserved.
|
||
*
|
||
* Contributor(s):
|
||
* Gareth Bailey (gb@kynesim.co.uk), Kynesim, Cambridge, UK
|
||
*
|
||
* ***** END LICENSE BLOCK *****
|
||
*/
|
||
|
||
#include "compat.h"
|
||
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#include <string.h>
|
||
#include <fcntl.h>
|
||
#include <sys/stat.h>
|
||
#include <assert.h>
|
||
#include <errno.h>
|
||
|
||
#include "misc_fns.h"
|
||
#include "printing_fns.h"
|
||
#include "version.h"
|
||
|
||
#define TS_PACKET_SIZE 188
|
||
|
||
#define TO_BE16(from,to) *to = (0xFF & (from>>8)); *(to+1) = (0xFF & from);
|
||
|
||
static uint8_t *create_out_packet(char *in_data, int in_len, uint16_t pid)
|
||
{
|
||
uint8_t *out_packet = malloc(TS_PACKET_SIZE);
|
||
uint8_t *ptr = out_packet;
|
||
uint16_t flags;
|
||
uint16_t flags_pid;
|
||
|
||
if (!ptr)
|
||
return NULL;
|
||
if (in_len > (TS_PACKET_SIZE - 4))
|
||
return NULL;
|
||
|
||
*ptr = 0x47;
|
||
ptr++;
|
||
|
||
/* Transport Error Indicator */
|
||
flags = 0<<15;
|
||
|
||
/* Payload Unit Start Indicator */
|
||
flags = flags | 0<<14;
|
||
|
||
/* Transport Priority */
|
||
flags = flags | 0<<13;
|
||
|
||
flags_pid = flags | pid;
|
||
TO_BE16(flags_pid,ptr);
|
||
ptr+=2;
|
||
|
||
*ptr = 0x11;
|
||
ptr++;
|
||
|
||
memcpy(ptr,in_data,in_len);
|
||
ptr+=in_len;
|
||
|
||
memset(ptr,0xFF,TS_PACKET_SIZE-(ptr-out_packet));
|
||
ptr+=TS_PACKET_SIZE-(ptr-out_packet);
|
||
|
||
assert((ptr-TS_PACKET_SIZE) == out_packet);
|
||
|
||
{
|
||
int i;
|
||
ptr = out_packet;
|
||
print_msg("Packet to be written is:\n");
|
||
for (i=0;i<TS_PACKET_SIZE;i++)
|
||
{
|
||
if (!(i%16)) print_msg("\n");
|
||
|
||
fprint_msg("%02x ",ptr[i]);
|
||
}
|
||
print_msg("\n\n");
|
||
}
|
||
|
||
return out_packet;
|
||
}
|
||
|
||
|
||
static int insert_packets(int file,
|
||
int out_file,
|
||
uint8_t *out_packet,
|
||
int *packet_numbers,int n_pack)
|
||
{
|
||
uint8_t buf[TS_PACKET_SIZE];
|
||
int rv;
|
||
int packets_read=0;
|
||
int bytes_read=0;
|
||
int packnum_i=0;
|
||
|
||
while (1)
|
||
{
|
||
rv = read(file, buf, TS_PACKET_SIZE);
|
||
|
||
if (rv == 0) break;
|
||
if (rv != TS_PACKET_SIZE) return rv;
|
||
|
||
bytes_read+=rv;
|
||
|
||
if (packet_numbers[packnum_i]==packets_read && packnum_i<n_pack)
|
||
{
|
||
fprint_msg("Writing new packet before packet %d...\n",packets_read);
|
||
|
||
rv = write(out_file,out_packet,TS_PACKET_SIZE);
|
||
assert(rv == TS_PACKET_SIZE);
|
||
packnum_i++;
|
||
}
|
||
|
||
rv = write(out_file,buf,rv);
|
||
|
||
assert(rv=TS_PACKET_SIZE);
|
||
|
||
packets_read++;
|
||
}
|
||
|
||
fprint_msg("\nRead a total of %d packets (%d bytes)\n",packets_read,bytes_read);
|
||
return 0;
|
||
}
|
||
|
||
static off_t get_file_size(int file)
|
||
{
|
||
struct stat stat_ret;
|
||
fstat(file, &stat_ret);
|
||
|
||
return stat_ret.st_size;
|
||
}
|
||
|
||
|
||
static int num_char_in_string(char *string,char c)
|
||
{
|
||
unsigned int i;
|
||
int a=0;
|
||
|
||
for (i=0; i<strlen(string);i++)
|
||
{
|
||
if (c == *(string+i))
|
||
a++;
|
||
}
|
||
return a;
|
||
}
|
||
|
||
static void print_usage()
|
||
{
|
||
print_msg(
|
||
"Usage: ts_packet_insert [switches] <infile>\n"
|
||
"\n"
|
||
);
|
||
REPORT_VERSION("ts_packet_insert");
|
||
print_msg(
|
||
"\n"
|
||
" Insert TS packets into a Transport Stream at positions\n"
|
||
" specified by the user.\n"
|
||
"\n"
|
||
"Input:\n"
|
||
" <infile> An H.222 Transport Stream file.\n"
|
||
"\n"
|
||
"Switches:\n"
|
||
" -err stdout Write error messages to standard output (the default)\n"
|
||
" -err stderr Write error messages to standard error (Unix traditional)\n"
|
||
" -p <positions> This a a colon (':') delimited string of numbers\n"
|
||
" between 0 and 1, representing how far through to put \n"
|
||
" each TS packet. E.g., -p 0.1:0.4:0.7:0.9 will insert\n"
|
||
" 4 packets at 10%%, 40%%, 70%% and 90%% through the file.\n"
|
||
" -pid <pid> The inserted packets will have the PID specfied.\n"
|
||
" If no PID is specified, then 0x68 will be used.\n"
|
||
" -s <string> The inserted packets will contain <string> as their\n"
|
||
" payload. This defaults to 'Inserted packet'.\n"
|
||
" -o <output file> The new TS file will be written out with the given name\n"
|
||
" (which defaults to out.ts)\n"
|
||
"For example:\n"
|
||
"\n"
|
||
" ts_packet_insert -p 0.3:0.6 -o out.ts -pid 89 -s \"AD=start\" in.ts\n"
|
||
);
|
||
}
|
||
|
||
/* bubble sort */
|
||
static void sort_positions(double *in_array,int size)
|
||
{
|
||
int sorted=0;
|
||
|
||
while (!sorted)
|
||
{
|
||
int i;
|
||
/* asume sorted */
|
||
sorted++;
|
||
|
||
for (i=0;i<size-1;i++)
|
||
{
|
||
if (in_array[i]>in_array[i+1])
|
||
{
|
||
double tmp;
|
||
|
||
tmp = in_array[i];
|
||
in_array[i] = in_array[i+1];
|
||
in_array[i+1] = tmp;
|
||
|
||
/* damn, go round again */
|
||
sorted=0;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
int main(int argc, char **argv)
|
||
{
|
||
char *output_file_path = "out.ts";
|
||
char *in_file_path = NULL;
|
||
long in_file_size=0;
|
||
/*an array of floats for the positions of packets to insert,values of 0-1*/
|
||
double *positions=NULL;
|
||
int *packet_numbers=NULL;
|
||
int n_pos = 0;
|
||
|
||
int argno = 1;
|
||
int arg_counter = 0;
|
||
|
||
uint32_t pid=0x68;
|
||
|
||
char *out_string="Inserted packet";
|
||
|
||
if (argc < 2)
|
||
{
|
||
print_usage();
|
||
return 0;
|
||
}
|
||
|
||
while (argno < argc)
|
||
{
|
||
if (argv[argno][0] == '-')
|
||
{
|
||
if (!strcmp("-p",argv[argno]))
|
||
{
|
||
char *endptr;
|
||
char *position_string;
|
||
int pos_index;
|
||
|
||
++argno;
|
||
|
||
free(positions);
|
||
|
||
n_pos = (num_char_in_string(argv[argno],':')+1);
|
||
positions = malloc(n_pos * sizeof(double));
|
||
if (!positions)
|
||
{
|
||
print_err("malloc failed");
|
||
exit(1);
|
||
}
|
||
|
||
position_string = strtok(argv[argno],":");
|
||
pos_index=0;
|
||
print_msg("Adding new packets at:");
|
||
while (1)
|
||
{
|
||
if (!position_string)
|
||
break;
|
||
|
||
positions[pos_index] = strtod(position_string,&endptr);
|
||
|
||
if (endptr == position_string || positions[pos_index]>1 || positions[pos_index]<0)
|
||
{
|
||
fprint_err("\nNot a valid floating point number for position (argument %d)\n",argno);
|
||
exit(1);
|
||
}
|
||
|
||
fprint_msg(" %d%%",(int)(positions[pos_index]*100));
|
||
|
||
position_string = strtok(NULL,":");
|
||
pos_index++;
|
||
}
|
||
print_msg("\n");
|
||
sort_positions(positions,n_pos);
|
||
assert(pos_index == n_pos);
|
||
|
||
}
|
||
else if (!strcmp("-err",argv[argno]))
|
||
{
|
||
CHECKARG("ts_packet_insert",argno);
|
||
if (!strcmp(argv[argno+1],"stderr"))
|
||
redirect_output_stderr();
|
||
else if (!strcmp(argv[argno+1],"stdout"))
|
||
redirect_output_stdout();
|
||
else
|
||
{
|
||
fprint_err("### ts_packet_insert: "
|
||
"Unrecognised option '%s' to -err (not 'stdout' or"
|
||
" 'stderr')\n",argv[argno+1]);
|
||
return 1;
|
||
}
|
||
argno++;
|
||
}
|
||
else if (!strcmp("-pid",argv[argno]))
|
||
{
|
||
int err;
|
||
CHECKARG("ts_packet_insert",argno);
|
||
err = unsigned_value("ts_packet_insert",argv[argno],argv[argno+1],0,&pid);
|
||
if (err) return 1;
|
||
argno++;
|
||
}
|
||
else if (!strcmp("-o",argv[argno]))
|
||
{
|
||
CHECKARG("ts_packet_insert",argno);
|
||
output_file_path = argv[++argno];
|
||
}
|
||
else if (!strcmp("-s",argv[argno]))
|
||
{
|
||
CHECKARG("ts_packet_insert",argno);
|
||
out_string = argv[++argno];
|
||
}
|
||
else if (!strcmp("-h",argv[argno]) || !strcmp("--help",argv[argno]))
|
||
{
|
||
print_usage();
|
||
return 0;
|
||
}
|
||
else
|
||
{
|
||
fprint_msg("\n *** Unknown option %s, ignoring.\n\n",argv[argno]);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
if (arg_counter == 0)
|
||
{
|
||
in_file_path = argv[argno];
|
||
arg_counter++;
|
||
}
|
||
else
|
||
{
|
||
fprint_err( "### ts_packet_insert: Unexpected '%s'\n", argv[argno]);
|
||
return 1;
|
||
}
|
||
}
|
||
argno++;
|
||
}
|
||
|
||
if (!in_file_path)
|
||
{
|
||
print_err("Error: No input file specified.\n");
|
||
exit(1);
|
||
}
|
||
|
||
fprint_msg("Reading from file: %s\n",in_file_path);
|
||
fprint_msg("Writing to file: %s\n",output_file_path);
|
||
fprint_msg("Inserting packets with PID %#x (%u)\n",pid,pid);
|
||
fprint_msg("Using output string: %s\n",out_string);
|
||
|
||
{
|
||
int out_file;
|
||
int in_file = open(in_file_path,O_RDONLY);
|
||
|
||
if (in_file<0)
|
||
{
|
||
fprint_err("Error: could not open %s for reading: %s\n",
|
||
in_file_path,strerror(errno));
|
||
exit(1);
|
||
}
|
||
|
||
out_file = open(output_file_path,O_WRONLY | O_TRUNC | O_CREAT,0644);
|
||
if (out_file<0)
|
||
{
|
||
fprint_err("Error: could not open %s for reading: %s\n",
|
||
output_file_path,strerror(errno));
|
||
exit(1);
|
||
}
|
||
|
||
in_file_size = get_file_size(in_file);
|
||
|
||
if (in_file_size % TS_PACKET_SIZE)
|
||
{
|
||
print_err("Error: TS file length is not a multiple of 188 bytes\n");
|
||
exit(1);
|
||
}
|
||
|
||
{
|
||
int num_pack = in_file_size / TS_PACKET_SIZE;
|
||
int i;
|
||
fprint_msg("\nInput file is %ld bytes long with ",in_file_size);
|
||
fprint_msg("%d TS packets\n",num_pack);
|
||
|
||
packet_numbers = malloc(n_pos * sizeof(int));
|
||
|
||
/* Find out which packets we insert before */
|
||
for (i=0;i<n_pos;i++)
|
||
{
|
||
packet_numbers[i] = (int)((double)num_pack * positions[i]);
|
||
}
|
||
|
||
|
||
{
|
||
/* create the packet to spit out */
|
||
uint8_t *out_packet = create_out_packet(out_string,
|
||
strlen(out_string)+1,
|
||
pid);
|
||
|
||
insert_packets(in_file,out_file,out_packet,packet_numbers,n_pos);
|
||
}
|
||
}
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
// Local Variables:
|
||
// tab-width: 8
|
||
// indent-tabs-mode: nil
|
||
// c-basic-offset: 2
|
||
// End:
|
||
// vim: set tabstop=8 shiftwidth=2 expandtab:
|