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博