/***************************************************************************** * * remotetrans.cpp - * * $Id$ * * Copyright 2004-2013 CrossWire Bible Society (http://www.crosswire.org) * CrossWire Bible Society * P. O. Box 2528 * Tempe, AZ 85280-2528 * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License as published by the * Free Software Foundation version 2. * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * General Public License for more details. * */ #include #include #include #include #include extern "C" { #include } using std::vector; SWORD_NAMESPACE_START namespace { void removeTrailingSlash(SWBuf &buf) { int len = buf.size(); if ((buf[len-1] == '/') || (buf[len-1] == '\\')) buf.size(len-1); } }; void StatusReporter::preStatus(long totalBytes, long completedBytes, const char *message) { } void StatusReporter::statusUpdate(double dtTotal, double dlNow) { } RemoteTransport::RemoteTransport(const char *host, StatusReporter *statusReporter) { this->statusReporter = statusReporter; this->host = host; u = "ftp"; p = "installmgr@user.com"; term = false; passive = true; unverifiedPeerAllowed = true; } RemoteTransport::~RemoteTransport() { } // override this method in your real transport class char RemoteTransport::getURL(const char *destPath, const char *sourceURL, SWBuf *destBuf) { SWLog::getSystemLog()->logWarning("RemoteTransport::getURL called but unsupported"); char retVal = -1; return retVal; } // override this method in your real transport class char RemoteTransport::putURL(const char *destURL, const char *sourcePath, SWBuf *sourceBuf) { SWLog::getSystemLog()->logWarning("RemoteTransport::putURL called but unsupported"); char retVal = -1; return retVal; } vector RemoteTransport::getDirList(const char *dirURL) { SWLog::getSystemLog()->logDebug("RemoteTransport::getDirList(%s)", dirURL); vector dirList; SWBuf dirBuf; if (!getURL("", dirURL, &dirBuf)) { char *start = dirBuf.getRawData(); char *end = start; while (start < (dirBuf.getRawData()+dirBuf.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; } SWLog::getSystemLog()->logDebug("getDirList: parsing item %s(%d)\n", start, end-start); int status = ftpparse(&item, start, (int)(end - start)); // in ftpparse.h, there is a warning that name is not necessarily null terminated SWBuf name; name.append(item.name, item.namelen); SWLog::getSystemLog()->logDebug("getDirList: got item %s\n", name.c_str()); if (status && name != "." && name != "..") { struct DirEntry i; i.name = name; i.size = item.size; i.isDirectory = (item.flagtrycwd == 1); dirList.push_back(i); } start = end; } } else { SWLog::getSystemLog()->logWarning("getDirList: failed to get dir %s\n", dirURL); } return dirList; } int RemoteTransport::copyDirectory(const char *urlPrefix, const char *dir, const char *dest, const char *suffix) { SWLog::getSystemLog()->logDebug("RemoteTransport::copyDirectory"); int retVal = 0; SWBuf url = SWBuf(urlPrefix) + SWBuf(dir); removeTrailingSlash(url); url += '/'; SWLog::getSystemLog()->logDebug("NetTransport: getting dir %s\n", url.c_str()); vector dirList = getDirList(url.c_str()); if (!dirList.size()) { SWLog::getSystemLog()->logWarning("NetTransport: failed to read dir %s\n", url.c_str()); return -1; } // append files in sub directories and calculate total download size unsigned int i = 0; long totalBytes = 0; for (;;) { if (i == dirList.size()) break; struct DirEntry &e = dirList.at(i); if (e.isDirectory) { SWBuf name(e.name); // &e will be invalidated after first insertion vector sd = getDirList((url + name + '/').c_str()); for (unsigned int ii = 0; ii < sd.size(); ii++) { sd[ii].name = name + '/' + sd[ii].name; dirList.push_back(sd[ii]); } dirList.erase(dirList.begin() + i); } else { totalBytes += e.size; i++; } } long completedBytes = 0; for (i = 0; i < dirList.size(); i++) { struct DirEntry &dirEntry = dirList[i]; SWBuf buffer = (SWBuf)dest; removeTrailingSlash(buffer); buffer += "/"; buffer += dirEntry.name; if (!strcmp(&buffer.c_str()[buffer.length()-strlen(suffix)], suffix)) { SWBuf buffer2 = "Downloading ("; buffer2.appendFormatted("%d", i+1); buffer2 += " of "; buffer2.appendFormatted("%d", dirList.size()); buffer2 += "): "; buffer2 += dirEntry.name; if (statusReporter) statusReporter->preStatus(totalBytes, completedBytes, buffer2.c_str()); FileMgr::createParent(buffer.c_str()); // make sure parent directory exists SWTRY { SWBuf url = (SWBuf)urlPrefix + (SWBuf)dir; removeTrailingSlash(url); url += "/"; url += dirEntry.name; if (getURL(buffer.c_str(), url.c_str())) { SWLog::getSystemLog()->logWarning("copyDirectory: failed to get file %s\n", url.c_str()); return -2; } completedBytes += dirEntry.size; } SWCATCH (...) {} if (term) { retVal = -3; break; } } } return retVal; } #if defined(__GNUC__) #pragma GCC diagnostic ignored "-Wdeprecated-declarations" #endif void StatusReporter::update(unsigned long totalBytes, unsigned long completedBytes) { statusUpdate(totalBytes, completedBytes); } SWORD_NAMESPACE_END