linx/SIC
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Initial commit

Browse Source
This commit is contained in:
胡楠
2023-02-28 09:33:09 +08:00
commit 5dedfaa5d9
36 changed files with 2491 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
code/datadifflib/Makefile Normal file
View File

@@ -0,0 +1,7 @@
scan:main.c md5.c
gcc main.c md5.c -static -lsqlite3 -lpthread -ldl -lm -o scan -Wall -g
.PHONY:clean
clean:
rm scan

699
code/datadifflib/main.c Normal file
View File

@@ -0,0 +1,699 @@
#include "scan.h"
/* 主函数
*
*
*
*/
//#define PRINT_TEST
int main(int argc, const char *argv[])
{
/*****************获取数据库*********************************************/
struct isoinfo iso;//镜像信息结构体
char content[1024];
memset(content,0,sizeof(content));
if(get_iso_info(&iso)!=0)//获取当前系统id
{
sprintf(content,"iso|2|iso_name:%s|镜像在数据库中未找到",iso.name);
}
else
{
sprintf(content,"iso|1|iso_name:%s|iso_version:%s|iso_md5:%s|iso_id:%d|iso_issue=%s",iso.name,iso.version,iso.md5,iso.id,iso.issue);
}
put_diff_log(content);
/******************获取镜像信息*****************************************/
FILE *fp;
char now_name[256];
struct pkginfo now_pkg,data_pkg;
struct fileinfo now_file,data_file;
struct datalib file_list;
//获取当前系统安装包列表
if ((fp = fopen(PKGLIST,"r")) == NULL)
{
printf("can't open PKGLIST\n");
return 0;
}
int progress_i=0,progress_j=0,progress_sum=0,progress_sleep=0;
char progress_bar[101]={0};
if ( argc >=2 )
{
progress_sleep=atoi(argv[1]);
}
//每读取一条，便在数据库中搜索该安装包是否存在。
while(fgets(now_name,255,fp) != NULL)
{
/***********************读取当前安装包列表*******************************/
get_now_pkg(now_name,&now_pkg);
if ( argc >= 3 )
{
progress_sum=atoi(argv[2]);
progress_j=progress_i*100/progress_sum;
printf("[%-100s] [%s] [%d%%]\n",progress_bar,now_pkg.name,progress_j);
fflush(stdout);
progress_bar[progress_j]='#';
progress_i++;
}
/***************************打印对比进度********************************/
memset(content,0,sizeof(content));
if(get_pkg_info(&iso,&now_pkg,&data_pkg,FROM_NOWISO) == 0)
{
sprintf(content,"pkg|1|name:%s|version:%s|md5:%s",data_pkg.name,data_pkg.version,data_pkg.md5);
put_diff_log(content);
//安装包是默认安装
}
else if(get_pkg_info(&iso,&now_pkg,&data_pkg,FROM_DATALIB) == 0)
{
sprintf(content,"pkg|2|name:%s|version:%s|md5:%s",data_pkg.name,data_pkg.version,data_pkg.md5);
put_diff_log(content);
//安装包在数据库中存在
}
else
{
sprintf(content,"pkg|3|name:%s|version:%s",now_pkg.name,now_pkg.version);
put_diff_log(content);
continue;
//安装包在数据库不存在
}
/**************根据当前安装包列表搜索数据库中该安装包是否存在*************/
get_datafile_list(data_pkg,&data_file,&file_list);
//获取安装包文件列表
int i=0;
for(i=file_list.ncolumn;i<((file_list.nrow+1)*file_list.ncolumn);i=i+file_list.ncolumn)
{
memset(&data_file,0,sizeof(data_file));
memset(&now_file,0,sizeof(now_file));
get_datafile_info(i,&file_list,&data_file);
//获取数据库中该安装包文件信息
if(get_nowfile_info(&now_file,&data_file) == -1)
{
//文件在当前系统不存在
memset(content,0,sizeof(content));
sprintf(content,"file|3|name@@%s|type@@%s|md5@@%s|permis@@%s|owner@@%d|genus@@%d|acl@@%s|capable@@%s|mac@@%s|"\
,data_file.pathname,data_file.type,data_file.md5,data_file.permis,data_file.owner,data_file.genus\
,data_file.acl,data_file.capable,data_file.mac);
clean_n(content);
put_diff_log(content);
continue;
}
//获取当前系统中该文件信息
diff_file_info(now_file,data_file);
//对比数据库中与当前系统中文件信息的差异
usleep(progress_sleep);
//对程序性能进行限制
}
sqlite3_free_table(file_list.db_result);
}
/**************对比当前安装包列表文件信息是否正常*************/
fclose(fp);
return 0;
}
/*
* 功能 根据当前检测系统来获取当前镜像信息，获取依据为本系统的/etc/issue文件。
* 输入 镜像信息结构体
* 输出 正确返回镜像id 错误返回 -1
*/
int get_iso_info(struct isoinfo *iso)
{
FILE *fp;
int i = 0;
//获取当前系统版本
if ((fp = fopen("/etc/issue","r")) == NULL)
{
printf("can't open /etc/issue\n");
return -1;
}
char c;
memset(iso->issue,0,sizeof(iso->issue));
while( (c=getc(fp)) != EOF)
{
if(c != '\n')
{
iso->issue[i] = c;
}
else
{
iso->issue[i] = ' ';
}
i++;
}
// printf("%s\n",iso->issue);
//获取数据库product表中的系统版本 和镜像id
sqlite3 *db;
int ret = 0;
ret = sqlite3_open(DATALIB,&db);
if(ret != SQLITE_OK)
{
printf("open database error : %s\n",sqlite3_errmsg(db));
sqlite3_close(db);
return -1;
}
/*获取数据库*/
int nrow,ncolumn;
char ** db_result;
ret = 0;
char select_query[1024];
iso->issue[strlen(iso->issue)-1]=0;
sprintf(select_query,"select product_id,product_name,product_version,product_md5 from product where product_issue=\"%s\";",iso->issue);
ret = sqlite3_get_table(db,select_query,&db_result,&nrow,&ncolumn,NULL);
if(ret != SQLITE_OK)
{
printf("openside database error : %s\n",sqlite3_errmsg(db));
sqlite3_close(db);
return -1;
}
if(nrow < 1)
{
printf("未找到该镜像\n");
sqlite3_close(db);
return -2;
}
for(i=ncolumn;i<(nrow+1)*ncolumn;i=i+ncolumn)
{
iso->id=atoi(db_result[i]);
strcpy(iso->name,db_result[i+1]);
strcpy(iso->version,db_result[i+2]);
strcpy(iso->md5,db_result[i+3]);
}
sqlite3_close(db);
fclose(fp);
sqlite3_free_table(db_result);
return 0;
}
/*
* 功能 对pkglist文件读取出的数据进行处理，输入到当前安装包信息结构体中。
*
* 输入 获取到的安装包名 当前安装包结构体
* 输出 正确返回0
*/
int get_now_pkg(char *now_name,struct pkginfo *now_pkg)
{
//获取当前系统安装包名
int a=0;
int b=0;
while(now_name[a] != '\n')
{
if(now_name[a] != ' ')
{
now_pkg->name[a] = now_name[a];
a++;
}
else
{
now_pkg->name[a] = '\0';
a++;
break;
}
}
//获取当前系统安装包版本
while(now_name[a] != '\n')
{
now_pkg->version[b] = now_name[a];
a++;
b++;
}
now_pkg->version[b] = '\0';
return 0;
}
/*
* 功能 根据标志位，获取数据库中安装包信息。
* 标志位
* FROM_PKGID:根据安装包id获取安装包信息
* FROM_NOWISO:根据安装包和安装包版本在当前产品默认安装包的包列表中查找
* FROM_DATALIB:根据安装包名和安装包版本遍历整个数据库
*
* 输入 镜像信息结构体 当前安装包信息结构体 数据库安装包信息结构体 获取标志位
* 输出 正确返回镜像id 错误返回 -1
*/
int get_pkg_info(struct isoinfo *iso,struct pkginfo *now_pkg,struct pkginfo *data_pkg,int statu)
{
sqlite3 *db;
int ret = 0;
ret = sqlite3_open(DATALIB,&db);
if(ret != SQLITE_OK)
{
printf("open database error : %s\n",sqlite3_errmsg(db));
sqlite3_close(db);
return -1;
}
/*获取数据库*/
int nrow,ncolumn;
char ** db_result;
ret = 0;
char select_query[2048];
switch (statu)
{
case FROM_PKGID:
//根据安装包id获取安装包信息
sprintf(select_query,"select pkg_name,pkg_version,pkg_id,pkg_md5 from pkg where pkg_id = %d;",now_pkg->id);
break;
case FROM_NOWISO:
//根据安装包和安装包版本在当前产品默认安装包的包列表中查找
sprintf(select_query,"select pkg.pkg_name,pkg.pkg_version,pkg.pkg_id,pkg.pkg_md5 from product_and_pkg,pkg where product_id=%d and pkg.pkg_name=\"%s\" and pkg.pkg_version=\"%s\" and product_and_pkg.pkg_id=pkg.pkg_id;",iso->id,now_pkg->name,now_pkg->version);
break;
case FROM_DATALIB:
//根据安装包名和安装包版本遍历整个数据库
sprintf(select_query,"select pkg.pkg_name,pkg.pkg_version,pkg.pkg_id,pkg.pkg_md5 from pkg where pkg_name=\"%s\" and pkg_version=\"%s\" ;",now_pkg->name,now_pkg->version);
break;
}
//printf("%s\n",select_query);
ret = sqlite3_get_table(db,select_query,&db_result,&nrow,&ncolumn,NULL);
if(ret != SQLITE_OK)
{
printf("未发现当前安装包信息\n");
printf("openside database error : %s\n",sqlite3_errmsg(db));
sqlite3_close(db);
return -1;
}
if(nrow != 0)
{
strcpy(data_pkg->name,db_result[4]);
strcpy(data_pkg->version,db_result[5]);
data_pkg->id=atoi(db_result[6]);
strcpy(data_pkg->md5,db_result[7]);
#ifdef PRINT_TEST
printf("pkg_name:%s pkg_version:%s pkg_id:%s pkg_md5:%s\n ",db_result[4],db_result[5],db_result[6],db_result[7]);
#endif
sqlite3_close(db);
sqlite3_free_table(db_result);
return 0;
}
else
{
sqlite3_close(db);
sqlite3_free_table(db_result);
return -2;
}
}
/*
* 功能 根据安装包信息，获取数据库中该安装包的文件列表
*
* 输入 数据库安装包信息结构体 数据库文件信息结构体
* 输出 正确返回0 错误返回-1
*/
int get_datafile_list(struct pkginfo data_pkg,struct fileinfo *data_file,struct datalib *file_list)
{
int ret = 0;
char select_query[2048];
sqlite3 *db;
ret = sqlite3_open(DATALIB,&db);
if(ret != SQLITE_OK)
{
printf("open database error : %s\n",sqlite3_errmsg(db));
sqlite3_close(db);
return -1;
}
/*获取数据库*/
sprintf(select_query,"select file_id,file_name,file_pathname,file_md5,file_type,file_owner,file_genus,file_permis,file_size,file_acl,file_capable,file_level,file_mac from file where pkg_id=%d;",data_pkg.id);
//根据安装包id获取其对应文件的信息
ret = sqlite3_get_table(db,select_query,&file_list->db_result,&file_list->nrow,&file_list->ncolumn,NULL);
if(ret != SQLITE_OK)
{
printf("未发现当前安装包信息\n");
printf("openside database error : %s\n",sqlite3_errmsg(db));
sqlite3_close(db);
return -1;
}
sqlite3_close(db);
return 0;
}
/* 将获取的文件信息转入到文件信息结构体中
*
* 输入 获取列表中第几个文件 安装包文件列表信息结构体 文件信息结构体
* 输出 无
*
*/
void get_datafile_info(int i,struct datalib *file_list,struct fileinfo *data_file)
{
data_file->id=atoi(file_list->db_result[i]);
strcpy(data_file->name,file_list->db_result[i+1]);
strcpy(data_file->pathname,file_list->db_result[i+2]);
strcpy(data_file->md5,file_list->db_result[i+3]);
strcpy(data_file->type,file_list->db_result[i+4]);
data_file->owner=atoi(file_list->db_result[i+5]);
data_file->genus=atoi(file_list->db_result[i+6]);
strcpy(data_file->permis,file_list->db_result[i+7]);
data_file->size=atoi(file_list->db_result[i+8]);
strcpy(data_file->acl,file_list->db_result[i+9]);
strcpy(data_file->capable,file_list->db_result[i+10]);
strcpy(data_file->level,file_list->db_result[i+11]);
strcpy(data_file->mac,file_list->db_result[i+12]);
#ifdef PRINT_TEST
printf("file_name=%s\n",data_file->name);
#endif
}
/* 获取当期系统的具体文件信息
*
* 输入 当前文件信息结构体 数据库文件信息结构体
* 输出 获取正常返回0 错误返回-1
*
*/
int get_nowfile_info( struct fileinfo *now_file,struct fileinfo *data_file )
{
struct stat filestat;
strcpy(now_file->name,data_file->name);
strcpy(now_file->pathname,data_file->pathname);
if(lstat(now_file->pathname,&filestat) < 0)
{
printf("cannot access the file %s\n",now_file->pathname);
perror("error : ");
return -1;
}
strcpy(now_file->md5,"");//清除now_file->md5
//获取文件类型
switch(filestat.st_mode & S_IFMT){
case S_IFDIR :
sprintf(now_file->type,"d");
break;
case S_IFBLK :
sprintf(now_file->type,"b");
//获取文件md5值
get_file_md5(now_file->pathname,now_file->md5);
break;
case S_IFSOCK :
sprintf(now_file->type,"s");
//获取文件md5值
get_file_md5(now_file->pathname,now_file->md5);
break;
case S_IFLNK :
sprintf(now_file->type,"l");
break;
case S_IFREG :
sprintf(now_file->type,"-");
//获取文件md5值
get_file_md5(now_file->pathname,now_file->md5);
break;
case S_IFCHR :
sprintf(now_file->type,"c");
break;
case S_IFIFO :
sprintf(now_file->type,"p");
break;
}
//获取文件大小
now_file->size = filestat.st_size;
//获取文件权限
sprintf(now_file->permis,"%o",(filestat.st_mode&S_ISUGV)+(filestat.st_mode&S_IRWXU)+(filestat.st_mode&S_IRWXG)+(filestat.st_mode&S_IRWXO));
//获取文件属主
now_file->owner=filestat.st_uid;
//获取文件属组
now_file->genus=filestat.st_gid;
//获取文件acl
get_acl(now_file);
//获取文件capable
get_cap(now_file);
//获取文件mac
get_mac(now_file);
//获取文件level
//now_file->level=NULL;
return 0 ;
}
/* 对比当前文件信息和数据库中文件信息的差异
*
* 输入 当前文件信息结构体 数据库文件信息结构体
* 输出 正确返回0
*
*/
int diff_file_info(struct fileinfo now_file,struct fileinfo data_file)
{
char content[9][512]={{0}};
char log[1024]={0};
int status=1;
sprintf(content[0],"name@%s",now_file.pathname);
if(strcmp(now_file.type,data_file.type) != 0)
{
sprintf(content[1],"type@%s@%s",now_file.type,data_file.type);
status=2;
}
if(now_file.type[0] != 'l')//如果文件类型为链接文件，则不需要比较md5值
{
if(strcmp(now_file.md5,data_file.md5) != 0)
{
sprintf(content[2],"md5@%s@%s",now_file.md5,data_file.md5);
status=2;
}
}
if(strcmp(now_file.permis,data_file.permis) != 0)
{
sprintf(content[3],"permis@%s@%s",now_file.permis,data_file.permis);
status=2;
}
if(now_file.owner != data_file.owner)
{
sprintf(content[4],"owner@%d@%d",now_file.owner,data_file.owner);
status=2;
}
if(now_file.genus != data_file.genus)
{
sprintf(content[5],"genus@%d@%d",now_file.genus,data_file.genus);
status=2;
}
if(strcmp(now_file.acl,data_file.acl) != 0)
{
sprintf(content[6],"acl@%s@%s",now_file.acl,data_file.acl);
status=2;
}
if(strcmp(now_file.capable,data_file.capable) != 0)
{
sprintf(content[7],"capable@%s@%s",now_file.capable,data_file.capable);
status=2;
}
if(strcmp(now_file.mac,data_file.mac) != 0)
{
sprintf(content[8],"mac@%s@%s",now_file.mac,data_file.mac);
status=2;
}
sprintf(log,"file|%d|%s|%s|%s|%s|%s|%s|%s|%s|%s|",status,content[0],content[1],content[2],content[3],content[4],content[5],content[6],content[7],content[8]);
clean_n(log);
put_diff_log(log);
return 0 ;
}
/* 打印信息到记录日志
*
* 输入 需要打印的信息指针
* 输出 正确返回0 失败返回-1
*/
int put_diff_log(char *content)
{
FILE *fp=NULL;
if ((fp = fopen(FILEDIFF,"a")) == NULL)
{
perror("can't open FILEDIFF:");
return -1;
}
fputs(content,fp);
putc('\n',fp);
fflush(fp);
fclose(fp);
return 0;
}
/* 获取shell指令的输出
*
* 输入 shell指令指针 输出信息指针 输出信息的大小
* 输出 正确返回0 错误返回-1
*/
int myshell(char *command,char *buff,int num)
{
FILE *fp;
if ((fp=popen(command,"r")) == NULL)
{
perror("popen fp error");
return -1;
}
fread(buff,sizeof(char),num,fp);
if (pclose(fp) == -1)
{
perror("pclose fp error");
return -2;
}
return 0;
}
/* 获取文件acl
*
* 将根据当前结构提中文件名获取该文件的acl并存储到结构体中
*
* 输入 文件信息结构体
* 输出 无返回
*/
void get_acl(struct fileinfo *now_file)
{
memset(now_file->acl,0,sizeof(now_file->acl));
if ( now_file->type[0] == 'c' || now_file->type[0] == 'p' || now_file->type[0] == 'l')
return ;
char command[1024]={0};
sprintf(command,"getfacl --absolute-name \"%s\" 2>&1",now_file->pathname);
myshell(command,now_file->acl,sizeof(now_file->acl));
now_file->acl[strlen(now_file->acl)-2]='\0';
}
/* 获取文件mac
*
* 将根据当前结构提中文件名获取该文件的mac并存储到结构体中
*
* 输入 文件信息结构体
* 输出 无返回
*/
void get_mac(struct fileinfo *now_file)
{
memset(now_file->mac,0,sizeof(now_file->mac));
if ( now_file->type[0] == 'c' || now_file->type[0] == 'p' || now_file->type[0] == 'l')
return ;
char command[1024]={0};
sprintf(command,"getfmac \"%s\" 2>/dev/null",now_file->pathname);
myshell(command,now_file->mac,sizeof(now_file->mac));
now_file->mac[strlen(now_file->mac)-1]='\0';
}
/* 获取文件cap
*
* 将根据当前结构提中文件名获取该文件的cap并存储到结构体中
*
* 输入 文件信息结构体
* 输出 无返回
*/
void get_cap(struct fileinfo *now_file)
{
memset(now_file->capable,0,sizeof(now_file->capable));
if ( now_file->type[0] == 'c' || now_file->type[0] == 'p' || now_file->type[0] == 'l')
return ;
char command[1024]={0};
sprintf(command,"getcap \"%s\" 2>/dev/null",now_file->pathname);
myshell(command,now_file->capable,sizeof(now_file->capable));
now_file->capable[strlen(now_file->capable)-1]='\0';
}
void clean_n(char *log){
int i=0;
while(i<strlen(log))
{
if(log[i] == '\n')
{
log[i]=' ';
}
i++;
}
}

