Add support for retrying file transfer.

[ffftp/ffftp.git] / mbswrapper.c

diff --git a/mbswrapper.c b/mbswrapper.c
index d44d1ef..72d8966 100644 (file)
--- a/mbswrapper.c
+++ b/mbswrapper.c
@@ -7,6 +7,7 @@
 #define UNICODE\r
 #define _UNICODE\r
 \r
+#include <stdio.h>\r
 #include <tchar.h>\r
 #include <direct.h>\r
 #include <windows.h>\r
@@ -2149,12 +2150,11 @@ START_ROUTINE
        while((c = GetNextCharM(_Str, &p)) > 0)\r
        {\r
                if(c == _Ch)\r
-               {\r
-                       r = (unsigned char*)_Str;\r
                        break;\r
-               }\r
                _Str = p;\r
        }\r
+       if(c == _Ch)\r
+               r = (unsigned char*)_Str;\r
 END_ROUTINE\r
        return r;\r
 }\r
@@ -2171,6 +2171,8 @@ START_ROUTINE
                        r = (unsigned char*)_Str;\r
                _Str = p;\r
        }\r
+       if(c == _Ch)\r
+               r = (unsigned char*)_Str;\r
 END_ROUTINE\r
        return r;\r
 }\r