X-Git-Url: https://git.jankratochvil.net/?p=nethome.git;a=blobdiff_plain;f=src%2Fstreamfer-client.C;fp=src%2Fstreamfer-client.C;h=ce4e92384ac79f86c79b8d41e9bbb7b785b8e09a;hp=e998ffd016d4a454306b427733a0e5e70acfda3b;hb=1ea922510816fff90f9f239f99b67f925c3398c2;hpb=6a8b49a9c2bf3dafa999d8d9e6a3726b581eb8bb

diff --git a/src/streamfer-client.C b/src/streamfer-client.C
index e998ffd..ce4e923 100644
--- a/src/streamfer-client.C
+++ b/src/streamfer-client.C
@@ -18,11 +18,25 @@ static string get_string(FILE *f,const char *fn,const char *what) {
   return buf;
 }
 
-// https://stackoverflow.com/a/8615450/2995591
-#include <glob.h> // glob(), globfree()
-#include <cstring> // memset()
-#include <vector>
-#include <string>
+static void last_stored_write(const char *last_stored_fn,const string &filename,uint64_t offset) {
+  int last_stored_fd(open(last_stored_fn,O_WRONLY|O_CREAT|O_TRUNC,0644));
+  if (last_stored_fd==-1)
+    fatal("Error storing filename to %s: %m",last_stored_fn);
+  write_safe(last_stored_fd,filename.c_str(),filename.length());
+  write_safe(last_stored_fd,'\n');
+  string offsetstr(stringf("%zu",(size_t)offset));
+  write_safe(last_stored_fd,offsetstr.c_str(),offsetstr.length());
+  write_safe(last_stored_fd,'\n');
+  if (close(last_stored_fd))
+    fatal("Error closing %s: %m",last_stored_fn);
+}
+
+static string my_basename(const string &path) {
+  size_t slash(path.find_last_of('/'));
+  if (slash!=string::npos)
+    return path.substr(slash+1);
+  return path;
+}
 
 int main(int argc,char **argv) {
   if (argc!=1+3)
@@ -54,8 +68,20 @@ int main(int argc,char **argv) {
   }
   write_safe(server_fd,last_found);
   write_safe(server_fd,offset);
+  { struct stat statbuf;
+    string last_found_basename(my_basename(last_found));
+    if (stat(last_found_basename.c_str(),&statbuf)) {
+      static const struct timespec mtim_zero{};
+      write_safe(server_fd,mtim_zero);
+    } else
+      write_safe(server_fd,statbuf.st_mtim);
+  }
   string last_got;
-  read_safe(server_fd,last_got);
+  try {
+    read_safe(server_fd,last_got);
+  } catch (SafeIOError) {
+    fatal("SafeIOError reading from the server %s",argv[1]);
+  }
   if (last_got.empty()) {
     warning("No more files to transfer");
     exit(EXIT_SUCCESS);
@@ -63,12 +89,7 @@ int main(int argc,char **argv) {
   struct timespec mtim;
   read_safe(server_fd,mtim);
   int file_fd;
-  string file_name;
-  size_t slash(last_got.find_last_of('/'));
-  if (slash!=string::npos)
-    file_name=last_got.substr(slash+1);
-  else
-    file_name=last_got;
+  string file_name(my_basename(last_got));;
   if (last_found==last_got) {
     file_fd=open(file_name.c_str(),O_WRONLY);
     if (file_fd==-1)
@@ -82,9 +103,15 @@ int main(int argc,char **argv) {
     if (file_fd==-1)
       fatal("Error creating %s: %m",file_name.c_str());
   }
+  last_stored_write(last_stored_fn,last_got,offset);
   uint64_t transferred(transfer(server_fd,"server fd",file_fd,file_name.c_str()));
-  if (!transferred)
-    return EXIT_FAILURE;
+  if (!transferred) {
+    struct stat statbuf;
+    int err(fstat(file_fd,&statbuf));
+    assert(!err);
+    if (memcmp(&mtim,&statbuf.st_mtim,sizeof(mtim))==0)
+      return EXIT_FAILURE;
+  }
   offset+=transferred;
   struct timespec mtim2[2];
   mtim2[0]=mtim; // atime
@@ -93,15 +120,6 @@ int main(int argc,char **argv) {
     fatal("Error setting timestamp of %s: %m",file_name.c_str());
   if (close(file_fd))
     fatal("Error closing %s: %m",file_name.c_str());
-  int last_stored_fd(open(last_stored_fn,O_WRONLY|O_CREAT|O_TRUNC,0644));
-  if (last_stored_fd==-1)
-    fatal("Error storing filename to %s: %m",last_stored_fn);
-  write_safe(last_stored_fd,last_got.c_str(),last_got.length());
-  write_safe(last_stored_fd,'\n');
-  string offsetstr(stringf("%zu",(size_t)offset));
-  write_safe(last_stored_fd,offsetstr.c_str(),offsetstr.length());
-  write_safe(last_stored_fd,'\n');
-  if (close(last_stored_fd))
-    fatal("Error closing %s: %m",last_stored_fn);
+  last_stored_write(last_stored_fn,last_got,offset);
   return EXIT_SUCCESS;
 }