/*
 * fakewav  --  hide in / extract from fake wav files
 * Dec 22 2001
 * by itouh
 */


#define PROGNAME "fakewav"
#define PROGVERSION "1.0"

#include <stdio.h>
#include <stdlib.h> /* exit() */
#include <sys/stat.h> /* stat() */
#include <stdarg.h> /* vfprintf() */
#include <string.h> /* strncpy(), strrchr() */

#define uchar unsigned char
#define ulong unsigned long
#define FILENAMELEN 200
#define PASSWDLEN 200
#define BUFLEN (1024*1024) //1MBytes
typedef enum {TRUE, FALSE} bool;
typedef enum {ENCRYPT, DECRYPT} crypt_type;

void mkOutfile(const char *fname_org, const bool allowOverwrite);
void arrangeEndian(void *v, const int size, FILE *fin);
void errexit(char *fmt, ...);
int cryptbyte(size_t size, FILE *fin, FILE *fout);
void writewavhd(FILE *fout, int size);
int checkwavhd(FILE *fin);
char *removeDirFromFilename(char *fname);

char passwd[PASSWDLEN+1]; //終端'\0'を格納するために +1する
ulong filebytecounter; //ファイル先頭からのバイト数をかぞえる(単調増加)
crypt_type actmode = ENCRYPT;

/*
 * handling arguments 
 */ 
int main(int argc, char *argv[]){
    int i;
    static bool passdefined = FALSE; //passwdが定義されるとTRUE、されてないとF
    static bool allowOverwrite = TRUE; //outputfileを上書きできればTRUE

    /*
     * arguments
     */
    if(argc<=1 || strncmp(argv[1],"-h",2)==0 || strncmp(argv[1],"-v",2)==0){
        fprintf(stderr, PROGNAME " version " PROGVERSION "\n");
        fprintf(stderr, "usage:  " PROGNAME " [-de] [-i] [-p] inputfilename..\n");
        fprintf(stderr, "option: -e,-c (default)  encrypt inputfiles and output 'inputfilename.wav'\n");
        fprintf(stderr, "        -d,-x            decrypt inputfiles and output files excluding '.wav'\n");
        fprintf(stderr, "        -i               disallow to overwrite existing files\n");
        fprintf(stderr, "        -p\"pass phrase\"  specify password/phrase\n");
        exit(1);
    }

    for(i=1; i<argc; i++){
        if(passdefined==FALSE && (strncmp(argv[i],"-z",2)==0 || strncmp(argv[i],"-p",2)==0)){
            passdefined = TRUE;
            memcpy(passwd, &(argv[i][2]), PASSWDLEN);
            *(passwd+PASSWDLEN) = '\0';
#ifdef DEBUG
            fprintf(stderr, "passphrase = '%s'\n", passwd);
#endif /* DEBUG */
        }else if(strncmp(argv[i],"-i",2)==0){
            allowOverwrite = FALSE;
        }else if(actmode!=ENCRYPT && (strncmp(argv[i],"-e",2)==0 || strncmp(argv[i],"-c",2)==0)){
            actmode = ENCRYPT;
        }else if(actmode!=DECRYPT && (strncmp(argv[i],"-d",2)==0 || strncmp(argv[i],"-x",2)==0)){
            actmode = DECRYPT;
        }else{
            //各input-fileの先頭が、パスフレーズの1文字めによって
            //暗号化されるように resyncする
            filebytecounter = 0;

            mkOutfile(argv[i], allowOverwrite);
        }
    }
    return EXIT_SUCCESS;
}



/*
 * ファイル名を決定する
 * 他の関数を呼び、ファイルを書き出す
 */
