知识屋:更实用的电脑技术知识网站
所在位置:首页 > 操作系统 > linux

linux下简单cp命令的实现

发布时间:2014-09-05 16:11:43作者:知识屋

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.zhishiwu.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.zhishiwu.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.zhishiwu.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.zhishiwu.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.zhishiwu.com  
        free(pDes);
117
     
118
    close(frd);
119
    close(fwd);
120
 
121
    return 0;
122
}

 
 
作者 bo博
(免责声明:文章内容如涉及作品内容、版权和其它问题,请及时与我们联系,我们将在第一时间删除内容,文章内容仅供参考)
收藏
  • 人气文章
  • 最新文章
  • 下载排行榜
  • 热门排行榜