231
code/datadifflib/md5.c Normal file
View File

@@ -0,0 +1,231 @@
#include "md5.h"
#include <memory.h>
unsigned char PADDING[] =
{
0x80,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0
};
void MD5Init(MD5_CTX *context)
{
context->count[0] = 0;
context->count[1] = 0;
context->state[0] = 0x67452301;
context->state[1] = 0xEFCDAB89;
context->state[2] = 0x98BADCFE;
context->state[3] = 0x10325476;
}
void MD5Update(MD5_CTX *context, unsigned char *input, unsigned int inputlen)
{
unsigned int i = 0;
unsigned int index = 0;
unsigned int partlen = 0;
index = (context->count[0] >> 3) & 0x3F;
partlen = 64 - index;
context->count[0] += inputlen << 3;
if(context->count[0] < (inputlen << 3))
context->count[1]++;
context->count[1] += inputlen >> 29;
if(inputlen >= partlen)
{
memcpy(&context->buffer[index], input,partlen);
MD5Transform(context->state, context->buffer);
for(i = partlen; i+64 <= inputlen; i+=64)
MD5Transform(context->state, &input[i]);
index = 0;
}
else
{
i = 0;
}
memcpy(&context->buffer[index], &input[i], inputlen-i);
}
void MD5Final(MD5_CTX *context, unsigned char digest[16])
{
unsigned int index = 0,padlen = 0;
unsigned char bits[8];
index = (context->count[0] >> 3) & 0x3F;
padlen = (index < 56)?(56-index):(120-index);
MD5Encode(bits, context->count, 8);
MD5Update(context, PADDING, padlen);
MD5Update(context, bits, 8);
MD5Encode(digest, context->state, 16);
}
void MD5Encode(unsigned char *output,unsigned int *input,unsigned int len)
{
unsigned int i = 0;
unsigned int j = 0;
while(j < len)
{
output[j] = input[i] & 0xFF;
output[j+1] = (input[i] >> 8) & 0xFF;
output[j+2] = (input[i] >> 16) & 0xFF;
output[j+3] = (input[i] >> 24) & 0xFF;
i++;
j += 4;
}
}
void MD5Decode(unsigned int *output, unsigned char *input, unsigned int len)
{
unsigned int i = 0;
unsigned int j = 0;
while(j < len)
{
output[i] = (input[j]) |
(input[j+1] << 8) |
(input[j+2] << 16) |
(input[j+3] << 24);
i++;
j += 4;
}
}
void MD5Transform(unsigned int state[4], unsigned char block[64])
{
unsigned int a = state[0];
unsigned int b = state[1];
unsigned int c = state[2];
unsigned int d = state[3];
unsigned int x[64];
MD5Decode(x,block,64);
FF(a, b, c, d, x[ 0], 7, 0xd76aa478); /* 1 */
FF(d, a, b, c, x[ 1], 12, 0xe8c7b756); /* 2 */
FF(c, d, a, b, x[ 2], 17, 0x242070db); /* 3 */
FF(b, c, d, a, x[ 3], 22, 0xc1bdceee); /* 4 */
FF(a, b, c, d, x[ 4], 7, 0xf57c0faf); /* 5 */
FF(d, a, b, c, x[ 5], 12, 0x4787c62a); /* 6 */
FF(c, d, a, b, x[ 6], 17, 0xa8304613); /* 7 */
FF(b, c, d, a, x[ 7], 22, 0xfd469501); /* 8 */
FF(a, b, c, d, x[ 8], 7, 0x698098d8); /* 9 */
FF(d, a, b, c, x[ 9], 12, 0x8b44f7af); /* 10 */
FF(c, d, a, b, x[10], 17, 0xffff5bb1); /* 11 */
FF(b, c, d, a, x[11], 22, 0x895cd7be); /* 12 */
FF(a, b, c, d, x[12], 7, 0x6b901122); /* 13 */
FF(d, a, b, c, x[13], 12, 0xfd987193); /* 14 */
FF(c, d, a, b, x[14], 17, 0xa679438e); /* 15 */
FF(b, c, d, a, x[15], 22, 0x49b40821); /* 16 */
/* Round 2 */
GG(a, b, c, d, x[ 1], 5, 0xf61e2562); /* 17 */
GG(d, a, b, c, x[ 6], 9, 0xc040b340); /* 18 */
GG(c, d, a, b, x[11], 14, 0x265e5a51); /* 19 */
GG(b, c, d, a, x[ 0], 20, 0xe9b6c7aa); /* 20 */
GG(a, b, c, d, x[ 5], 5, 0xd62f105d); /* 21 */
GG(d, a, b, c, x[10], 9, 0x2441453); /* 22 */
GG(c, d, a, b, x[15], 14, 0xd8a1e681); /* 23 */
GG(b, c, d, a, x[ 4], 20, 0xe7d3fbc8); /* 24 */
GG(a, b, c, d, x[ 9], 5, 0x21e1cde6); /* 25 */
GG(d, a, b, c, x[14], 9, 0xc33707d6); /* 26 */
GG(c, d, a, b, x[ 3], 14, 0xf4d50d87); /* 27 */
GG(b, c, d, a, x[ 8], 20, 0x455a14ed); /* 28 */
GG(a, b, c, d, x[13], 5, 0xa9e3e905); /* 29 */
GG(d, a, b, c, x[ 2], 9, 0xfcefa3f8); /* 30 */
GG(c, d, a, b, x[ 7], 14, 0x676f02d9); /* 31 */
GG(b, c, d, a, x[12], 20, 0x8d2a4c8a); /* 32 */
/* Round 3 */
HH(a, b, c, d, x[ 5], 4, 0xfffa3942); /* 33 */
HH(d, a, b, c, x[ 8], 11, 0x8771f681); /* 34 */
HH(c, d, a, b, x[11], 16, 0x6d9d6122); /* 35 */
HH(b, c, d, a, x[14], 23, 0xfde5380c); /* 36 */
HH(a, b, c, d, x[ 1], 4, 0xa4beea44); /* 37 */
HH(d, a, b, c, x[ 4], 11, 0x4bdecfa9); /* 38 */
HH(c, d, a, b, x[ 7], 16, 0xf6bb4b60); /* 39 */
HH(b, c, d, a, x[10], 23, 0xbebfbc70); /* 40 */
HH(a, b, c, d, x[13], 4, 0x289b7ec6); /* 41 */
HH(d, a, b, c, x[ 0], 11, 0xeaa127fa); /* 42 */
HH(c, d, a, b, x[ 3], 16, 0xd4ef3085); /* 43 */
HH(b, c, d, a, x[ 6], 23, 0x4881d05); /* 44 */
HH(a, b, c, d, x[ 9], 4, 0xd9d4d039); /* 45 */
HH(d, a, b, c, x[12], 11, 0xe6db99e5); /* 46 */
HH(c, d, a, b, x[15], 16, 0x1fa27cf8); /* 47 */
HH(b, c, d, a, x[ 2], 23, 0xc4ac5665); /* 48 */
/* Round 4 */
II(a, b, c, d, x[ 0], 6, 0xf4292244); /* 49 */
II(d, a, b, c, x[ 7], 10, 0x432aff97); /* 50 */
II(c, d, a, b, x[14], 15, 0xab9423a7); /* 51 */
II(b, c, d, a, x[ 5], 21, 0xfc93a039); /* 52 */
II(a, b, c, d, x[12], 6, 0x655b59c3); /* 53 */
II(d, a, b, c, x[ 3], 10, 0x8f0ccc92); /* 54 */
II(c, d, a, b, x[10], 15, 0xffeff47d); /* 55 */
II(b, c, d, a, x[ 1], 21, 0x85845dd1); /* 56 */
II(a, b, c, d, x[ 8], 6, 0x6fa87e4f); /* 57 */
II(d, a, b, c, x[15], 10, 0xfe2ce6e0); /* 58 */
II(c, d, a, b, x[ 6], 15, 0xa3014314); /* 59 */
II(b, c, d, a, x[13], 21, 0x4e0811a1); /* 60 */
II(a, b, c, d, x[ 4], 6, 0xf7537e82); /* 61 */
II(d, a, b, c, x[11], 10, 0xbd3af235); /* 62 */
II(c, d, a, b, x[ 2], 15, 0x2ad7d2bb); /* 63 */
II(b, c, d, a, x[ 9], 21, 0xeb86d391); /* 64 */
state[0] += a;
state[1] += b;
state[2] += c;
state[3] += d;
}
int get_file_md5(const char *file_path, char *md5_str)
{
int i;
int fd;
int ret;
unsigned char data[READ_DATA_SIZE];
unsigned char md5_value[MD5_SIZE];
MD5_CTX md5;
fd = open(file_path, O_RDONLY);
if (-1 == fd)
{
perror("open");
return -1;
}
// init md5
MD5Init(&md5);
while (1)
{
ret = read(fd, data, READ_DATA_SIZE);
if (-1 == ret)
{
perror("read");
return -1;
}
MD5Update(&md5, data, ret);
if (0 == ret || ret < READ_DATA_SIZE)
{
break;
}
}
close(fd);
MD5Final(&md5, md5_value);
for(i = 0; i < MD5_SIZE; i++)
{
snprintf(md5_str + i*2, 2+1, "%02x", md5_value[i]);
}
md5_str[MD5_STR_LEN] = '\0'; // add end
return 0;
}

