[lttng-dev] [Babeltrace RFC PATCH 15/28] Add MinGW implementation of nftw()

[Ikaheimonen, JP]

Add MinGW implementation of nftw().
This implementation works only with plain directories
containing plain files.
---
 converter/babeltrace.c          |  2 +-
 include/babeltrace/compat/ftw.h | 73 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 74 insertions(+), 1 deletion(-)
 create mode 100644 include/babeltrace/compat/ftw.h

diff --git a/converter/babeltrace.c b/converter/babeltrace.c
index a34af17..7b3db49 100644
--- a/converter/babeltrace.c
+++ b/converter/babeltrace.c
@@ -48,7 +48,7 @@
 #include <fcntl.h>
 #include <unistd.h>
 #include <inttypes.h>
-#include <ftw.h>
+#include <babeltrace/compat/ftw.h>
 #include <string.h>
 
 #include <babeltrace/ctf-ir/metadata.h>	/* for clocks */
diff --git a/include/babeltrace/compat/ftw.h b/include/babeltrace/compat/ftw.h
new file mode 100644
index 0000000..f1e7bfd
--- /dev/null
+++ b/include/babeltrace/compat/ftw.h
@@ -0,0 +1,73 @@
+#ifndef _BABELTRACE_INCLUDE_COMPAT_FTW_H
+#define _BABELTRACE_INCLUDE_COMPAT_FTW_H
+
+#ifndef __MINGW32__
+#include <ftw.h>
+#else
+
+#include <windows.h>
+
+struct FTW {
+	int base;
+	int level;
+};
+
+
+#define FTW_F 0
+#define FTW_D 1
+#define FTW_DNR 2
+#define FTW_NS 3
+#define FTW_SL 4
+
+static inline
+int nftw(const char *dirpath,
+	int (*fn) (const char *fpath, const struct stat *sb,
+		int typeflag, struct FTW *ftwbuf),
+	int nopenfd, int flags)
+{
+	CHAR szSearch[MAX_PATH]         = {0};
+	CHAR szDirectory[MAX_PATH]      = {0};
+	HANDLE hFind                    = NULL;
+	WIN32_FIND_DATAA FindFileData;
+
+	sprintf(szSearch, "%s%s", dirpath, "\\*");
+	hFind = FindFirstFileA(szSearch, &FindFileData);
+	if(hFind == INVALID_HANDLE_VALUE)
+	{
+		/* print error */
+		return -1;
+	}
+	do {
+
+		if((FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY)
+		{
+			/* this is a directory */
+			/* skip '..' */
+			if (strcmp(FindFileData.cFileName, "..") == 0)
+				continue;
+
+			/* return '.' but do not recurse into it */
+			if (strcmp(FindFileData.cFileName, ".") == 0)
+			{
+				(*fn)(dirpath, 0, FTW_D, 0);
+				continue;
+			}
+
+			/* for others, recurse into them */
+			sprintf(szDirectory, "%s/%s", dirpath, FindFileData.cFileName);
+			nftw(szDirectory, fn, nopenfd, flags);
+
+		}
+		else
+		{
+			/* for files, report but do not recurse */
+			sprintf(szDirectory, "%s/%s", dirpath, FindFileData.cFileName);
+			(*fn)(szDirectory, 0, FTW_F, 0);
+		}
+	} while(FindNextFileA(hFind, &FindFileData));
+
+	FindClose(hFind);
+	return 0;
+}
+#endif
+#endif
-- 
1.8.1.msysgit.1