實現文件複製命令

原創 w_bo713 2019-02-22 23:29

 

一、實驗目的
學習使用linux和windows下有關目錄讀寫和文件讀寫的API函數。
二、實驗內容
完成一個目錄複製命令mycp，包括目錄下的文件和子目錄。
說明：
Linux: creat，read，write等系統調用
Windows: CreateFile(), ReadFile(), WriteFile(), CloseHandle（）函數。
三、實驗環境
Ubuntu10.10和Windows7
四、程序設計與實現
主要算法是打開源目錄，遍歷源目錄中的每一項，如果是文件，就在目標目錄的相應位置創建文件，然後複製文件內容，如果是文件夾，在目標位置建立文件夾，然後遞歸進入該文件夾繼續操作。這裏要注意的是在建立文件夾和文件的時候，要同時把源文件的權限複製到目標文件上去。
主要使用了一下API函數：
windows下：
HANDLE CreateFile(
　　LPCTSTR lpFileName, //指向文件名的指針
　　DWORD dwDesiredAccess, //訪問模式（寫/讀）
　　DWORD dwShareMode, //共享模式
　　LPSECURITY_ATTRIBUTES lpSecurityAttributes, //指向安全屬性的指針
　　DWORD dwCreationDisposition, //如何創建
　　DWORD dwFlagsAndAttributes, //文件屬性
　　HANDLE hTemplateFile //用於複製文件句柄
　　);
BOOL WriteFile(
　　HANDLE hFile, // 文件句柄
　　LPCVOID lpBuffer, // 數據緩存區指針
　　DWORD nNumberOfBytesToWrite, // 你要寫的字節數
　　LPDWORD lpNumberOfBytesWritten, // 用於保存實際寫入字節數的存儲區域的指針
　　LPOVERLAPPED lpOverlapped // OVERLAPPED結構體指針
　　);
BOOL ReadFile(
　　HANDLE hFile, //文件的句柄
　　LPVOID lpBuffer, //用於保存讀入數據的一個緩衝區
　　DWORD nNumberOfBytesToRead, //要讀入的字符數
　　LPDWORD lpNumberOfBytesRead, //指向實際讀取字節數的指針
　　LPOVERLAPPED lpOverlapped //如文件打開時指定了FILE_FLAG_OVERLAPPED，那麼必須，用這個參數引用一個特殊的結構。該結構定義了一次異步讀取操作。否則，應將這個參數設爲NULL
　　);
BOOL CloseHandle(
　　HANDLE hObject //代表一個已打開對象handle
　　);
linux下：
int creat(const char * pathname, mode_tmode);//_tmode爲文件的狀態碼
int read(int handle, void *buf, int nbyte);//nbyte爲最大讀取字節數，返回值爲實際讀取字節數
int write(int handel, void *buf, int nbyte); //nbyte爲最大寫入字節數，返回值爲實際寫入字節數

