#include <stdio.h>
#include <stdlib.h>
#include <memory.h>

static unsigned int  DHL_PsiAPI_CalculateCRC(const unsigned char *data, unsigned int len)
{
	static unsigned int crcTable[256];
	static bool bFirst=true;
	if (bFirst)
	{
		bFirst = false;
		for (int i=0; i<256; i++) {
			unsigned int crc = 0;
			for (int j=7; j>=0; j--) {
				if (((i >> j) ^ (crc >> 31)) & 1) crc=(crc<<1)^0x04C11DB7;
				else crc<<=1;
			}
			crcTable[i] = crc;
		}
	}
	unsigned int crc = 0xFFFFFFFF;
	for (int i = 0; i < len; ++i) crc = (crc << 8) ^ crcTable[(crc >> 24) ^ (*data++)];
	return crc;
}

int main(int argc, char* argv[])
{
	// 입력 인자 갯수 확인
	if (argc != 5)
	{
		printf("%s ModelName VersionName FileName\n", argv[0]);
		return -1;
	}
	
	// 입력 파일이 존재하는 지 확인
	FILE *fp = fopen(argv[3],"rb");
	if (fp == 0)
	{
		printf("%s ModelName VersionName FileName\n", argv[0]);
		printf("Can not find file %s\n", argv[3]);
		return -2;
	}
	
	// 입력 파일의 길이 확인
	fseek(fp, 0L, SEEK_END);
	int nFileSize = ftell(fp);
	if (nFileSize == 0)
	{
		printf("%s ModelName VersionName FileName\n", argv[0]);
		printf("Can not find file %s\n", argv[1]);
		printf("File Size is Zero\n");
		return -3;
	}
	
	// 입력 파일을 읽어 들임
	fseek(fp, 0L, SEEK_SET);
	unsigned char *data = (unsigned char*)malloc(nFileSize);
	if (data == 0)
	{
		printf("%s ModelName VersionName FileName\n", argv[0]);
		printf("Can not find file %s\n", argv[1]);
		printf("Out of Memory\n");
		fclose(fp);
		return -4;
	}
	fread( data, 1, nFileSize, fp);
	fclose(fp);
	
	// 정보 구조체
	struct INFO
	{
		char strModelName[10]; //제품모델명
		char strVersionName[8]; // S/W 버전
		char strFileSize[8]; // 전체 파일 길이
		char strPosition[8]; // 데이터의 시작 위치
		char strLength[8]; // 데이터의 길이
		char strMaxBlockNumber[8]; // 마지막 블럭 번호
		char strBlockNumber[8]; // 몇번째 블럭인지 0부터 시작
	} info;
	
	memset(&info, 0, sizeof(info));
	
	// 구조체에 모델이름 복사
	int maxLen = strlen(argv[1]);
	if (maxLen > 9) maxLen = 9;
	memcpy(info.strModelName, argv[1], maxLen);
	
	// 구조체에 버전이름 복사
	maxLen = strlen(argv[2]);
	if (maxLen > 7) maxLen = 7;
	memcpy(info.strVersionName, argv[2], maxLen);
#if 1
	// CMB에 특화된 버전 이름 표기 1.0.00
	// DST_Main.cpp의 버전 규칙과 일치시켜야 한다.
	sprintf(info.strVersionName, "%d.%d.%d", 
		atoi(argv[2])/1000, 
		(atoi(argv[2])%1000)/100, 
		(atoi(argv[2])%1000)%100);
	printf("VERSION : %s\n", info.strVersionName);
#endif
	// 구조체에 파일 길이 복사
	sprintf(info.strFileSize, "%d", nFileSize); 
	
	//출력 파일 생성 입력파일위에 .ts 추가
	//char strOutFileName[4096];
	//sprintf(strOutFileName, "%s.ts", argv[3]);
	fp = fopen(argv[4], "wb");
	
	// 한 패킷에 들어갈 순수한 데이터의 최대 길이를 계산
	int packet_length = 4031+17;
	int max_data_length = packet_length - 9 - sizeof(info) - 4; //total - header - private header - crc

	// 구조체에 가장 큰 블럭번호 계산
	int nMaxBlockNumber =  (nFileSize -1)  / max_data_length;
	sprintf(info.strMaxBlockNumber, "%d", nMaxBlockNumber); 
	
	for (int k = 0; k < 2; k++) // 스트림 제너레이터에서 Wrap around 시 빠질것을 고려하여 2바퀴 돌린다.
	{	
		int nBlockNumber = 0; // 현재 블럭 번호 
		for (int nPosition = 0; nPosition < nFileSize; nPosition += max_data_length)
		{
			// 구조체에 파일상 위치 저장
			sprintf(info.strPosition, "%d", nPosition); 
			
			// 구조체에 현 블럭번호  저장
			sprintf(info.strBlockNumber, "%d", nBlockNumber); 
			
			// 저장할 데이터 길이
			int nLength = (nPosition + max_data_length >= nFileSize) ? nFileSize - nPosition : max_data_length;
			sprintf(info.strLength, "%d", nLength); 
			
			//printf("nLength = %d\n", nLength);
			//printf("max_data_length = %d\n", max_data_length);
			
			unsigned char section[5000];
			memset(section, 0xFF, 5000);
			int nPos = 0; // 배열내의 현재 위치
			section[nPos++] = 0x00; 
			section[nPos++] = 0xF5; // table id
			section[nPos++] = 0xF0 + ((packet_length - 4)/256); // section syntax indicator
			section[nPos++] = (packet_length - 4) % 256; // length
			section[nPos++] = 0x00; 
			section[nPos++] = 0x01;
			section[nPos++] = 0xC1;
			section[nPos++] = nPosition & 0xFF;
			section[nPos++] = 0xFF;
			
			int nLen = (section[2] & 0xF)*256 + section[3];
			// 정보 복사
			memcpy(&section[nPos], &info, sizeof(info));
			nPos += sizeof(info);
			
			// 데이터 복사
			memcpy(&section[nPos], &data[nPosition], nLength);
			
			// CRC 삽입
			unsigned int crc = DHL_PsiAPI_CalculateCRC(&section[1], nLen-1);
			
			section[nLen+0] = (crc >> 24) & 0xFF;
			section[nLen+1] = (crc >> 16) & 0xFF;
			section[nLen+2] = (crc >> 8) & 0xFF;
			section[nLen+3] = (crc) & 0xFF;	
			
			int nSectionLength = nLen;
			for (int i = 0; i <  nSectionLength; i+= 184) // 188에서 header 4바이트 제거
			{
				unsigned char packet[188];
				memset(packet, 0xFF, 188);
				int nPos = 0; // 배열내의 현재 위치
				packet[nPos++] = 0x47; //Sync
				packet[nPos++] = (i==0) ? 0x5F : 0x1F; // payload unit start indicator + 0x1FF5
				packet[nPos++] = 0xF5; // pid 0x1FF5
				static int continuity_counter = 0;
				packet[nPos++] = 0x10 + ((continuity_counter++)%16); // Adaptation field control + continuity counter
				// 데이터 복사
				memcpy(&packet[nPos], &section[i], 184);
				fwrite(packet, 1, 188, fp); // 파일 저장
			}
			// printf("i = %d data_length = %d packet[3] = 0x%02X\n", i, data_length, packet[3]);
			unsigned char packet[188];
			memset(packet, 0x00, 188);
			packet[0] = 0x47; //Sync
			packet[1] = 0x1F;
			packet[2] = 0xFF;
			packet[3] = 0x10;
			for ( int i = 0; i < 100; i++) // 스트림 제너레이터 특성에 따라 적당히 조절한다.
			{
				fwrite(packet, 1, 188, fp); // 파일 저장
			}
			
			nBlockNumber++;
		}
	}
	fclose(fp);
	return 0;
}