62
code/datadifflib/md5.h Normal file
View File

@@ -0,0 +1,62 @@
#ifndef MD5_H
#define MD5_H
#define READ_DATA_SIZE 1024
#define MD5_SIZE 16
#define MD5_STR_LEN (MD5_SIZE * 2)
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
typedef struct
{
unsigned int count[2];
unsigned int state[4];
unsigned char buffer[64];
} MD5_CTX;
#define F(x,y,z) ((x & y) | (~x & z))
#define G(x,y,z) ((x & z) | (y & ~z))
#define H(x,y,z) (x^y^z)
#define I(x,y,z) (y ^ (x | ~z))
#define ROTATE_LEFT(x,n) ((x << n) | (x >> (32-n)))
#define FF(a,b,c,d,x,s,ac) \
{ \
a += F(b,c,d) + x + ac; \
a = ROTATE_LEFT(a,s); \
a += b; \
}
#define GG(a,b,c,d,x,s,ac) \
{ \
a += G(b,c,d) + x + ac; \
a = ROTATE_LEFT(a,s); \
a += b; \
}
#define HH(a,b,c,d,x,s,ac) \
{ \
a += H(b,c,d) + x + ac; \
a = ROTATE_LEFT(a,s); \
a += b; \
}
#define II(a,b,c,d,x,s,ac) \
{ \
a += I(b,c,d) + x + ac; \
a = ROTATE_LEFT(a,s); \
a += b; \
}
void MD5Init(MD5_CTX *context);
void MD5Update(MD5_CTX *context, unsigned char *input, unsigned int inputlen);
void MD5Final(MD5_CTX *context, unsigned char digest[16]);
void MD5Transform(unsigned int state[4], unsigned char block[64]);
void MD5Encode(unsigned char *output, unsigned int *input, unsigned int len);
void MD5Decode(unsigned int *output, unsigned char *input, unsigned int len);
int get_file_md5(const char *file_path, char *value);
#endif

