diff options
Diffstat (limited to 'test/test_dir.c')
| -rw-r--r-- | test/test_dir.c | 145 |
1 files changed, 145 insertions, 0 deletions
diff --git a/test/test_dir.c b/test/test_dir.c new file mode 100644 index 0000000..44a7afd --- /dev/null +++ b/test/test_dir.c @@ -0,0 +1,145 @@ +#include "vdefine.h" +#define SIZE_BUFF 16*16*16*16*16*16*16*16-1//6*16*16*16*16*16 +#include <dirent.h> +#include <sys/stat.h> +#include <unistd.h> +#define SIZE_FILE 50 +typedef struct buff_inf_h +{ + char *buff; + size_t size; +}buff_inf; + +buff_inf *buff_get(char *vfile) { + buff_inf *buff_inf_h = (buff_inf *)malloc(sizeof(buff_inf)); + + FILE *fr = NULL; + char *buff_ori = (char *)malloc(sizeof(char *)*SIZE_BUFF);//CH + char *buff = buff_ori;//CH + memset(buff, 0, sizeof(buff)); + if ((fr = fopen(vfile, "rb")) == NULL) + { + perror("fopen error:"); + buff_inf_h->buff = NULL; + buff_inf_h->size = 0; + return buff_inf_h;//-1; + } + size_t i = 0;//unsigned int i = 0;//size_t i = 0; + //int j = 0; + while (!feof(fr)) + { + fread(buff, 1, 1, fr); + buff += 1; + i++; + } + buff_inf_h->buff = (char *)malloc(sizeof(char *) * (i-1));//(i-2)); + memcpy(buff_inf_h->buff, buff - i, i-1);//i-2); + buff_inf_h->size = i-1;//i - 2; + //buff_inf_h.buff = buff - i; + //buff_inf_h.size = i; + free(buff_ori);//CH--free(buff - i);//? + buff_ori = NULL; + fclose(fr); + return buff_inf_h; + +} +int main(int argc, char *argv[]) +{ + DIR *dirp; + struct dirent *dp; + char *vdir = argv[1]; + char *vfile_ori = (char *)malloc(SIZE_FILE); + memcpy(vfile_ori,vdir,strlen(vdir)); + strcat(vfile_ori, "/"); + char *vfile = (char *)malloc(SIZE_FILE); + //strcat(vdir,".dir"); + dirp = opendir(vdir); + struct stat info; + + buff_inf *buff_inf_h; + //CH--char *buff;// = (char *)malloc(sizeof(char *)*size);//; + int ret; + int ret_sig = atoi(argv[2]); + int sig_show_all = atoi(argv[3]); + int sig_choose_all = atoi(argv[4]); + size_t size; + while ((dp = readdir(dirp)) != NULL) + { + memcpy(vfile,vfile_ori,SIZE_FILE); + strncat(vfile, dp->d_name, strlen(dp->d_name));//strcat(vfile,dp->d_name);//vfile = dp->d_name;//fileinfo->name;//&((fileinfo->name)[0]);//argv[1];// "0003.dat"; + + stat(vfile, &info); + if(S_ISDIR(info.st_mode)) + continue; + buff_inf_h = buff_get(vfile); + if (buff_inf_h->size == 0) + { + printf("read file failed of %s.\n",vfile); + return -1; + } + if (sig_choose_all == 0) + { + int divide_h = 10; + int divide_s = 0; + int divide_e = 1; + + size = (buff_inf_h->size / divide_h)*(divide_e - divide_s); + + //CH--buff = (char *)malloc(sizeof(char *)*size); + //CH--memcpy(buff, buff_inf_h->buff + (buff_inf_h->size / divide_h)*divide_s, size); + + //printf("buff_whole is %p.\n", buff_inf_h->buff); + //printf("buff_inf_h.size/divide_h is %x.\n", buff_inf_h->size / divide_h); + //printf("buff starts from(>=) %p.\n", buff); + //printf("buff ends at(<) %p.\n", buff + size - 1);//+size + ret = AV_container_identify(buff_inf_h->buff + (buff_inf_h->size / divide_h)*divide_s, size);;//CCH--ret = AV_container_identify(buff, size);//CH--(const char *)? + //CH--free(buff); + //CH--buff = NULL; + //printf("define : %d.\n", ret); + //printf("size of whole buff is %d.\n", buff_inf_h->size);//size - 1 + //printf("size of chosen buff is %d.\n", size); + } + else + ret = AV_container_identify(buff_inf_h->buff, buff_inf_h->size); + if (sig_show_all == 0) + { + if (ret != ret_sig) + { + printf("-----------------data of [%s] is kind of [%d].\n", dp->d_name, ret); + //printf("define : %d.\n", ret); + printf("size of whole buff is %d.\n", buff_inf_h->size); + if (sig_choose_all == 0) + printf("size of chosen buff is %d.\n", size); + + } + } + else + { + printf("-----------------data of [%s] is kind of [%d].\n", dp->d_name, ret); + printf("size of whole buff is %d.\n", buff_inf_h->size); + if (sig_choose_all == 0) + printf("size of chosen buff is %d.\n", size); + } + free(buff_inf_h->buff);//CH-- + buff_inf_h->buff = NULL;//CH-- + free(buff_inf_h);//CH-- + buff_inf_h = NULL;//CH-- + } + /* + free(buff); + + free(buff_inf_h->buff); + buff_inf_h = NULL; + free(buff_inf_h); + return ret; + */ + + //CH--free(buff_inf_h->buff); + //CH--buff_inf_h = NULL; + //CH--free(buff_inf_h); + + free(vfile); + free(vfile_ori); + + return ret; +} |