windows版本：

 

  1. #include <windows.h>  
  2. #include <windowsx.h>  
  3. #include <tchar.h>  
  4. #include <stdlib.h>  
  5. #include <stdio.h>  
  6. #include <string.h>  
  7.  
  8. #define chDIMOF(Array) (sizeof(Array)/sizeof(Array[0]))  
  9.  
  10. static BOOL IsChildDir(WIN32_FIND_DATA * lpFindData)  
  11. {  
  12.     return ((lpFindData->dwFileAttributes&  
  13.         FILE_ATTRIBUTE_DIRECTORY)!=0)&&  
  14.         (lstrcmp(lpFindData->cFileName,__TEXT("."))!=0)&&  
  15.         (lstrcmp(lpFindData->cFileName,__TEXT(".."))!=0);  
  16. }  
  17.  
  18. static BOOL FindNextChildDir(HANDLE hFindFile,WIN32_FIND_DATA *lpFindData)  
  19. {  
  20.     BOOL fFound=FALSE;  
  21.       
  22.     do 
  23.     {  
  24.         fFound=FindNextFile(hFindFile,lpFindData);  
  25.     }while (fFound&&!IsChildDir(lpFindData));  
  26.  
  27.     return fFound;  
  28. }  
  29.  
  30. static HANDLE FindFirstChildDir(LPTSTR szPath,WIN32_FIND_DATA * lpFindData)  
  31. {  
  32.     BOOL fFound;  
  33.     HANDLE hFindFile=FindFirstFile(szPath,lpFindData);  
  34.  
  35.     if (hFindFile!=INVALID_HANDLE_VALUE)  
  36.     {  
  37.         fFound=IsChildDir(lpFindData);  
  38.  
  39.         if (!fFound)  
  40.         {  
  41.             fFound=FindNextChildDir(hFindFile,lpFindData);  
  42.         }  
  43.  
  44.         if (!fFound)  
  45.         {  
  46.             FindClose(hFindFile);  
  47.             hFindFile=INVALID_HANDLE_VALUE;  
  48.         }  
  49.     }  
  50.     return hFindFile;  
  51. }  
  52.  
  53. typedef struct 
  54. {  
  55.     int nDepth;  
  56.     BOOL fRecurse;  
  57.     TCHAR szBuf[1000];  
  58.     int nIndent;  
  59.     BOOL fOk;  
  60.     BOOL fIsDir;  
  61.     WIN32_FIND_DATA FindData;  
  62. }DIRWALKDATA,*LPDIRWALKDATA;  
  63.  
  64. static void DirWalkRecurse(LPDIRWALKDATA pDW,LPTSTR pszRootPath,LPTSTR pszdestPath)  
  65. {  
  66.     HANDLE hFind;  
  67.     LPTSTR temp;  
  68.     LPTSTR ttemp;  
  69.     temp=(LPTSTR)malloc(100*sizeof(LPTSTR));  
  70.     ttemp=(LPTSTR)malloc(100*sizeof(LPTSTR));  
  71.     pDW->nDepth++;  
  72.     pDW->nIndent=3*pDW->nDepth;  
  73.     _stprintf(pDW->szBuf,__TEXT("%*s"),pDW->nIndent,__TEXT(""));  
  74.     GetCurrentDirectory(chDIMOF(pDW->szBuf)-pDW->nIndent,&pDW->szBuf[pDW->nIndent]);  
  75.  
  76.     //printf("%s\n",pDW->szBuf);  
  77.  
  78.     hFind=FindFirstFile(__TEXT("*.*"),&pDW->FindData);  
  79.     pDW->fOk=(hFind!=INVALID_HANDLE_VALUE);  
  80.       
  81.     while(pDW->fOk)  
  82.     {  
  83.         pDW->fIsDir=pDW->FindData.dwFileAttributes&  
  84.             FILE_ATTRIBUTE_DIRECTORY;  
  85.         if (!pDW->fIsDir||  
  86.             (!pDW->fRecurse&&IsChildDir(&pDW->FindData)))  
  87.         {  
  88.             _stprintf(pDW->szBuf,  
  89.                 pDW->fIsDir?__TEXT("%*s[%s]"):__TEXT("%*s%s"),  
  90.                 pDW->nIndent,__TEXT(""),  
  91.                 pDW->FindData.cFileName);  
  92.  
  93.             //printf("%s\n",pDW->szBuf);  
  94.             lstrcpy(temp,pszRootPath);  
  95.             lstrcat(temp,"\\");  
  96.             lstrcat(temp,pDW->FindData.cFileName);  
  97.             lstrcpy(ttemp,pszdestPath);  
  98.             lstrcat(ttemp,"\\");  
  99.             lstrcat(ttemp,pDW->FindData.cFileName);  
  100.             HANDLE hfile=CreateFile(  
  101.                 ttemp,  
  102.                 GENERIC_READ|GENERIC_WRITE,  
  103.                 0,  
  104.                 NULL,  
  105.                 OPEN_ALWAYS,  
  106.                 pDW->FindData.dwFileAttributes,  
  107.                 NULL);  
  108.             HANDLE dfile=CreateFile(  
  109.                 temp,  
  110.                 GENERIC_READ|GENERIC_WRITE,  
  111.                 0,  
  112.                 NULL,  
  113.                 OPEN_EXISTING,  
  114.                 pDW->FindData.dwFileAttributes,  
  115.                 NULL);  
  116.             DWORD filesize=GetFileSize(dfile,NULL);  
  117.             char* buffer=new char[filesize+1];  
  118.             DWORD readsize;  
  119.             ReadFile(dfile,buffer,filesize,&readsize,NULL);  
  120.             WriteFile(hfile,buffer,filesize,&readsize,NULL);  
  121.             buffer[filesize]=0;  
  122.             CloseHandle(hfile);  
  123.             CloseHandle(dfile);  
  124.         }  
  125.         else if(pDW->fIsDir&&IsChildDir(&pDW->FindData))  
  126.         {  
  127.             SECURITY_ATTRIBUTES attribute;  
  128.             attribute.nLength = sizeof(attribute);  
  129.             attribute.lpSecurityDescriptor = NULL;  
  130.             attribute.bInheritHandle = FALSE;  
  131.             SetCurrentDirectory(pszdestPath);  
  132.             CreateDirectory(  
  133.                 pDW->FindData.cFileName,  
  134.                 &attribute  
  135.                 );   
  136.             SetCurrentDirectory(pszRootPath);  
  137.         }  
  138.  
  139.         pDW->fOk=FindNextFile(hFind,&pDW->FindData);  
  140.     }  
  141.  
  142.     if (hFind != INVALID_HANDLE_VALUE)  
  143.     {  
  144.         FindClose(hFind);  
  145.     }  
  146.  
  147.     if (pDW->fRecurse)  
  148.     {  
  149.         hFind=FindFirstChildDir(__TEXT("*.*"),&pDW->FindData);  
  150.         pDW->fOk=(hFind!=INVALID_HANDLE_VALUE);  
  151.         while(pDW->fOk)  
  152.         {  
  153.             if (SetCurrentDirectory(pDW->FindData.cFileName))  
  154.             {  
  155.                 lstrcpy(temp,pszRootPath);  
  156.                 lstrcat(temp,"\\");  
  157.                 lstrcat(temp,pDW->FindData.cFileName);  
  158.                 lstrcpy(ttemp,pszdestPath);  
  159.                 lstrcat(ttemp,"\\");  
  160.                 lstrcat(ttemp,pDW->FindData.cFileName);  
  161.                 DirWalkRecurse(pDW,temp,ttemp);  
  162.                 SetCurrentDirectory(__TEXT(".."));  
  163.             }  
  164.             pDW->fOk=FindNextChildDir(hFind,&pDW->FindData);  
  165.         }  
  166.         if (hFind!=INVALID_HANDLE_VALUE)  
  167.         {  
  168.             FindClose(hFind);  
  169.         }  
  170.     }  
  171.  
  172.     pDW->nDepth--;  
  173. }  
  174.  
  175. void DirWalk(LPTSTR pszRootPath,LPTSTR pszdestPath,BOOL fResourse)  
  176. {  
  177.     TCHAR szCurrDir[_MAX_DIR];  
  178.     DIRWALKDATA DW;  
  179.       
  180.     GetCurrentDirectory(chDIMOF(szCurrDir),szCurrDir);  
  181.       
  182.     SetCurrentDirectory(pszRootPath);  
  183.  
  184.     DW.nDepth=-1;  
  185.     DW.fRecurse=fResourse;  
  186.       
  187.     DirWalkRecurse(&DW,pszRootPath,pszdestPath);  
  188.     SetCurrentDirectory(szCurrDir);  
  189. }  
  190.  
  191. int main(int argc,char *argv[])  
  192. {  
  193.     char tmp[100],ttmp[100];  
  194.     int i;  
  195.     if (argc!=3)  
  196.     {  
  197.         printf("Argument Error\n");  
  198.         exit(0);  
  199.     }  
  200.       
  201.  
  202.     strcpy(tmp,argv[1]);  
  203.     if (strlen(tmp)!=1&&tmp[strlen(tmp)-1] == '\\')  
  204.     {  
  205.          tmp[strlen(tmp)-1]=0;  
  206.     }  
  207.     for ( i = strlen(tmp); i >= 0; i--)  
  208.     {  
  209.          if ( tmp[i] == '\\')  
  210.          {  
  211.              break;  
  212.          }  
  213.     }  
  214.     //printf("%s\n",&tmp[i+1]);  
  215.     strcpy(ttmp,argv[2]);  
  216.     if (strcmp(&tmp[i+1],"")&&strcmp(&tmp[i+1],"..")&&strcmp(&tmp[i+1],"."))  
  217.     {  
  218.  
  219.         if (ttmp[strlen(ttmp)-1] != '\\')  
  220.         {  
  221.             strcat(ttmp,"/");  
  222.         }  
  223.         strcat(ttmp,&tmp[i+1]);  
  224.         //printf("cp %s %s\n",tmp,ttmp);  
  225.         SECURITY_ATTRIBUTES attribute;  
  226.         attribute.nLength = sizeof(attribute);  
  227.         attribute.lpSecurityDescriptor = NULL;  
  228.         attribute.bInheritHandle = FALSE;  
  229.         SetCurrentDirectory(argv[2]);  
  230.         CreateDirectory(  
  231.             &tmp[i+1],  
  232.             &attribute  
  233.             );   
  234.     }  
  235.     DirWalk((char *)argv[1],(char *)ttmp,TRUE);  
  236.  
  237.     return 0;  
  238. }