123
code/datadifflib/pkg.sh Executable file
View File

@@ -0,0 +1,123 @@
#!/bin/bash
WORKDIR=./pkg
PKGLIST=$WORKDIR/pkglist
#硬限制
#限制进程最大内存使用100M
ulimit -Hv 102400
#限制进程最大虚拟内存为200M
ulimit -Hv 204800
#限制进程最大cpu占用时间为1秒
ulimit -Ht 1
#程序处理延迟(微秒)
progress_sleep=500
#初始化环境
function data_init(){
if [ -d $WORKDIR ];then
rm -rf $WORKDIR/log
fi
mkdir $WORKDIR/log
if [ -f $WORKDIR/pkglist ];then
rm pkglist
fi
}
#获取当前安装包列表
function data_pkglist(){
iso_version=`grep -o -E "42|60|80|76|77" /etc/issue`
case $iso_version in
42)
pkginfo -i >> $PKGLIST
;;
60|80)
#dpkg -l | awk '{if($1=="ii"){sub(":amd64","",$2);sub("[0-9]:","",$3);print $2"_"$3"_"$4}}' >> $PKGLIST
awk 'BEGIN{i=0;j=0;k=0}{if($1 == "Package:"){A[i]=$2;i+=1};if($1 == "Version:"){B[j]=$2;j+=1}if($1 == "Architecture:"){C[k]=$2;k=k+1}}END{for(i=0;i<k;i++)printf "%s_%s_%s\n",A[i],B[i],C[i]}' /var/lib/dpkg/status >> pkglist
;;
76|77)
rpm -qa | while read pkglist
do
pkg_isoname=`echo $pkglist | awk -F "-[0-9]" '{printf $1}'`
pkgversion=`ECTYPE_TRANSMIT_NOT_PRESENTho $pkglist | awk -F "$pkgname-" '{printf $2}'`
echo $pkgname $pkgversion >> $PKGLIST
done
;;
*)
echo "版本识别失败"
exit
;;
esac
}
#搭建工作环境
function data_env(){
tail -n +39 $0 > $WORKDIR/pkg.tar.gz
tar xf $WORKDIR/pkg.tar.gz
cd $WORKDIR/pkg
}
#对对比程序资源进行设置
function data_set(){
echo "设置进程资源"
}
#启动对比程序
function data_run(){
data_sum=`cat $PKGLIST | wc -l`
./scan $data_sum $progress_sleep
data_pid=$?
}
#监视对比程序运行状态
function data_control(){
echo "监视程序运行状态"
}
#生成检测报告
function data_report(){
echo "生成对比报告"
}
function main(){
data_init
data_pkglist
data_env
data_set
data_run
data_control
data_report
}
main
exit 0

