diff options
Diffstat (limited to 'RealtimeSync/watcher.cpp')
| -rw-r--r-- | RealtimeSync/watcher.cpp | 198 |
1 files changed, 198 insertions, 0 deletions
diff --git a/RealtimeSync/watcher.cpp b/RealtimeSync/watcher.cpp new file mode 100644 index 00000000..4949f01a --- /dev/null +++ b/RealtimeSync/watcher.cpp @@ -0,0 +1,198 @@ +#include "watcher.h" +#include "../shared/systemFunctions.h" +#include "functions.h" +#include <wx/intl.h> +#include <wx/filefn.h> +#include "../shared/fileHandling.h" + +#ifdef FFS_WIN +#include <wx/msw/wrapwin.h> //includes "windows.h" + +#elif defined FFS_LINUX +#include <wx/timer.h> +#include <exception> +#include "../shared/inotify/inotify-cxx.h" +#include "../shared/fileTraverser.h" +#endif + + +#ifdef FFS_WIN +class ChangeNotifications +{ +public: + ~ChangeNotifications() + { + for (std::vector<HANDLE>::const_iterator i = arrayHandle.begin(); i != arrayHandle.end(); ++i) + if (*i != INVALID_HANDLE_VALUE) + ::FindCloseChangeNotification(*i); + } + + void addHandle(const HANDLE hndl) + { + arrayHandle.push_back(hndl); + } + + size_t getSize() + { + return arrayHandle.size(); + } + + const HANDLE* getArray() + { + return &arrayHandle[0]; //client needs to check getSize() before calling this method! + } + +private: + std::vector<HANDLE> arrayHandle; +}; + +#elif defined FFS_LINUX +class DirsOnlyTraverser : public FreeFileSync::TraverseCallback +{ +public: + DirsOnlyTraverser(std::vector<std::string>& dirs) : m_dirs(dirs) {} + + virtual ReturnValue onFile(const DefaultChar* shortName, const Zstring& fullName, const FileInfo& details) + { + return TRAVERSING_CONTINUE; + } + virtual ReturnValDir onDir(const DefaultChar* shortName, const Zstring& fullName) + { + m_dirs.push_back(fullName.c_str()); + return ReturnValDir(ReturnValDir::Continue(), this); + } + virtual ReturnValue onError(const wxString& errorText) + { + throw FreeFileSync::FileError(errorText); + } + +private: + std::vector<std::string>& m_dirs; +}; +#endif + + +void RealtimeSync::waitForChanges(const std::vector<wxString>& dirNames, WaitCallback* statusHandler) +{ + if (dirNames.empty()) //pathological case, but check is needed later + return; + +#ifdef FFS_WIN + ChangeNotifications notifications; + + for (std::vector<wxString>::const_iterator i = dirNames.begin(); i != dirNames.end(); ++i) + { + const Zstring formattedDir = FreeFileSync::getFormattedDirectoryName(i->c_str()); + + if (formattedDir.empty()) + throw FreeFileSync::FileError(_("Please fill all empty directory fields.")); + else if (!FreeFileSync::dirExists(formattedDir)) + throw FreeFileSync::FileError(wxString(_("Directory does not exist:")) + wxT("\n") + + wxT("\"") + formattedDir + wxT("\"") + wxT("\n\n") + + FreeFileSync::getLastErrorFormatted()); + + const HANDLE rv = ::FindFirstChangeNotification( + formattedDir.c_str(), //__in LPCTSTR lpPathName, + true, //__in BOOL bWatchSubtree, + FILE_NOTIFY_CHANGE_FILE_NAME | + FILE_NOTIFY_CHANGE_DIR_NAME | + FILE_NOTIFY_CHANGE_SIZE | + FILE_NOTIFY_CHANGE_LAST_WRITE); //__in DWORD dwNotifyFilter + + if (rv == INVALID_HANDLE_VALUE) + { + const wxString errorMessage = wxString(_("Could not initialize directory monitoring:")) + wxT("\n\"") + *i + wxT("\""); + throw FreeFileSync::FileError(errorMessage + wxT("\n\n") + FreeFileSync::getLastErrorFormatted()); + } + + notifications.addHandle(rv); + } + + + while (true) + { + const DWORD rv = ::WaitForMultipleObjects( //NOTE: notifications.getArray() returns valid pointer, because it cannot be empty in this context + notifications.getSize(), //__in DWORD nCount, + notifications.getArray(), //__in const HANDLE *lpHandles, + false, //__in BOOL bWaitAll, + UI_UPDATE_INTERVAL); //__in DWORD dwMilliseconds + if (WAIT_OBJECT_0 <= rv && rv < WAIT_OBJECT_0 + notifications.getSize()) + return; //directory change detected + else if (rv == WAIT_FAILED) + throw FreeFileSync::FileError(wxString(_("Error when monitoring directories.")) + wxT("\n\n") + FreeFileSync::getLastErrorFormatted()); + else if (rv == WAIT_TIMEOUT) + statusHandler->requestUiRefresh(); + } + +#elif defined FFS_LINUX + std::vector<std::string> fullDirList; + + //add all subdirectories + for (std::vector<wxString>::const_iterator i = dirNames.begin(); i != dirNames.end(); ++i) + { + const Zstring formattedDir = FreeFileSync::getFormattedDirectoryName(i->c_str()); + + if (formattedDir.empty()) + throw FreeFileSync::FileError(_("Please fill all empty directory fields.")); + + else if (!FreeFileSync::dirExists(formattedDir)) + throw FreeFileSync::FileError(wxString(_("Directory does not exist:")) + wxT("\n") + + wxT("\"") + formattedDir + wxT("\"") + wxT("\n\n") + + FreeFileSync::getLastErrorFormatted()); + + fullDirList.push_back(formattedDir.c_str()); + //get all subdirectories + DirsOnlyTraverser traverser(fullDirList); + FreeFileSync::traverseFolder(formattedDir, false, &traverser); //don't traverse into symlinks (analog to windows build) + } + + try + { + Inotify notifications; + notifications.SetNonBlock(true); + + for (std::vector<std::string>::const_iterator i = fullDirList.begin(); i != fullDirList.end(); ++i) + { + try + { + InotifyWatch newWatch(*i, //dummy object: InotifyWatch may be destructed safely after Inotify::Add() + IN_DONT_FOLLOW | //don't follow symbolic links + IN_ONLYDIR | //watch directories only + IN_CLOSE_WRITE | + IN_CREATE | + IN_DELETE | + IN_DELETE_SELF | + IN_MODIFY | + IN_MOVE_SELF | + IN_MOVED_FROM | + IN_MOVED_TO ); + notifications.Add(newWatch); + } + catch (const InotifyException& e) + { + const wxString errorMessage = wxString(_("Could not initialize directory monitoring:")) + wxT("\n\"") + *i + wxT("\""); + throw FreeFileSync::FileError(errorMessage + wxT("\n\n") + e.GetMessage().c_str()); + } + } + + while (true) + { + notifications.WaitForEvents(); //called in non-blocking mode + + if (notifications.GetEventCount() > 0) + return; //directory change detected + + wxMilliSleep(RealtimeSync::UI_UPDATE_INTERVAL); + statusHandler->requestUiRefresh(); + } + } + catch (const InotifyException& e) + { + throw FreeFileSync::FileError(wxString(_("Error when monitoring directories.")) + wxT("\n\n") + e.GetMessage().c_str()); + } + catch (const std::exception& e) + { + throw FreeFileSync::FileError(wxString(_("Error when monitoring directories.")) + wxT("\n\n") + e.what()); + } +#endif +} |