linux下简单cp命令的实现

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.atcpu.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.atcpu.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.atcpu.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.atcpu.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.atcpu.com  
        free(pDes);
117
    
118
    close(frd);
119
    close(fwd);
120

121
    return 0;
122
}