BIN
code/datadifflib/scan Executable file
View File

Binary file not shown.

130
code/datadifflib/scan.c Normal file
View File

@@ -0,0 +1,130 @@
#include "scan.h"
//轮循读取文件名
void read_filename(char *dir)
{
DIR *dfd;
struct dirent *dp;
char name[MAX_NAME];
if ((dfd = opendir(dir)) == NULL)
{
printf("dir_order:can't open %s\n %s",dir,strerror(errno));
return ;
}
while ((dp = readdir(dfd)) != NULL)
{
if (strncmp(dp->d_name, ".", 1) == 0)
continue;
if (strlen(dir) + strlen(dp->d_name) + 2 > sizeof(name))//确定申请的空间够大
{
printf("dir_order:name %s %s too long\n",dir,dp->d_name);
return ;
}
else
{
memset(name,0,sizeof(char)*MAX_NAME);
sprintf(name,"%s/%s",dir,dp->d_name);
sprintf(file.name,name);
if(4 == dp->d_type)//判断文件为目录，则递归查询
{
read_filename(name);
printf("%s\n",name);
}
printf("%s\n" ,name);
}
}
closedir(dfd);
return ;
}
//获取文件部分信息
void read_filestat(struct fileinfo file)
{
struct stat filestat;
if(lstat(file.name,&filestat) <= 0)
{
printf("cannot access the file %s",file.name);
return;
}
//获取文件类型
switch(filestat.st_mode & S_IFMT){
case S_IFDIR :
sprintf(file.type,"目录");
break;
case S_IFSOCK :
sprintf(file.type," 套接字 ");
break;
case S_IFLNK :
sprintf(file.type," 软链接 ");
break;
case S_IFREG :
sprintf(file.type," 文件 ");
break;
case S_IFCHR :
sprintf(file.type," 字符设备 ");
break;
case S_IFIFO :
sprintf(file.type," 管道 ");
break;
}
//获取文件大小
//printf("%8ld\n" ,file.st_size);//打印文件名和文件大小。
file.size = filestat.st_size;
//获取文件权限
int i = 8;
while(i >= 0)
{
if((filestat.st_mode)& 1<<i)
{
switch(i%3)
{
case 2:
file.mod[9-i] = 'r';
break;
case 1:
file.mod[9-i] = 'w';
break;
case 0:
file.mod[9-i] = 'x';
break;
}
}
else
{
file.mod[9-i] = '-';
}
i--;
}
//获取文件属主
sprintf(file.uid,"%s",getpwuid(filestat.st_uid)->pw_name);
//获取文件属组
sprintf(file.gid,"%s",getgrgid(filestat.st_gid)->gr_name);
}
#if 0
void read_filemd5(struct fileinfo file)
{
}
#endif
#if 0
int main(int argc, const char *argv[])
{
char *dir = NULL;
char *name = NULL;
dir = (char *)malloc(MAX_FILE);
printf("请输入要读取的目录\n");
scanf("%s",dir);
read_filename(dir);
printf("文件名：%s\n",file.name);
return 0;
}
#endif

