多線程實現文件拷貝(Linux下C++)

1. //copyfile.cc
2. #include <iostream>
3. #include <cstring>
4. #include <stdio.h>
5. #include <sys/types.h>
6. #include <sys/stat.h>
7. #include <fcntl.h>
8. #include <unistd.h>
9. #include <pthread.h>
10. using namespace std;
12. /************************************
13. *使用指定線程實現從文件的拷貝
14. *創建時間：2011.07.28
15. *修改時間：2011.07.29
16. *作者：hahaya
17. ***********************************/
19. //最大使用的線程數
20. const int MAX_THREADS = 5;
22. typedef struct TAG_INFO
23. {
24. char *fromfile; //源地址
25. char *tofile; //目的地址
26. int num; //啟動的第i-1個進程
27. }info;
29. //st_size的類型為__off_t
30. int get_size(const char *filename)
31. {
32. struct stat st;
33. memset(&st, 0, sizeof(st));
34. stat(filename, &st);
35. return st.st_size;
36. }
38. void* threadDL(void *param)
39. {
40. info info1 = *((info*)param);
42. FILE *fin = fopen(info1.fromfile, "r+");
43. FILE *fout = fopen(info1.tofile, "w+");
45. int size = get_size(info1.fromfile);
46. //將文件指針分別設置在每個線程要讀和寫的位置
47. fseek(fin, size*(info1.num)/MAX_THREADS, SEEK_SET);
48. fseek(fout, size*(info1.num)/MAX_THREADS, SEEK_SET);
50. char buff[1024] = {'\0'};
51. int len = 0;
52. int total = 0;
53. while((len = fread(buff, 1, sizeof(buff), fin)) > 0)
54. {
55. fwrite(buff, 1, len, fout);
56. total += len;
57. //如果讀入的數據大於文件總大小除線程總數則停止讀入，因為每個線程要讀或寫的數據就等於文件總大小除線程總數
58. //可能會多寫入一些數據，下一次寫入時會覆蓋多寫入的數據，所以不用擔心
59. if(total > size/MAX_THREADS)
60. {
61. break;
62. }
63. }
65. fclose(fin);
66. fclose(fout);
68. }
70. int main(int argc, char *argv[])
71. {
72. //先創建一個與文件1同樣大小的文件
73. creat(argv[2], 0777);
74. truncate(argv[2], get_size(argv[1]));
76. pthread_t pid[MAX_THREADS];
77. info info1;
78. //啟動指定線程數的線程
79. for(int i = 0; i < MAX_THREADS; i++)
80. {
81. memset(&info1, 0, sizeof(info1));
82. info1.fromfile = argv[1];
83. info1.tofile = argv[2];
84. info1.num = i;
85. pthread_create(&pid[i], NULL, threadDL, (void*)&info1);
86. }
87. //等待線程結束
88. for(int j = 0; j < MAX_THREADS; j++)
89. {
90. //pthread_join不能用在創建進程的for循環中，否則創建第一個進程後會等待第一個進程結束後創建第二個進程
91. pthread_join(pid[j], NULL);
92. }
94. cout << "file copy success......" << endl;
95. return 0;
97. }

程序運行截圖：

復制後的文件是完整的，可以解壓，如下圖: