菜鸟刚入门写的查找文件或者文件夹并加密的程序，希望大牛们多多指教，能帮我优化程序

mkshtk

程序需要实现功能：输入name和path，查找并判断name是文件还是文件夹，若name是文件则选择加密，若是目录则遍历该目录，对下面的文件进行加密。
      加密算法暂时没有写，只是大概要实现功能。希望大牛们多多指点，帮我删减优化掉一些字段，减小耦合度，能实现模块化。谢谢

1. # include <dirent.h>
   
2. # include <string.h>
   
3. # include <stdio.h>
   
4. # include <sys/stat.h>
   
5. # include <stdlib.h>
   
6. # include <unistd.h>
   
7. # include <stdbool.h>
   
8. 
   
9. /*
   
10. 函数名：jiami()
    
11. 作用：解密算法
    
12. 参数：* stream--需要加密的字符串指针
    
13.           * store1--加密完成后的字符串指针
    
14. */
    
15. int jiami(char *stream1, char *store1)
    
16. {       
    
17.         int i, len;
    
18.         len = strlen(stream1);
    
19.         for(i=0; i<len; i++)
    
20.         {       
    
21.                 store1[i] = stream1[i];
    
22.         }
    
23.         return 0;
    
24. }
    
25. 
    
26. /*
    
27. 函数名：jiemi()
    
28. 作用：解密算法
    
29. 参数：* stream--需要解密的字符串指针
    
30.           * store--解密完成后的字符串指针
    
31. */
    
32. int jiemi(char *stream2, char *store2)
    
33. {       
    
34.         int i, len;
    
35.         len = strlen(stream2);
    
36.         for(i=0; i<len; i++)
    
37.         {       
    
38.                 store2[i] = stream2[i]+1;
    
39.         }
    
40.         return 0;
    
41. }
    
42. 
    
43. /*
    
44. 函数名：En_Code()
    
45. 作用：加密函数
    
46. 参数：* pSrc--需要加密的字符串指针
    
47.           * pDst--生成目标文件的文件指针
    
48. */
    
49. void En_Code(char *pSrc, FILE *pDst)
    
50. {       
    
51.         char *out1 = (char *)malloc(sizeof(pSrc)*100);
    
52.         jiami(pSrc,out1);
    
53.         fwrite(out1, strlen(out1), 1, pDst);
    
54.         free(out1);
    
55.         out1 = NULL;
    
56. }
    
57. 
    
58. 
    
59. /*
    
60. 函数名：De_Code()
    
61. 作用：解密函数
    
62. 参数：* pSrc--需要解密的字符串指针
    
63.           * pDst--生成目标文件的文件指针
    
64. */
    
65. void De_Code(char *pSrc, FILE *pDst)
    
66. {       
    
67.         char *out2 = malloc(sizeof(pSrc)*100);
    
68.         jiemi(pSrc,out2);
    
69.         fwrite(out2, strlen(out2), 1, pDst);
    
70.         free(out2);
    
71.         out2 = NULL;
    
72. }
    
73. 
    
74. 
    
75. /*
    
76. 函数名：Open_file()
    
77. 作用：判断源文件和需要生成的目标文件是否能够打开
    
78. 参数：*Src_file--源文件指针
    
79.           *Dst_file--生成目标文件指针
    
80. */
    
81. int  Open_file(char *Src_file, char *Dst_file)
    
82. {
    
83.         FILE *fp1, *fp2;
    
84.         long curpos, flen;
    
85.         char *buff;
    
86.         fp1=fopen(Src_file, "rb");
    
87.         fp2=fopen(Dst_file, "wb+");
    
88.         if(fp1 == NULL)
    
89.         {
    
90.                 printf("cannot open source file.\n");
    
91.                 exit(1);
    
92.         }
    
93.                 if(fp2 == NULL)
    
94.         {
    
95.                 printf("cannot open destination file.\n");
    
96.                 exit(1);
    
97.         }
    
98.        
    
99.         curpos = ftell(fp1);
    
100.         fseek(fp1, 0L, SEEK_END);
     
101.         flen = ftell(fp1);
     
102.         buff = (char *)malloc(flen*10);
     
103.         memset(buff, 0, flen*10);
     
104.         if(buff == NULL)
     
105.         {
     
106.                 fclose(fp1);
     
107.                 return 0;
     
108.         }
     
109.         fseek(fp1, curpos, SEEK_SET);
     
110.         fread(buff, flen, 1, fp1);
     
111.         En_Code(buff,fp2);
     
112.         free(buff);
     
113.         buff = NULL;
     
114.         fclose(fp1);
     
115.         fclose(fp2);
     
116.         return 0;
     
117. }
     
118. 
     
119. /*
     
120. 函数名：File_Exist()
     
121. 作用：判断指定路径下文件是否存在，不查询含子目录
     
122. 参数：* fullpath---完整的路径加上文件名
     
123. */
     
124. bool File_Exist(char * fullpath)
     
125. {
     
126.         if((access(fullpath,F_OK)) == 0)
     
127.                 return true;
     
128.         else
     
129.                 return false;
     
130. }
     
131. 
     
132. /*
     
133. 函数名：List()
     
134. 作用：循环便利目录，并对查找的文件进行加密等操作
     
135. 参数：* path--要查找的目录地址指针
     
136. */
     
137. void List(char * path)
     
