注册
登录
| 关键词： nbsp LEN pDes write include line 文件 目录 源文件 argv |
linux下简单cp命令的实现 实现功能： $./cp ~/filename ~/OtherName //文件到文件的拷贝 $./cp ~/directory/filename . //文件到当前目录的拷贝 $./cp ~/directory/filename ~/directory/ //文件到目录的拷贝 不白费口舌，直接上代码才是王道！ 001 #include <stdio.h> 002 #include <stdlib.h> 003 #include <sys/stat.h> 004 #include <sys/types.h> 005 #include <fcntl.h> 006 #include <errno.h> 007 #include <unistd.h> 008 #include <string.h> 009 www.2cto.com 010 #define BUF_SIZE 1024 011 #define PATH_LEN 128 012 013 void my_err(char *err_string, int line ) 014 { 015 fprintf(stderr,"line:%d ",line); 016 perror(err_string); 017 exit(1); 018 } 019 020 void copy_data(const int frd,const int fwd) 021 { 022 int read_len = 0, write_len = 0; 023 unsigned char buf[BUF_SIZE], *p_buf; 024 025 while ( (read_len = read(frd,buf,BUF_SIZE)) ) { 026 027 if (-1 == read_len) { 028 my_err("Read error", __LINE__); 029 } 030 else if (read_len > 0) { //把读取部分写入目标文件 031 p_buf = buf; www.2cto.com 032 while ( (write_len = write(fwd,p_buf,read_len)) ) { 033 if(write_len == read_len) { 034 break; 035 } 036 else if (write_len > 0) { //只写入部分 037 p_buf += write_len; 038 read_len -= write_len; 039 } 040 else if(-1 == write_len) { 041 my_err("Write error", __LINE__); 042 } 043 } 044 if (-1 == write_len) break; 045 } 046 } 047 } 048 049 int main(int argc, char **argv) 050 { 051 052 int frd, fwd; //读写文件描述符 053 int len = 0; 054 char *pSrc, *pDes; //分别指向源文件路径和目标文件路径 055 struct stat src_st,des_st; 056 057 if (argc < 3) { 058 printf("用法 ./MyCp <源文件路径> <目标文件路径>\n"); 059 my_err("arguments error ", __LINE__); 060 } 061 062 frd = open(argv[1],O_RDONLY); 063 if (frd == -1) { 064 my_err("Can not opne file", __LINE__); 065 } 066 067 if (fstat(frd,&src_st) == -1) { 068 my_err("stat error",__LINE__); 069 } www.2cto.com 070 /*检查源文件路径是否是目录*/ 071 if (S_ISDIR(src_st.st_mode)) { 072 my_err("略过目录",__LINE__); 073 } 074 075 pDes = argv[2]; 076 stat(argv[2],&des_st); 077 if (S_ISDIR(des_st.st_mode)) { //目标路径是目录，则使用源文件的文件名 078 079 len = strlen(argv[1]); 080 pSrc = argv[1] + (len-1); //指向最后一个字符 081 /*先找出源文件的文件名*/ 082 while (pSrc >= argv[1] && *pSrc != '/') { 083 pSrc--; 084 } 085 pSrc++;//指向源文件名 086 087 len = strlen(argv[2]); 088 // . 表示复制到当前工作目录 089 if (1 == len && '.' == *(argv[2])) { 090 len = 0; //没有申请空间，后面就不用释放 091 pDes = pSrc; 092 } 093 else { //复制到某目录下，使用源文件名 094 pDes = (char *)malloc(sizeof(char)*PATH_LEN); 095 www.2cto.com if (NULL == pDes) { 096 my_err("malloc error ", __LINE__); 097 } 098 099 strcpy(pDes,argv[2]); 100 101 if ( *(pDes+(len-1)) != '/' ) { //目录缺少最后的'/'，则补上’/‘ 102 strcat(pDes,"/"); 103 } 104 strcat(pDes+len,pSrc); 105 } 106 } 107 108 /* 打开目标文件， 使权限与源文件相同*/ 109 fwd = open(pDes,O_WRONLY | O_CREAT | O_TRUNC,src_st.st_mode); 110 if (fwd == -1) { 111 my_err("Can not creat file", __LINE__); 112 } 113 copy_data(frd,fwd); 114 //puts("end of copy"); 115 if (len > 0 && pDes != NULL) 116 www.2cto.com free(pDes); 117 118 close(frd); 119 close(fwd); 120 121 return 0; 122 } 作者 bo博 |
|
声明：文章版权归原作者所有 部分文章转自互联网 如有侵权请联系
[邮箱地址] 删除
|