void mkOutfile(const char *fname_org, const bool allowOverwrite){
    FILE *fin, *fout;
    struct stat info;
    char fname[FILENAMELEN+1];
    char newfname[FILENAMELEN+1];
    long inputfilesize;
    int res;

    /*
     *  input-fileのファイルサイズを調べる
     */
    strcpy(fname, fname_org);
    fname[FILENAMELEN] = '\0';

    if(stat(fname, &info) != 0){
        //ファイルが無いとき
        errexit("'%s' does not exist!", fname);
    }
    inputfilesize = info.st_size;
    if((fin=fopen(fname_org,"rb")) == NULL)
        errexit("unable to read '%s'!", fname);

    /*
     * output-file のファイル名を決定する
     */
    if(actmode == ENCRYPT){
	//output-filenameからディレクトリの部分を削除する
	char *p = removeDirFromFilename(fname);

        memset(newfname, 0x00, FILENAMELEN+1);
        sprintf(newfname, "%s.wav", p);
        //文字列が終端していない場合のために'\0'を入れる
        newfname[FILENAMELEN] = '\0';
    }else{
        //actmode == DECRYPT
        char *p;

        // inputfilenameから '.wav'を削除する
        // 拡張子(suffix)がついてないファイルの場合、エラー終了する
        if((p=strrchr(fname, '.'))==NULL)
            errexit("'%s' has no suffix!",fname);
        *p = '\0'; // '.wav'を削除

	p = removeDirFromFilename(fname);
        memset(newfname, 0x00, FILENAMELEN+1);
        strcpy(newfname, p);
        newfname[FILENAMELEN] = '\0';
    }


    /*
     * output-fileが既にあるかどうか調べる
     */
    if(allowOverwrite==FALSE && stat(newfname,&info)==0){
        //上書き許可がなくて、ファイルが既にあったとき
        errexit("'%s' has already existed!", newfname);
    }


    /*
     * wavヘッダを付加/削除
     */
    if(actmode == ENCRYPT){
	if((fout=fopen(newfname,"wb")) == NULL)
	    errexit("unable to write '%s'!", newfname);
        writewavhd(fout, inputfilesize);
    }else{
        //actmode == DECRYPT
        if(checkwavhd(fin))
            errexit("'%s' is not a wav file!", fname);

        //wavヘッダを読みとばす
        if(fseek(fin, 44, SEEK_SET)!=0)
            errexit("error occurs in reading '%s'!", fname);
        inputfilesize -= 44;

	//ここより前に置くと、wavファイルじゃなかったとき、
	// 0バイトのファイルが残る
	if((fout=fopen(newfname,"wb")) == NULL)
	    errexit("unable to write '%s'!", newfname);
    }


    /*
     * input-fileを読み込み、output-fileに書き出す
     */
#ifdef DEBUG
    fprintf(stderr, "inputfilesize(excluding wav header) = %ld\n", inputfilesize);
#endif /* DEBUG */
    while(inputfilesize > BUFLEN){
        if((res=cryptbyte(BUFLEN, fin, fout))!=0)
            errexit("error %d in reading '%s' and writing '%s'", res, fname, newfname);
        inputfilesize -= BUFLEN;
    }
    if(inputfilesize != 0){
        if((res=cryptbyte(inputfilesize, fin, fout))!=0)
            errexit("error %d in reading '%s' and writing '%s'", res, fname, newfname);
    }
    fclose(fout);
    fclose(fin);
    printf("    wrote '%s'\n", newfname);

    return;
}    



/*
 * ファイルを指定バイトだけ読みこみ、暗号化/復号化して、書き込む
 * out: 1==error, 0==ok
 */
int cryptbyte(size_t size, FILE *fin, FILE *fout){
    uchar buf[BUFLEN];
    int i;
    static int passlen = 0;

    if(passlen==0){
        passlen = strlen(passwd);
        if(passlen==0)
            passlen = 1;
#ifdef DEBUG
        fprintf(stderr, "passlen = %d\n", passlen);
#endif /* DEBUG */
    }

    /*
     * バイト列を読みこみ
     */
    if(fread(buf, 1, size, fin)!=size)
        return 1;
    /*
     * ファイルの暗号化
     * 1バイトづつ passwdとファイルの中身を足してmoduloとったもの
     * ただし、passwdはビット反転させる
     */
    for(i=0; i<size; i++){
        uchar passchar = 0xff - passwd[filebytecounter%passlen];
        long nn; //1バイト暗号化/復号化したもの、moduloする前
        if(actmode == ENCRYPT)
            nn = *(buf+i) + passchar;
        else{
            //actmode == DECRYPT
            //256を足してるのは、passcharを引いたときマイナスにならないため
            //あとで、256で modulo するので、数値はもんだいない
            nn = *(buf+i) - passchar + 256;
        }
            
#ifdef DEBUG
        fprintf(stderr,
                "buf=0x%02x, pass=0x%02x'%c', passchar=0x%02x, newbuf=0x%02x\n",
                *(uchar*)(buf+i), passwd[filebytecounter%passlen],
                passwd[filebytecounter%passlen], passchar, (uchar)nn%256);
#endif /* DEBUG */

        //0xff(255)でmoduloとすると、..254,0,1,..となり、255が抜ける
        *(buf+i) = nn % 256;

        filebytecounter++;
    }

    /*
     * バイト列を出力
     */
    if(fwrite(buf, 1, size, fout)!=size)
        return 2;
    return 0;
}



/*
 * あなたの使っているCPUのエンディアンを実行時にチェックして
 * もし必要があれば、値を適宜並びかえ、
 * ファイルに書き込む
 */
