source/copy.cpp

   1 #include <errno.h>
   2 #include <fstream>
   3 #include <iostream>
   4 #include <msp/path/utils.h>
   5 #include "builder.h"
   6 #include "copy.h"
   7 #include "package.h"
   8
   9 using namespace std;
  10 using namespace Msp;
  11
  12 Copy::Copy(Builder &b, const Package &pkg, const Path::Path &s, const Path::Path &d):
  13         Action(b),
  14         src(s),
  15         dest(d),
  16         worker(0)
  17 {
  18         announce(pkg.get_name(), "COPY", dest[-1]);
  19         if(builder.get_verbose()>=2)
  20                 cout<<s<<" -> "<<d<<'\n';
  21
  22         if(!builder.get_dry_run())
  23                 worker=new Worker(*this);
  24 }
  25
  26 int Copy::check()
  27 {
  28         if(!worker)
  29         {
  30                 signal_done.emit();
  31                 return 0;
  32         }
  33
  34         if(worker->get_done())
  35         {
  36                 signal_done.emit();
  37                 worker->join();
  38                 return worker->get_error()?1:0;
  39         }
  40
  41         return -1;
  42 }
  43
  44 Copy::~Copy()
  45 {
  46         delete worker;
  47 }
  48
  49 void Copy::Worker::main()
  50 {
  51         Path::mkpath(copy.dest.subpath(0, copy.dest.size()-1), 0755);
  52
  53         if(unlink(copy.dest.str().c_str())<0 && errno!=ENOENT)
  54         {
  55                 int err=errno;
  56                 cerr<<"Can't unlink "<<copy.dest<<": "<<strerror(err)<<'\n';
  57                 done=error=true;
  58                 return;
  59         }
  60
  61         ifstream in(copy.src.str().c_str());
  62         if(!in)
  63         {
  64                 cerr<<"Can't open "<<copy.src<<" for reading\n";
  65                 done=error=true;
  66                 return;
  67         }
  68
  69         ofstream out(copy.dest.str().c_str());
  70         if(!out)
  71         {
  72                 cerr<<"Can't open "<<copy.dest<<" for writing\n";
  73                 done=error=true;
  74                 return;
  75         }
  76
  77         char buf[16384];
  78         while(!in.eof())
  79         {
  80                 in.read(buf, sizeof(buf));
  81                 out.write(buf, in.gcount());
  82         }
  83
  84         struct stat st;
  85         Path::stat(copy.src, st);
  86         chmod(copy.dest.str().c_str(), st.st_mode&0777);
  87
  88         done=true;
  89 }