diff options
author | Wonsik Kim <wonsik@google.com> | 2017-03-31 00:18:13 +0000 |
---|---|---|
committer | Wonsik Kim <wonsik@google.com> | 2017-03-31 00:18:13 +0000 |
commit | 395e29472fa012c4177d981d9ce699625b706f4f (patch) | |
tree | a7aef5560033a0c94516ecf62c22ae0de281400a /init/builtins.cpp | |
parent | 82bac0de6d95bcdf45729516f6a4f29eb2681118 (diff) |
Revert "init: use read_file and write_file to implement do_copy builtin"
This reverts commit 82bac0de6d95bcdf45729516f6a4f29eb2681118.
Change-Id: Ibfdf453bd85ba4fcd1948b78bd22e639a883653e
Diffstat (limited to 'init/builtins.cpp')
-rw-r--r-- | init/builtins.cpp | 58 |
1 files changed, 54 insertions, 4 deletions
diff --git a/init/builtins.cpp b/init/builtins.cpp index c7dffb105..43eb378d7 100644 --- a/init/builtins.cpp +++ b/init/builtins.cpp @@ -839,11 +839,61 @@ static int do_write(const std::vector<std::string>& args) { } static int do_copy(const std::vector<std::string>& args) { - std::string data; - if (read_file(args[1].c_str(), &data)) { - return write_file(args[2].c_str(), data.data()) ? 0 : 1; + char *buffer = NULL; + int rc = 0; + int fd1 = -1, fd2 = -1; + struct stat info; + int brtw, brtr; + char *p; + + if (stat(args[1].c_str(), &info) < 0) + return -1; + + if ((fd1 = open(args[1].c_str(), O_RDONLY|O_CLOEXEC)) < 0) + goto out_err; + + if ((fd2 = open(args[2].c_str(), O_WRONLY|O_CREAT|O_TRUNC|O_CLOEXEC, 0660)) < 0) + goto out_err; + + if (!(buffer = (char*) malloc(info.st_size))) + goto out_err; + + p = buffer; + brtr = info.st_size; + while(brtr) { + rc = read(fd1, p, brtr); + if (rc < 0) + goto out_err; + if (rc == 0) + break; + p += rc; + brtr -= rc; } - return 1; + + p = buffer; + brtw = info.st_size; + while(brtw) { + rc = write(fd2, p, brtw); + if (rc < 0) + goto out_err; + if (rc == 0) + break; + p += rc; + brtw -= rc; + } + + rc = 0; + goto out; +out_err: + rc = -1; +out: + if (buffer) + free(buffer); + if (fd1 >= 0) + close(fd1); + if (fd2 >= 0) + close(fd2); + return rc; } static int do_chown(const std::vector<std::string>& args) { |