linux版本：

 

  1. #include <stdio.h>  
  2. #include <stdlib.h>  
  3. #include <string.h>  
  4. #include <dirent.h>  
  5. #include <sys/stat.h>  
  6. #include <dirent.h>  
  7. #include <fcntl.h>  
  8. #define N 4096  
  9.  
  10. int cp(char dir[],char tdir[],int k)  
  11. {  
  12.     int i,j,n;  
  13.     int fd,tfd;  
  14.     DIR *dp;  
  15.     char buf[N];  
  16.     char tmp[N],ttmp[N],ch[N];  
  17.     struct dirent *dirp[N],*dirptmp;  
  18.     struct stat st[N];  
  19.  
  20.     if ( (dp = opendir(dir)) == NULL)  
  21.     {  
  22.         printf("cp: cannot access %s: No such file or directory\n",dir);  
  23.         return 1;  
  24.     }  
  25.  
  26.     i=0;  
  27.     n=0;  
  28.     while ( (dirp[i] = readdir(dp)) != NULL)  
  29.     {  
  30.         if (dirp[i]->d_name[0]=='.')  
  31.         {  
  32.             continue;  
  33.         }  
  34.  
  35.         strcpy(tmp,dir);  
  36.         if (tmp[strlen(tmp)-1] != '/')  
  37.         {  
  38.             strcat(tmp,"/");  
  39.         }  
  40.         strcat(tmp,dirp[i]->d_name);  
  41.         //printf("%s\n",tmp);  
  42.         if (lstat(tmp,&st[i]) < 0)  
  43.         {  
  44.              printf("Cannot open the directory\n");  
  45.              break;  
  46.         }  
  47.  
  48.         i++;  
  49.         n++;  
  50.     }  
  51.  
  52.     for ( i = 0; i < n; i++)  
  53.     {  
  54.         //printf("%d\n",i);  
  55.         memset(tmp,0,sizeof(tmp));  
  56.         memset(ttmp,0,sizeof(ttmp));  
  57.         //printf("%s\n",dirp[i]->d_name);  
  58.         if (S_ISDIR(st[i].st_mode))  
  59.         {  
  60.             strcpy(tmp,dir);  
  61.             if (tmp[strlen(tmp)-1] != '/')  
  62.             {  
  63.                 strcat(tmp,"/");  
  64.             }  
  65.             strcat(tmp,dirp[i]->d_name);  
  66.             strcpy(ttmp,tdir);  
  67.             if (ttmp[strlen(ttmp)-1] != '/')  
  68.             {  
  69.                 strcat(ttmp,"/");  
  70.             }  
  71.             strcat(ttmp,dirp[i]->d_name);  
  72.             //printf("cp %s %s\n",tmp,ttmp);  
  73.             if (mkdir(ttmp,st[i].st_mode) != 0)  
  74.             {  
  75.                 //printf("The directory %s is exist.\n",ttmp);  
  76.                 //continue;  
  77.             }  
  78.             cp(tmp,ttmp,k+1);  
  79.             //printf("%s\n",tmp);  
  80.         }  
  81.         else if(S_ISLNK(st[i].st_mode))  
  82.         {  
  83.             continue;  
  84.         }  
  85.         else 
  86.         {  
  87.             strcpy(tmp,dir);  
  88.             if (tmp[strlen(tmp)-1] != '/')  
  89.             {  
  90.                 strcat(tmp,"/");  
  91.             }  
  92.             strcat(tmp,dirp[i]->d_name);  
  93.             fd=open(tmp,O_RDONLY);  
  94.             if (fd == -1)  
  95.             {  
  96.                 printf("open file error\n");  
  97.                 continue;  
  98.             }  
  99.             strcpy(ttmp,tdir);  
  100.             strcat(ttmp,"/");  
  101.             strcat(ttmp,dirp[i]->d_name);  
  102.  
  103.             tfd=creat(ttmp,0644);  
  104.             if (tfd == -1)  
  105.             {  
  106.                 printf("open file error\n");  
  107.                 continue;  
  108.             }  
  109.             //printf("file %s %s\n",tmp,ttmp);  
  110.             while((j=read(fd,buf,512))>0)  
  111.             {  
  112.                write(tfd,buf,j);  
  113.             }  
  114.  
  115.             close(fd);  
  116.             close(tfd);  
  117.         }  
  118.     }  
  119.     closedir(dp);  
  120.     return 0;  
  121. }  
  122. int main(int argc, char *argv[])  
  123. {  
  124.     int i;  
  125.     char tmp[N],ttmp[N];  
  126.     struct stat st[N];  
  127.  
  128.     if (argc < 2)  
  129.     {  
  130.         printf("Argument error\n");  
  131.         return 0;  
  132.     }  
  133.     else if (argc == 2)  
  134.     {  
  135.         printf("Argument error\n");  
  136.         return 0;  
  137.     }  
  138.     else if (argc == 3)  
  139.     {  
  140.         strcpy(tmp,argv[1]);  
  141.         if (strlen(tmp)!=1&&tmp[strlen(tmp)-1] == '/')  
  142.         {  
  143.             tmp[strlen(tmp)-1]=0;  
  144.         }  
  145.         for ( i = strlen(tmp); i >= 0; i--)  
  146.         {  
  147.             if ( tmp[i] == '/')  
  148.             {  
  149.                 break;  
  150.             }  
  151.         }  
  152.         //printf("%s\n",&tmp[i+1]);  
  153.         strcpy(ttmp,argv[2]);  
  154.         if (strcmp(&tmp[i+1],"")&&strcmp(&tmp[i+1],"..")&&strcmp(&tmp[i+1],"."))  
  155.         {  
  156.  
  157.             if (ttmp[strlen(ttmp)-1] != '/')  
  158.             {  
  159.                 strcat(ttmp,"/");  
  160.             }  
  161.             strcat(ttmp,&tmp[i+1]);  
  162.             //printf("cp %s %s\n",tmp,ttmp);  
  163.             if (lstat(tmp,&st[i]) < 0)  
  164.             {  
  165.                  printf("Cannot open the directory\n");  
  166.                  exit(0);  
  167.             }  
  168.             mkdir(ttmp,st[i].st_mode);  
  169.         }  
  170.         //printf("cp %s %s\n",argv[1],ttmp);  
  171.         cp(argv[1],ttmp,0);//參數應爲絕對路徑  
  172.     }  
  173.     else 
  174.     {  
  175.         printf("Argument error\n");  
  176.         return 0;  
  177.     }  
  178.     return 0;  
  179. }

 

