1 files changed, 145 insertions, 0 deletions
diff --git a/src/mgr/installmgr.cpp b/src/mgr/installmgr.cpp
new file mode 100644
index 0000000..4e1cb35
--- /dev/null
+++ b/src/mgr/installmgr.cpp
@@ -0,0 +1,145 @@
+/*****************************************************************************
+ * InstallMgr functions to be made into something usefully exposed by
+ * master Glassey
+ *
+ */
+ 
+
+#include <installmgr.h>
+
+#include <fcntl.h>
+#ifndef __GNUC__
+#include <io.h>
+#else
+#include <unistd.h>
+#endif
+
+#include <curl/curl.h>
+#include <curl/types.h>
+#include <curl/easy.h>
+#include <defs.h>
+#include <vector>
+
+using namespace std;
+
+SWORD_NAMESPACE_START
+
+static InstallMgr_init _InstallMgr_init;
+
+InstallMgr_init::InstallMgr_init() {
+	curl_global_init(CURL_GLOBAL_DEFAULT);
+}
+
+InstallMgr_init::~InstallMgr_init() {
+	curl_global_cleanup();
+}
+
+
+int my_fwrite(void *buffer, size_t size, size_t nmemb, void *stream) {
+	struct FtpFile *out=(struct FtpFile *)stream;
+	if (out && !out->stream) {
+		/* open file for writing */
+		out->stream=fopen(out->filename, "wb");
+		if (!out->stream)
+			return -1; /* failure, can't open file to write */
+	}
+	return fwrite(buffer, size, nmemb, out->stream);
+}
+
+
+int my_fprogress(void *clientp, double dltotal, double dlnow, double ultotal, double ulnow) {
+	if (clientp) {
+		(*(void (*)(double, double))clientp)(dltotal, dlnow);
+	}
+	return 0;
+}
+
+
+char FTPURLGetFile(void *session, const char *dest, const char *sourceurl, bool passive, void (*status_callback)(double dltotal, double dlnow)) {
+	struct FtpFile ftpfile = {dest, NULL};
+	char retVal = 0;
+
+	CURL *curl = (CURL *)session;
+	CURLcode res;
+	
+	if (curl) {
+		curl_easy_setopt(curl, CURLOPT_URL, sourceurl);
+	
+		curl_easy_setopt(curl, CURLOPT_USERPWD, "ftp:installmgr@user.com");
+		curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, my_fwrite);
+		if (!passive)
+			curl_easy_setopt(curl, CURLOPT_FTPPORT, "-");
+		curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0);
+		curl_easy_setopt(curl, CURLOPT_PROGRESSDATA, status_callback);
+		curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, my_fprogress);
+		/* Set a pointer to our struct to pass to the callback */
+		curl_easy_setopt(curl, CURLOPT_FILE, &ftpfile);
+
+		/* Switch on full protocol/debug output */
+		curl_easy_setopt(curl, CURLOPT_VERBOSE, TRUE);
+
+		res = curl_easy_perform(curl);
+
+		if(CURLE_OK != res) {
+			retVal = -1;
+		}
+	}
+
+	if (ftpfile.stream)
+		fclose(ftpfile.stream); /* close the local file */
+	return retVal;
+}
+
+
+
+vector<struct ftpparse> FTPURLGetDir(void *session, const char *dirurl, bool passive) {
+
+	vector<struct ftpparse> dirList;
+	
+	if (!FTPURLGetFile(session, "dirlist", dirurl, passive)) {
+		int fd = open("dirlist", O_RDONLY|O_BINARY);
+		long size = lseek(fd, 0, SEEK_END);
+		lseek(fd, 0, SEEK_SET);
+		char *buf = new char [ size + 1 ];
+		read(fd, buf, size);
+		close(fd);
+		char *start = buf;
+		char *end = start;
+		while (start < (buf+size)) {
+			struct ftpparse item;
+			bool looking = true;
+			for (end = start; *end; end++) {
+				if (looking) {
+					if ((*end == 10) || (*end == 13)) {
+						*end = 0;
+						looking = false;
+					}
+				}
+				else if ((*end != 10) && (*end != 13))
+					break;
+			}
+			int status = ftpparse(&item, start, end - start);
+			if (status)
+				dirList.push_back(item);
+			start = end;
+		}
+	}
+	return dirList;
+}
+
+
+void *FTPOpenSession() {
+	CURL *curl;
+
+	curl = curl_easy_init();
+	return curl;
+}
+
+
+void FTPCloseSession(void *session) {
+	CURL *curl = (CURL *)session;
+	curl_easy_cleanup(curl);
+}
+
+SWORD_NAMESPACE_END
+