Add SMB implementation to UWP

Author: cscd98

libretro-common/vfs/vfs_implementation_uwp.cpp

@@ -53,6 +53,10 @@
 #include <uwp/uwp_async.h>
 #include <uwp/std_filesystem_compat.h>
 
+#ifdef HAVE_SMBCLIENT
+#include "vfs_implementation_smb.h"
+#endif
+
 namespace
 {
    /* UWP deals with paths containing / instead of
@@ -90,6 +94,9 @@ struct libretro_vfs_implementation_file
     int fd;
     unsigned hints;
     enum vfs_scheme scheme;
+#ifdef HAVE_SMBCLIENT
+    intptr_t smb_fh;
+#endif
 };
 
 #define RFILE_HINT_UNBUFFERED (1 << 8)
@@ -99,6 +106,13 @@ int retro_vfs_file_close_impl(libretro_vfs_implementation_file* stream)
     if (!stream)
         return -1;
 
+#ifdef HAVE_SMBCLIENT
+    if (stream->scheme == VFS_SCHEME_SMB)
+    {
+        retro_vfs_file_close_smb(stream);
+    }
+#endif
+
     if (stream->fp)
         fclose(stream->fp);
 
@@ -117,6 +131,11 @@ int retro_vfs_file_close_impl(libretro_vfs_implementation_file* stream)
 
 int retro_vfs_file_error_impl(libretro_vfs_implementation_file* stream)
 {
+#ifdef HAVE_SMBCLIENT
+    if (stream->scheme == VFS_SCHEME_SMB)
+        return retro_vfs_file_error_smb(stream);
+#endif
+
     return ferror(stream->fp);
 }
 
@@ -142,7 +161,13 @@ int64_t retro_vfs_file_tell_impl(libretro_vfs_implementation_file* stream)
         return -1;
 
     if ((stream->hints & RFILE_HINT_UNBUFFERED) == 0)
+    {
+#ifdef HAVE_SMBCLIENT
+        if (stream->scheme == VFS_SCHEME_SMB)
+           return retro_vfs_file_tell_smb(stream);
+#endif
         return _ftelli64(stream->fp);
+    }
     if ((val = lseek(stream->fd, 0, SEEK_CUR)) < 0)
         return -1;
 
@@ -159,7 +184,13 @@ int64_t retro_vfs_file_seek_internal(
         return -1;
 
     if ((stream->hints & RFILE_HINT_UNBUFFERED) == 0)
+    {
+#ifdef HAVE_SMBCLIENT
+        if (stream->scheme == VFS_SCHEME_SMB)
+            return retro_vfs_file_seek_smb(stream, offset, whence);
+#endif
         return _fseeki64(stream->fp, offset, whence);
+    }
     if ((val = lseek(stream->fd, (off_t)offset, whence)) < 0)
         return -1;
 
@@ -175,25 +206,51 @@ int64_t retro_vfs_file_seek_impl(libretro_vfs_implementation_file* stream,
 int64_t retro_vfs_file_read_impl(libretro_vfs_implementation_file* stream,
     void* s, uint64_t len)
 {
-    if (!stream || (!stream->fp && stream->fh == INVALID_HANDLE_VALUE) || !s)
-      return -1;
+    if (!stream)
+        return -1;
 
-   if (stream->fh != INVALID_HANDLE_VALUE)
-   {
-      DWORD _bytes_read;
-      ReadFile(stream->fh, (char*)s, len, &_bytes_read, NULL);
-      return (int64_t)_bytes_read;
-   }
+#ifdef HAVE_SMBCLIENT
+    if (stream->scheme == VFS_SCHEME_SMB)
+        return retro_vfs_file_read_smb(stream, s, len);
+#endif
+
+    if ((!stream->fp && stream->fh == INVALID_HANDLE_VALUE) || !s)
+        return -1;
+
+    if (stream->fh != INVALID_HANDLE_VALUE)
+    {
+        DWORD _bytes_read;
+        ReadFile(stream->fh, (char*)s, len, &_bytes_read, NULL);
+        return (int64_t)_bytes_read;
+    }
 
     if ((stream->hints & RFILE_HINT_UNBUFFERED) == 0)
-       return fread(s, 1, (size_t)len, stream->fp);
+        return fread(s, 1, (size_t)len, stream->fp);
     return read(stream->fd, s, (size_t)len);
 }
 
 
 int64_t retro_vfs_file_write_impl(libretro_vfs_implementation_file* stream, const void* s, uint64_t len)
 {
-    if (!stream || (!stream->fp && stream->fh == INVALID_HANDLE_VALUE) || !s)
+    if (!stream)
+        return -1;
+
+#ifdef HAVE_SMBCLIENT
+    if ((stream->hints & RFILE_HINT_UNBUFFERED) == 0 &&
+        stream->scheme == VFS_SCHEME_SMB)
+    {
+        int64_t pos = 0;
+        ssize_t ret = -1;
+
+        pos = retro_vfs_file_tell_smb(stream);
+        ret = retro_vfs_file_write_smb(stream, s, len);
+        if (ret != -1 && pos + ret > stream->size)
+            stream->size = pos + ret;
+        return ret;
+    }
+#endif
+
+    if ((!stream->fp && stream->fh == INVALID_HANDLE_VALUE) || !s)
         return -1;
 
     if (stream->fh != INVALID_HANDLE_VALUE)
@@ -264,6 +321,9 @@ libretro_vfs_implementation_file* retro_vfs_file_open_impl(
     stream->mapsize   = 0;
     stream->mapped    = NULL;
     stream->scheme    = VFS_SCHEME_NONE;
+#ifdef HAVE_SMBCLIENT
+    stream->smb_fh    = 0;
+#endif
 
 #ifdef VFS_FRONTEND
    if (     path
@@ -280,6 +340,20 @@ libretro_vfs_implementation_file* retro_vfs_file_open_impl(
          path             += sizeof("vfsonly://")-1;
 #endif
 
+#ifdef HAVE_SMBCLIENT
+   if (     path
+         && path[0] == 's'
+         && path[1] == 'm'
+         && path[2] == 'b'
+         && path[3] == ':'
+         && path[4] == '/'
+         && path[5] == '/'
+         && path[6] != '\0')
+   {
+        stream->scheme    = VFS_SCHEME_SMB;
+   }
+#endif
+
     path_wide    = utf8_to_utf16_string_alloc(path);
     windowsize_path(path_wide);
     path_wstring = path_wide;
@@ -344,6 +418,16 @@ libretro_vfs_implementation_file* retro_vfs_file_open_impl(
            ? OPEN_ALWAYS
            : CREATE_ALWAYS;
 
+#ifdef HAVE_SMBCLIENT
+    if (stream->scheme == VFS_SCHEME_SMB)
+    {
+        if (!retro_vfs_file_open_smb(stream, path, mode, hints))
+            goto error;
+
+        return stream;
+    }
+#endif
+
     if ((file_handle = CreateFile2FromAppW(path_wstring.data(), desireAccess,
                 FILE_SHARE_READ, creationDisposition, NULL)) == INVALID_HANDLE_VALUE)
        goto error;
@@ -658,6 +742,10 @@ struct libretro_vfs_implementation_dir
     HANDLE directory;
     bool next;
     char path[PATH_MAX_LENGTH];
+#ifdef HAVE_SMBCLIENT
+    intptr_t smb_directory;
+    char smb_path[PATH_MAX_LENGTH];
+#endif
 };
 
 libretro_vfs_implementation_dir* retro_vfs_opendir_impl(
@@ -680,6 +768,23 @@ libretro_vfs_implementation_dir* retro_vfs_opendir_impl(
 
     _len = strlcpy(path_buf, name, sizeof(path_buf));
 
+#ifdef HAVE_SMBCLIENT
+   if (name[0]=='s' && name[1]=='m' && name[2]=='b' &&
+       name[3]==':' && name[4]=='/' && name[5]=='/' && name[6] != '\0')
+   {
+      intptr_t dh = retro_vfs_opendir_smb(name, include_hidden);
+      if (dh < 0)
+      {
+         free(rdir->orig_path);
+         free(rdir);
+         return NULL;
+      }
+      rdir->smb_directory = dh;
+      rdir->smb_path[0] = '\0';
+      return rdir;
+   }
+#endif
+
     /* Non-NT platforms don't like extra slashes in the path */
     if (path_buf[_len - 1] != '\\')
         path_buf[_len++]   = '\\';
@@ -709,6 +814,22 @@ libretro_vfs_implementation_dir* retro_vfs_opendir_impl(
 
 bool retro_vfs_readdir_impl(libretro_vfs_implementation_dir* rdir)
 {
+#ifdef HAVE_SMBCLIENT
+    if (rdir->smb_directory != 0)
+    {
+       struct smbc_dirent *de = retro_vfs_readdir_smb(rdir->smb_directory);
+       if (!de)
+          return false;
+       strlcpy(rdir->smb_path, de->name, sizeof(rdir->smb_path));
+       return true;
+    }
+    /* If we opened an SMB path but failed, do not fall through to native readdir */
+    if (rdir->orig_path &&
+        rdir->orig_path[0]=='s' && rdir->orig_path[1]=='m' && rdir->orig_path[2]=='b' &&
+        rdir->orig_path[3]==':' && rdir->orig_path[4]=='/' && rdir->orig_path[5]=='/' )
+       return false;
+#endif
+
     if (rdir->next)
         return (FindNextFileW(rdir->directory, &rdir->entry) != 0);
 
@@ -718,6 +839,10 @@ bool retro_vfs_readdir_impl(libretro_vfs_implementation_dir* rdir)
 
 const char* retro_vfs_dirent_get_name_impl(libretro_vfs_implementation_dir* rdir)
 {
+#ifdef HAVE_SMBCLIENT
+    if (rdir->smb_directory != 0)
+       return rdir->smb_path;
+#endif
     char* name = utf16_to_utf8_string_alloc(rdir->entry.cFileName);
     memset(rdir->entry.cFileName, 0, sizeof(rdir->entry.cFileName));
     strlcpy((char*)rdir->entry.cFileName, name, sizeof(rdir->entry.cFileName));
@@ -728,6 +853,22 @@ const char* retro_vfs_dirent_get_name_impl(libretro_vfs_implementation_dir* rdir
 
 bool retro_vfs_dirent_is_dir_impl(libretro_vfs_implementation_dir* rdir)
 {
+#ifdef HAVE_SMBCLIENT
+    if (rdir->smb_directory != 0)
+    {
+       char full[PATH_MAX_LENGTH];
+       const char *name = retro_vfs_dirent_get_name_impl(rdir);
+
+       if (!name)
+          return false;
+
+       fill_pathname_join_special(full, rdir->orig_path, name, sizeof(full));
+       int32_t sz = 0;
+       int st = retro_vfs_stat_smb(full, &sz);
+
+       return (st & RETRO_VFS_STAT_IS_DIRECTORY) != 0;
+    }
+#endif
     const WIN32_FIND_DATA* entry = (const WIN32_FIND_DATA*)&rdir->entry;
     return entry->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
 }
@@ -737,6 +878,14 @@ int retro_vfs_closedir_impl(libretro_vfs_implementation_dir* rdir)
     if (!rdir)
         return -1;
 
+#ifdef HAVE_SMBCLIENT
+    if (rdir->smb_directory != 0)
+    {
+        retro_vfs_closedir_smb(rdir->smb_directory);
+        rdir->smb_directory = 0;
+    }
+#endif
+
     if (rdir->directory != INVALID_HANDLE_VALUE)
         FindClose(rdir->directory);