root/trunk/0024_URIStreamOpener/URIStreamOpenerTest.cpp

User picture

Author: Setsu

Revision: 348 («Previous)


File Size: 4.5 KB

(June 16, 2010 11:06 UTC) Almost 2 years ago

0024_URIStreamOpener

 
Show/hide line numbers
//
//  URIStreamOpenerTest.cpp
//
//  Created by Setsu on 6/2/2010.
//  Copyright 2010 RoundSquare Inc. All rights reserved.
//
// Permission is hereby granted, free of charge, to any person or organization
// obtaining a copy of the software and accompanying documentation covered by
// this license (the "Software") to use, reproduce, display, distribute,
// execute, and transmit the Software, and to prepare derivative works of the
// Software, and to permit third-parties to whom the Software is furnished to
// do so, all subject to the following:
// 
// The copyright notices in the Software and this entire statement, including
// the above license grant, this restriction and the following disclaimer,
// must be included in all copies of the Software, in whole or in part, and
// all derivative works of the Software, unless such copies or derivative
// works are solely in the form of machine-executable object code generated by
// a source language processor.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
// SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
// FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
//

//----------------------------------------
//	include
//----------------------------------------
#include <Poco/Logger.h>
#include <Poco/PatternFormatter.h>
#include <Poco/FormattingChannel.h>
#include <Poco/ConsoleChannel.h>
#include <Poco/URI.h>
#include <Poco/Format.h>
#include <Poco/URIStreamOpener.h>
#include <Poco/Net/HTTPStreamFactory.h>
#include <Poco/Net/FTPStreamFactory.h>
#include <Poco/StreamCopier.h>
#include <Poco/TemporaryFile.h>
#include <Poco/Path.h>
#include <Poco/Thread.h>

#include <sstream>
#include <fstream>

#include "ScopedLogMessage.h"

//----------------------------------------
//	TestURIStreamOpener
//----------------------------------------
void TestURIStreamOpener(ScopedLogMessage& msg, const Poco::URI& uri)
{
	msg.Message(Poco::format(" --- %s ---", uri.toString()));
	if("ftp" == uri.getScheme())
	{
		std::string password;
		std::string user(uri.getUserInfo());
		if("" == user)
		{
			user = "anonymous";
			password = Poco::Net::FTPStreamFactory::getAnonymousPassword();
		}
		else
		{
			Poco::Net::FTPPasswordProvider* pProvider = Poco::Net::FTPStreamFactory::getPasswordProvider();
			if(NULL != pProvider)
			{
				password = pProvider->password(user, uri.getAuthority());
			}
		}
		msg.Message(Poco::format("     (user=%s, password=%s)", user, password));
	}
	try
	{
		std::auto_ptr<std::istream> pStr(Poco::URIStreamOpener::defaultOpener().open(uri));
		std::stringstream ss;
		Poco::StreamCopier::copyStream(*pStr.get(), ss);

		Poco::Thread* p_thread = Poco::Thread::current();
		int threadID = (0 == p_thread) ? 0:p_thread->id();

		msg.Message(Poco::format(" <start>\n%s[%d]  <end>", ss.str(), threadID));
	}
	catch(Poco::Exception& exc)
	{
		msg.Message(Poco::format("    %s", exc.displayText()));
	}
}

//----------------------------------------
//	PrepareConsoleLogger
//----------------------------------------
void PrepareConsoleLogger(const std::string& name, int level=Poco::Message::PRIO_INFORMATION)
{
	Poco::FormattingChannel* pFCConsole = new Poco::FormattingChannel(new Poco::PatternFormatter("%t"));
	pFCConsole->setChannel(new Poco::ConsoleChannel);
	pFCConsole->open();

	Poco::Logger::create(name, pFCConsole, level);
}

//----------------------------------------
//	main
//----------------------------------------
int main(int /*argc*/, char** /*argv*/)
{
	PrepareConsoleLogger(Poco::Logger::ROOT, Poco::Message::PRIO_INFORMATION);

	ScopedLogMessage msg("URIStreamOpenerTest ", "start", "end");

	Poco::Net::HTTPStreamFactory::registerFactory();
	Poco::Net::FTPStreamFactory::registerFactory();

	// HTTP
	Poco::URI uriHTTP("http://poco.roundsquare.net/downloads/test.txt");
	TestURIStreamOpener(msg, uriHTTP);

	// FTP
	Poco::URI uriFTP("ftp://ftp.iij.ad.jp/pub/linux/debian/dists/README");
	TestURIStreamOpener(msg, uriFTP);

	// file
	Poco::TemporaryFile tempFile;
	std::string path = tempFile.path();
	std::ofstream ostr(path.c_str());
	ostr << "Hello, world!" << std::endl;
	ostr.close();
	Poco::URI uriFile;
	uriFile.setScheme("file");
	uriFile.setPath(Poco::Path(path).toString(Poco::Path::PATH_UNIX));
	TestURIStreamOpener(msg, uriFile);

	return 0;
}