cp.c 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177
  1. #include <u.h>
  2. #include <libc.h>
  3. #define DEFB (8*1024)
  4. int failed;
  5. int gflag;
  6. int uflag;
  7. int xflag;
  8. void copy(char *from, char *to, int todir);
  9. int copy1(int fdf, int fdt, char *from, char *to);
  10. void
  11. main(int argc, char *argv[])
  12. {
  13. Dir *dirb;
  14. int todir, i;
  15. ARGBEGIN {
  16. case 'g':
  17. gflag++;
  18. break;
  19. case 'u':
  20. uflag++;
  21. gflag++;
  22. break;
  23. case 'x':
  24. xflag++;
  25. break;
  26. default:
  27. goto usage;
  28. } ARGEND
  29. todir=0;
  30. if(argc < 2)
  31. goto usage;
  32. dirb = dirstat(argv[argc-1]);
  33. if(dirb!=nil && (dirb->mode&DMDIR))
  34. todir=1;
  35. if(argc>2 && !todir){
  36. fprint(2, "cp: %s not a directory\n", argv[argc-1]);
  37. exits("bad usage");
  38. }
  39. for(i=0; i<argc-1; i++)
  40. copy(argv[i], argv[argc-1], todir);
  41. if(failed)
  42. exits("errors");
  43. exits(0);
  44. usage:
  45. fprint(2, "usage:\tcp [-gux] fromfile tofile\n");
  46. fprint(2, "\tcp [-x] fromfile ... todir\n");
  47. exits("usage");
  48. }
  49. int
  50. samefile(Dir *a, char *an, char *bn)
  51. {
  52. Dir *b;
  53. int ret;
  54. ret = 0;
  55. b=dirstat(bn);
  56. if(b != nil)
  57. if(b->qid.type==a->qid.type)
  58. if(b->qid.path==a->qid.path)
  59. if(b->qid.vers==a->qid.vers)
  60. if(b->dev==a->dev)
  61. if(b->type==a->type){
  62. fprint(2, "cp: %s and %s are the same file\n", an, bn);
  63. ret = 1;
  64. }
  65. free(b);
  66. return ret;
  67. }
  68. void
  69. copy(char *from, char *to, int todir)
  70. {
  71. Dir *dirb, dirt;
  72. char name[256];
  73. int fdf, fdt, mode;
  74. if(todir){
  75. char *s, *elem;
  76. elem=s=from;
  77. while(*s++)
  78. if(s[-1]=='/')
  79. elem=s;
  80. sprint(name, "%s/%s", to, elem);
  81. to=name;
  82. }
  83. if((dirb=dirstat(from))==nil){
  84. fprint(2,"cp: can't stat %s: %r\n", from);
  85. failed = 1;
  86. return;
  87. }
  88. mode = dirb->mode;
  89. if(mode&DMDIR){
  90. fprint(2, "cp: %s is a directory\n", from);
  91. free(dirb);
  92. failed = 1;
  93. return;
  94. }
  95. if(samefile(dirb, from, to)){
  96. free(dirb);
  97. failed = 1;
  98. return;
  99. }
  100. mode &= 0777;
  101. fdf=open(from, OREAD);
  102. if(fdf<0){
  103. fprint(2, "cp: can't open %s: %r\n", from);
  104. free(dirb);
  105. failed = 1;
  106. return;
  107. }
  108. fdt=create(to, OWRITE, mode);
  109. if(fdt<0){
  110. fprint(2, "cp: can't create %s: %r\n", to);
  111. close(fdf);
  112. free(dirb);
  113. failed = 1;
  114. return;
  115. }
  116. if(copy1(fdf, fdt, from, to)==0 && (xflag || gflag || uflag)){
  117. nulldir(&dirt);
  118. if(xflag){
  119. dirt.mtime = dirb->mtime;
  120. dirt.mode = dirb->mode;
  121. }
  122. if(uflag)
  123. dirt.uid = dirb->uid;
  124. if(gflag)
  125. dirt.gid = dirb->gid;
  126. if(dirfwstat(fdt, &dirt) < 0)
  127. fprint(2, "cp: warning: can't wstat %s: %r\n", to);
  128. }
  129. free(dirb);
  130. close(fdf);
  131. close(fdt);
  132. }
  133. int
  134. copy1(int fdf, int fdt, char *from, char *to)
  135. {
  136. char *buf;
  137. long n, n1, rcount;
  138. int rv;
  139. char err[ERRMAX];
  140. buf = malloc(DEFB);
  141. /* clear any residual error */
  142. err[0] = '\0';
  143. errstr(err, ERRMAX);
  144. rv = 0;
  145. for(rcount=0;; rcount++) {
  146. n = read(fdf, buf, DEFB);
  147. if(n <= 0)
  148. break;
  149. n1 = write(fdt, buf, n);
  150. if(n1 != n) {
  151. fprint(2, "cp: error writing %s: %r\n", to);
  152. failed = 1;
  153. rv = -1;
  154. break;
  155. }
  156. }
  157. if(n < 0) {
  158. fprint(2, "cp: error reading %s: %r\n", from);
  159. failed = 1;
  160. rv = -1;
  161. }
  162. free(buf);
  163. return rv;
  164. }