138. {
     
139.         struct dirent *ent = NULL;
     
140.         DIR *pDir;
     
141.         char dir[512];
     
142.         struct stat statbuf;
     
143.         char * store_path = "store";
     
144.         char store_fullpath[512];
     
145. 
     
146.         if((pDir = opendir(path)) == NULL)
     
147.         {
     
148.                 fprintf(stderr, "cannot open directory:%s\n", path);
     
149.                 return;
     
150.         }
     
151. 
     
152.         while((ent = readdir(pDir)) != NULL)
     
153.         {
     
154.                 snprintf(dir, 512, "%s/%s", path, ent->d_name);
     
155.                 lstat(dir, &statbuf);
     
156. 
     
157.                 if(S_ISDIR(statbuf.st_mode))
     
158.                 {
     
159.                         if(strcmp(".", ent->d_name) == 0 || strcmp("..", ent->d_name) == 0)
     
160.                                 continue;
     
161.                         List(dir);
     
162.                 }
     
163.                
     
164.                 if(S_ISREG(statbuf.st_mode))
     
165.                 {
     
166.                         snprintf(store_fullpath, 512, "%s/%s", store_path, ent->d_name);
     
167.                         Open_file(dir, store_fullpath);
     
168.                 }
     
169.         }
     
170.         closedir(pDir);
     
171. }
     
172. 
     
173. /*
     
174. 函数名：Search_Dir()
     
175. 作用：遍历查找指定目录下文件是否存在
     
176. 参数：* path---查找的路径
     
177.           * name--文件名
     
178. */
     
179. bool Search_Dir(char * path, char *name)
     
180. {       
     
181.         char dir[50];
     
182.         struct dirent * ent;
     
183.         DIR *pdir;
     
184.         struct stat statbuf;
     
185. 
     
186.         if((pdir = opendir(path)) == NULL)
     
187.         {
     
188.                 fprintf(stderr, "Cannot open directory:%s\n", path);
     
189.                 return false;
     
190.         }
     
191. 
     
192.         while((ent = readdir(pdir))!=NULL)
     
193.         {
     
194.                 if((strncmp(ent->d_name,".",1))==0 || (strncmp(ent->d_name,"..",2))==0)
     
195.                         continue;
     
196.                 snprintf(dir,50,"%s/%s", path, ent->d_name);
     
197.                 lstat(dir, &statbuf);
     
198.                
     
199.                 if(strncmp(name,ent->d_name,sizeof(name))==0)
     
200.                 {
     
201.                         if(S_ISDIR(statbuf.st_mode))
     
202.                         {
     
203.                                 printf("2:'%s' is dir and exist!-----path:'%s'\n",name,dir);
     
204.                                 List(dir);
     
205.                         }
     
206.                         if(S_ISREG(statbuf.st_mode))
     
207.                         {       
     
208.                                 printf("'%s' id file and exist!-----full path:'%s'\n",name,dir);       
     
209.                         }
     
210.                         return true;
     
211.                 }
     
212.                 else
     
213.                 {                       
     
214.                         if(S_ISDIR(statbuf.st_mode))
     
215.                         {
     
216.                                 Search_Dir(dir,name);
     
217.                         }
     
218.                 }
     
219.         }
     
220.         closedir(pdir);
     
221.         return 0;
     
222. }
     
223. 
     
224. int main()
     
225. {       
     
226.         struct stat main_buff;
     
227.         char name[50];
     
228.         char path[50];
     
229.         char path_name[50];
     
230.         char output[50];
     
231.         printf("Please input the path:\n");
     
232.         scanf("%s", path);
     
233. 
     
234.         printf("Please input the file name:\n");
     
235.         scanf("%s", name);
     
236. 
     
237.         printf("Please input the output path and file name:\n");
     
238.         scanf("%s", output);
     
239. 
     
240.         snprintf(path_name, 50, "%s/%s", path, name);
     
241. 
     
242.         if(File_Exist(path_name))
     
243.         {
     
244.                 printf("1:The file '%s' exist!---path:'%s'\n", name, path_name);
     
245.                 lstat(path_name, &main_buff);
     
246.                 if(S_ISDIR(main_buf.st_mode))
     
247.                 {
     
248.                         if(strcmp(".", ent->d_name) == 0 || strcmp("..", ent->d_name) == 0)
     
249.                                 continue;
     
250.                         List(dir);
     
251.                 }
     
252.                
     
253.                 if(S_ISREG(main_buff.st_mode))
     
254.                 {
     
255.                         Open_file(path_name,output);
     
256.                 }
     
257.         }
     
258.         else
     
259.                 Search_Dir(path, name);
     
260.         return 0;
     
261. }
     

复制代码

bruceteen

看到第一个函数
len = sizeof(stream1); 就挺扯的

mkshtk

回复 2# bruceteen


    不好意思，应该是strlen，忘记改了

foxwb

int jiami(char *stream1, char *store1)
{        
        int i, len;
        len = strlen(stream1);
        for(i=0; i<len; i++)
        {        
                store1 = stream1;                 <- 啥意思？干吗放在循环里？
        }
        return 0;
}

mkshtk

回复 4# foxwb


    准备用来加密的，现在还没写加密函数哦

绝恋的情

进来学习下

cobras

建议将代码直接用 code代码 框起来。楼主的代码好像掉了一些运算符。

cobras

建议改成命令行参数输入。这样适用性好一些。还有就是只有加密流程，没有解密流程。可以做成自动模式（在加密文件中增加唯一标识）。或者通过参数手动指定。
至于加密和解密算法，可以留到最后来实现。还有就是提示：加密和解密算法一般都需要密匙。无密匙的算法很容易破解。

mkshtk

回复 8# cobras


    谢谢大牛指点啊，我目前还只是想简单的完成这个流程，解密的话我打算加入判断让用户选择加密或者解密。至于算法还没研究过呢，等整个框架都完成了，在慢慢的局部改进。目前在文件夹加密这块卡住了，而且感觉越写耦合度越高。。等彻底完成了再贴出来，希望大牛们能给我这个菜鸟指点下，谢谢啊

mkshtk

回复 7# cobras


    额。第一次发代码，还不知道怎么怎么贴代码...