Changeset 82687 in vbox for trunk/src/VBox/Runtime/tools

Timestamp:

Jan 9, 2020 10:45:36 AM (5 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

135606

Message:

IPRT/FTP: More protocol handling and callback work. bugref:9437

File:

• trunk/src/VBox/Runtime/tools/RTFTPServer.cpp (modified) (7 diffs)

trunk/src/VBox/Runtime/tools/RTFTPServer.cpp

@@ -57 +57 @@
 *********************************************************************************************************************************/
 /** Set by the signal handler when the FTP server shall be terminated. */
-static volatile bool g_fCanceled = false;
+static volatile bool  g_fCanceled  = false;
+static char          *g_pszRootDir = NULL;
 
 
@@ -145 +146 @@
 }
 
+static DECLCALLBACK(int) onUserConnect(PRTFTPCALLBACKDATA pData, const char *pcszUser)
+{
+    RT_NOREF(pData, pcszUser);
+
+    RTPrintf("User '%s' connected", pcszUser);
+
+    return VINF_SUCCESS;
+}
+
+static DECLCALLBACK(int) onUserAuthenticate(PRTFTPCALLBACKDATA pData, const char *pcszUser, const char *pcszPassword)
+{
+    RT_NOREF(pData, pcszUser, pcszPassword);
+
+    RTPrintf("Authenticating user '%s' ...", pcszUser);
+
+    return VINF_SUCCESS;
+}
+
+static DECLCALLBACK(int) onUserDisonnect(PRTFTPCALLBACKDATA pData)
+{
+    RT_NOREF(pData);
+
+    RTPrintf("User disconnected");
+
+    return VINF_SUCCESS;
+}
+
+static DECLCALLBACK(int) onPathSetCurrent(PRTFTPCALLBACKDATA pData, const char *pcszCWD)
+{
+    RT_NOREF(pData, pcszCWD);
+
+    RTPrintf("Setting current directory to '%s'\n", pcszCWD);
+
+    return VINF_SUCCESS;
+}
+
+static DECLCALLBACK(int) onPathGetCurrent(PRTFTPCALLBACKDATA pData, char *pszPWD, size_t cbPWD)
+{
+    RT_NOREF(pData, pszPWD, cbPWD);
+
+    return VINF_SUCCESS;
+}
+
+static DECLCALLBACK(int) onList(PRTFTPCALLBACKDATA pData, void **ppvData, size_t *pcbData)
+{
+    RT_NOREF(pData, ppvData, pcbData);
+
+    return VINF_SUCCESS;
+}
+
 int main(int argc, char **argv)
 {
@@ -154 +205 @@
     char     szAddress[64]         = "localhost";
     uint16_t uPort                 = 2121;
-
-    /* By default use the current directory as serving root directory. */
-    char     szRootDir[RTPATH_MAX];
-    rc = RTPathGetCurrent(szRootDir, sizeof(szRootDir));
-    if (RT_FAILURE(rc))
-        return RTMsgErrorExit(RTEXITCODE_FAILURE, "Retrieving current directory failed: %Rrc", rc);
 
     /*
@@ -190 +235 @@
             case 'p':
                 uPort = ValueUnion.u16;
+                break;
+
+            case 'r':
+                g_pszRootDir = RTStrDup(ValueUnion.psz);
                 break;
 
@@ -224 +273 @@
     }
 
+    if (!g_pszRootDir)
+    {
+        char szRootDir[RTPATH_MAX];
+
+        /* By default use the current directory as serving root directory. */
+        rc = RTPathGetCurrent(szRootDir, sizeof(szRootDir));
+        if (RT_FAILURE(rc))
+            return RTMsgErrorExit(RTEXITCODE_FAILURE, "Retrieving current directory failed: %Rrc", rc);
+
+        g_pszRootDir = RTStrDup(szRootDir);
+        if (!g_pszRootDir)
+            return RTMsgErrorExit(RTEXITCODE_FAILURE, "Allocating current directory failed");
+    }
+
     /* Install signal handler. */
     rc = signalHandlerInstall();
@@ -231 +294 @@
          * Create the FTP server instance.
          */
+        RTFTPSERVERCALLBACKS Callbacks;
+        RT_ZERO(Callbacks);
+        Callbacks.pfnOnUserConnect      = onUserConnect;
+        Callbacks.pfnOnUserAuthenticate = onUserAuthenticate;
+        Callbacks.pfnOnUserDisconnect   = onUserDisonnect;
+        Callbacks.pfnOnPathSetCurrent   = onPathSetCurrent;
+        Callbacks.pfnOnPathGetCurrent   = onPathGetCurrent;
+        Callbacks.pfnOnList             = onList;
+
         RTFTPSERVER hFTPServer;
-        rc = RTFTPServerCreate(&hFTPServer, szAddress, uPort, szRootDir);
+        rc = RTFTPServerCreate(&hFTPServer, szAddress, uPort, &Callbacks);
         if (RT_SUCCESS(rc))
         {
             RTPrintf("Starting FTP server at %s:%RU16 ...\n", szAddress, uPort);
-            RTPrintf("Root directory is '%s'\n", szRootDir);
+            RTPrintf("Root directory is '%s'\n", g_pszRootDir);
 
             RTPrintf("Running FTP server ...\n");
@@ -264 +336 @@
     }
 
+    RTStrFree(g_pszRootDir);
+
     /* Set rcExit on failure in case we forgot to do so before. */
     if (RT_FAILURE(rc))