source: svn/zas_dstar/candidates/tsmux.cpp

#include <stdio.h>
#include <stdlib.h>
#include <memory.h>

int main(int argc, char* argv[])
{
        // ÀÔ·Â ÀÎÀÚ °¹¼ö È®ÀÎ
        if (argc != 4)
        {
                printf("%s update_file ts_file output_file\n", argv[0]);
                return -1;
        }
        // ÀÔ·Â ÆÄÀÏÀÌ Á¸ÀçÇÏ´Â Áö È®ÀÎ
        FILE *fp = fopen(argv[1],"rb");
        if (fp == 0)
        {
                printf("Can not open file %s\n", argv[1]);
                printf("%s update_file ts_file output_file\n", argv[0]);
                return -2;
        }
        // ÀÔ·Â ÆÄÀÏÀÇ ±æÀÌ È®ÀÎ
        fseek(fp, 0L, SEEK_END);
        int nFileSize = ftell(fp);
        if (nFileSize == 0)
        {
                printf("%s update_file ts_file output_file\n", argv[0]);
                printf("File Size is Zero\n");
                return -5;
        }
        // ÀÔ·Â ÆÄÀÏÀ» ÀÐ¾î µéÀÓ µÎ¹ø ¹Ýº¹ÇÑ´Ù
        fseek(fp, 0L, SEEK_SET);
        unsigned char *data = (unsigned char*)malloc(nFileSize*2);
        if (data == 0)
        {
                printf("%s update_file ts_file output_file\n", argv[0]);
                printf("Out of Memory\n");
                fclose(fp);
                return -6;
        }
        fread( data, 1, nFileSize, fp);
        memcpy(&data[nFileSize], &data[0], nFileSize); // µÚ·Î º¹Á¦
        fclose(fp);
        // ÀÔ·Â ÆÄÀÏÀÌ Á¸ÀçÇÏ´Â Áö È®ÀÎ
        fp = fopen(argv[2],"rb");
        if (fp == 0)
        {
                printf("Can not open file %s\n", argv[2]);
                printf("%s update_file ts_file output_file\n", argv[0]);
                return -3;
        }
        remove(argv[3]); // Ãâ·Â ÆÄÀÏ »èÁ¦
        FILE *fp_output = fopen(argv[3],"wb");
        if (fp_output == 0)
        {
                printf("Can not open file %s\n", argv[3]);
                printf("%s update_file ts_file output_file\n", argv[0]);
                return -4;
        }
   for (int i = 0; i < nFileSize*2; i+=188)
   {
                char buff[188];
                if (fread( buff, 1, 188, fp) != 188)
                {
                        printf("rewind\n");
                        fseek(fp, 0L, SEEK_SET);
                        fread( buff, 1, 188, fp);
                }
                unsigned short pid = (buff[1]&0x1F)*256 + buff[2];
                if (pid < 256)
                {
                        fwrite( buff, 1, 188, fp_output);
                        i-=188;
                        continue;
                }
                else
                {
                        fwrite( &data[i], 1, 188, fp_output);
                }
        }
        fclose(fp);
        fclose(fp_output);
        return 0;
}