發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章

Linux基本操作命令

wbzjacky 2019-02-24 13:12:38

444句英語

wbzjacky 2019-02-24 13:12:37

真實的模擬***綜合實驗

wbzjacky 2019-02-24 13:12:37

三層交換機的HSRP、vlan、端口聚合

wbzjacky 2019-02-24 13:12:37

HSRP和二層交換機的端口聚合、vlan

wbzjacky 2019-02-24 13:12:37

路遇一個“乞討者“

楊木藝 2019-02-24 14:19:35

如果同事暗中傷害你，應該怎麼辦？

這個饅頭有餡 2019-02-24 13:59:08

職場中，抱怨越多的員工，越被領導瞧不起！

這個饅頭有餡 2019-02-24 13:59:08

老程序員被裁，應屆生卻能月薪 1.3 萬？這你能忍？

前端高達 2019-02-24 13:48:04

遇到到處蹭吃卻從不請客吃飯的主怎麼辦？

樑軍年 2019-02-24 13:26:35

高標準機房綜合配線安裝

wbzjacky 2019-02-24 13:12:38

IPsec ***實驗

wbzjacky 2019-02-24 13:12:37

網絡系統集成

wbzjacky 2019-02-24 13:12:37

CISCO路由AAA的Easy ***

wbzjacky 2019-02-24 13:12:37

CISCO訪問控制列表 企業網絡管理的必殺技

wbzjacky 2019-02-24 13:12:37