daemon/sync.c

   1 /* libguestfs - the guestfsd daemon
   2  * Copyright (C) 2009 Red Hat Inc.
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17  */
  18
  19 #include <config.h>
  20
  21 #ifdef HAVE_WINDOWS_H
  22 #include <windows.h>
  23 #endif
  24
  25 #include <stdio.h>
  26 #include <unistd.h>
  27
  28 #include "daemon.h"
  29 #include "actions.h"
  30
  31 #ifdef WIN32
  32 static int sync_win32 (void);
  33 #endif
  34
  35 int
  36 do_sync (void)
  37 {
  38   if (sync_disks () == -1) {
  39     reply_with_perror ("sync");
  40     return -1;
  41   }
  42
  43   return 0;
  44 }
  45
  46 /* Older versions of libguestfs used to issue separate 'umount_all'
  47  * and 'sync' commands just before closing the handle.  Since
  48  * libguestfs 1.9.7 the library issues this 'internal_autosync'
  49  * internal operation instead, allowing more control in the daemon.
  50  */
  51 int
  52 do_internal_autosync (void)
  53 {
  54   int r;
  55
  56   r = do_umount_all ();
  57
  58   sync_disks ();
  59
  60   return r;
  61 }
  62
  63 /* This is a replacement for sync(2) which is called from
  64  * this file and from other places in the daemon.  It works
  65  * on Windows too.
  66  */
  67 int
  68 sync_disks (void)
  69 {
  70 #if defined(HAVE_SYNC)
  71   sync ();
  72   return 0;
  73 #elif defined(WIN32)
  74   return sync_win32 ();
  75 #else
  76 #error "no known sync() API"
  77 #endif
  78 }
  79
  80 #ifdef WIN32
  81 static int
  82 sync_win32 (void)
  83 {
  84   DWORD n1, n2;
  85
  86   n1 = GetLogicalDriveStrings (0, NULL);
  87   if (n1 == 0)
  88     return -1;
  89
  90   TCHAR buffer[n1+2]; /* sic */
  91   n2 = GetLogicalDriveStrings (n1, buffer);
  92   if (n2 == 0)
  93     return -1;
  94
  95   TCHAR *p = buffer;
  96
  97   /* The MSDN example code itself assumes that there is always one
  98    * drive in the system.  However we will be better than that and not
  99    * make the assumption ...
 100    */
 101   while (*p) {
 102     HANDLE drive;
 103     DWORD drive_type;
 104
 105     if (verbose)
 106       fprintf (stderr, "sync_win32: examining drive %s\n", p);
 107
 108     /* Ignore removable drives. */
 109     drive_type = GetDriveType (p);
 110     if (drive_type == DRIVE_FIXED) {
 111       /* To open the volume you have to specify the volume name, not
 112        * the mount point.  MSDN documents use of the constant 50
 113        * below.
 114        */
 115       TCHAR volname[50];
 116       if (!GetVolumeNameForVolumeMountPoint (p, volname, 50))
 117         return -1;
 118
 119       drive = CreateFile (volname, GENERIC_READ|GENERIC_WRITE,
 120                           FILE_SHARE_READ|FILE_SHARE_WRITE,
 121                           NULL, OPEN_EXISTING, 0, 0);
 122       if (drive == INVALID_HANDLE_VALUE)
 123         return -1;
 124       if (verbose)
 125         fprintf (stderr, "sync_win32: flushing %s\n", volname);
 126
 127       BOOL r;
 128       /* This always fails in Wine:
 129        * http://bugs.winehq.org/show_bug.cgi?id=14915
 130        */
 131       r = FlushFileBuffers (drive);
 132       CloseHandle (drive);
 133       if (!r)
 134         return -1;
 135     }
 136
 137     /* Skip to next \0 character. */
 138     while (*p++);
 139   }
 140
 141   return 0;
 142 }
 143 #endif /* WIN32 */