void arrangeEndian(void *v, const int size, FILE *fout){
    typedef enum { UNKNOWN, LITTLE, BIG } kindofendian;
    static kindofendian endian = UNKNOWN;
    char *p = v;
    int i;

    //エンディアンの判定
    if(endian==UNKNOWN){
        long n = 0x12345678;
        if(*(char*)(&n) == 0x12)
            endian = BIG;
        else
            endian = LITTLE;
#ifdef DEBUG
        fprintf(stderr, "endian = %s\n", endian==LITTLE ? "Little":"Big");
#endif /* DEBUG */
    }

    //エンディアンにあわせて値を並べる
    //wavファイルは、数値をリトルエンディアンで持つので、
    //CPUがビッグエンディアンのときだけ、並びかえる
    if(endian==LITTLE){
        for(i=0; i<size; i++){
            if(fwrite(p+i, 1, 1, fout)!=1)
                errexit("error occurs in writing an wav file!");
        }
    }else{
        //endian==BIG
        for(i=size-1; i>=0; i--){
            if(fwrite(p+i, 1, 1, fout)!=1)
                errexit("error occurs in writing an wav file!");
        }
    }
    return;
}       



/*
 * エラー文を表示して、このプログラムを終了する
 */
void errexit(char *fmt, ...){
        va_list ap;
        va_start(ap, fmt);
        fprintf(stderr, "%s: ", PROGNAME);
        vfprintf(stderr, fmt, ap);
        fprintf(stderr, "\n");
        va_end(ap);

        exit(EXIT_FAILURE);
}



/* WAVヘッダの構造
 *   数値はリトルエンディアン、符号あり

string      filesize-8  string      string         wavfilesize-8=inputfsize+44
R  I  F  F  n  n  n  n  W  A  V  E  f  m  t ' '    ' 'は半角空白
                                                   fmtsize:16(見たまま)
           |fmt〜                                  pcm:1(無圧縮)
fmt-size   |pcm   chanl sample/sec  byte/sec       chnl:2(stereo)
10 00 00 00|01 00 02 00 44 ac 00 00 10 b1 02 00    sample:44100
                                                   byte:176400 = sample * blok
      〜fmt|                       |data〜         block:4byte
block bit  |string      datasize   |data              =2byte(=16bit)*2(=stereo)
04 00 10 00|d  a  t  a  n  n  n  n |n  n..         bit:16(stereo16bit)   
                                                   datasize:wavfilesize-44
                                                      = inputfilesize(見たまま)
44100 == 0xac44
176400 == 0x2b110
 */

/*
 * wav ファイルのヘッダ部分を書き出す
 */
void writewavhd(FILE *fout, int inputfilesize){
    long n;
    short h;

    fprintf(fout, "RIFF");

    n = inputfilesize + 44;
    arrangeEndian(&n, sizeof(n), fout);

    fprintf(fout, "WAVEfmt ");

    n = 16; //16bit
    arrangeEndian(&n, sizeof(n), fout);

    h = 1; //pcm-type: MS non-compression
    arrangeEndian(&h, sizeof(h), fout);

    h = 2; //channel: stereo
    arrangeEndian(&h, sizeof(h), fout);

    n = 44100;
    arrangeEndian(&n, sizeof(n), fout);

    n = 176400; //44100(sampling/sec) * 2byte(16bit) * 2(stereo)
    arrangeEndian(&n, sizeof(n), fout);

    h = 4; //2byte(16bit) * 2(stereo)
    arrangeEndian(&h, sizeof(h), fout);

    h = 16;
    arrangeEndian(&h, sizeof(h), fout);

    fprintf(fout, "data");

    n = inputfilesize;
    arrangeEndian(&n, sizeof(n), fout);
}



/*
 * ファイル先頭を読み込み、wavファイルであるか照合する
 * out: 1..error, 0..OK
 */ 
int checkwavhd(FILE *fin){
    const int checksize = 16;
    char wavbuf[checksize];

    // wavファイルかどうかチェックする
    if(fread(wavbuf, 1, checksize, fin)!=checksize)
        return 1;
    if(strncmp(wavbuf, "RIFF", 4)!=0 || strncmp(wavbuf+8,"WAVEfmt ",8)!=0)
        return 1;
    return 0;
}



/*
 * 出力ファイル名がディレクトリを含んでいれば削除する
 * その結果、出力はカレントディレクトリになされる
 */
char *removeDirFromFilename(char *fname){
    int i;
    char *p = fname;

    for(i=0; i<strlen(fname); i++){
	if(*(p+i)=='\\')
	    *(p+i) = '/'; //backslashをslashに置き換え
    }

    if((p=strrchr(fname, '/'))==NULL)
	p = fname; //inputfilename の最初の文字
    else
	p++; // slashの次の文字

    return p;
}

//end of file