98
code/datadifflib/scan.h Normal file
View File

@@ -0,0 +1,98 @@
#include <stdio.h>
#include <dirent.h>
#include <malloc.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>
#include <pwd.h>
#include <grp.h>
#include <sqlite3.h>
#include <stdlib.h>
#include "md5.h"
#define MAX 512
#define MAX_NAME 512
#define MAX_FILE 512
#define MAX_ACL 1024
#define MAX_ID 32
#define MAX_TYPE 10
#define MAX_MD5 33
#define MAX_MOD 16
#define MAX_PATH 4096
#define FILEISO 1
#define FILEPKG 2
#define FILEFIL 3
#define FILEMD5 4
#define FILEPER 5
#define FILEOWN 6
#define FILEGEN 7
#define FILETYP 8
#define FROM_PKGID 10
#define FROM_NOWISO 20
#define FROM_DATALIB 30
#define S_ISUGV 07000
#define md5_t char
#define DATALIB "./fileinfo.db"
#define PKGLIST "./pkglist"
#define FILEDIFF "./diff.log"
struct isoinfo{
char name[MAX_NAME]; //镜像名
char version[MAX_NAME];//镜像版本
int id; //镜像id
md5_t md5[MAX_NAME]; //镜像md5_t
char issue[MAX_NAME];
} ;
struct pkginfo{
char name[MAX_NAME]; //安装包名
char version[MAX_NAME];//安装包版本
int id; //安装包id
md5_t md5[MAX_MD5]; //安装包md5
} ;
struct fileinfo{
int id; //文件id
char name[MAX_NAME]; //文件名
char pathname[MAX_PATH];//文件绝对路径名
md5_t md5[MAX_MD5]; //文件md5值
char type[MAX_TYPE]; //文件类型
int owner; //文件属主
int genus; //文件属组
char permis[MAX_ID]; //文件权限
off_t size; //文件大小
char acl[MAX_ACL]; //文件acl
char capable[MAX]; //文件能力
char mac[MAX]; //文件mac
char level[MAX]; //文件等级
} ;
struct datalib{
int nrow;
int ncolumn;
char ** db_result;
} ;
int get_iso_info(struct isoinfo *iso); //获取当前镜像信息
int get_now_pkg(char *now_name,struct pkginfo *now_pkg); //处理当前安装包列表
int get_pkg_info(struct isoinfo *iso,struct pkginfo *now_pkg,struct pkginfo *data_pkg,int statu);//获取安装包信息
int get_datafile_list(struct pkginfo data_pkg,struct fileinfo *data_file,struct datalib *file_list);//根据安装包信息获取对应文件列表
int get_nowfile_info( struct fileinfo *now_file,struct fileinfo *data_file );//获取当前系统文件具体信息
int diff_file_info(struct fileinfo now_file,struct fileinfo data_file);//对比当前系统文件信息和数据库文件信息
void get_datafile_info(int i,struct datalib *file_list,struct fileinfo *data_file);//将获取的文件信息转入到文件信息结构体中
void get_cap(struct fileinfo *now_file);
void get_mac(struct fileinfo *now_file);
void get_acl(struct fileinfo *now_file);
int put_diff_log(char *content);
int myshell(char *command,char *buff,int num);//调用shell命令
void clean_n(char *log);