mirror of
https://github.com/aria2/aria2.git
synced 2025-04-05 05:27:38 +03:00
120e2de096
Added --http-auth-challenge option. If it is set to true(by default), aria2 sends HTTP authorization header only when it is requested by the server. If false is set, then authorization header is always sent to the server. This is useful for servers that don't respond 401 code when authentication is required. There is an exception: if username and password are embedded in URI, authorization header is always sent to the server regardless of this option. * src/AuthConfigFactory.cc * src/HttpSkipResponseCommand.cc * src/OptionHandlerFactory.cc * src/prefs.cc * src/prefs.h * src/usage_text.h * test/AuthConfigFactoryTest.cc * test/HttpRequestTest.cc
217 lines
7 KiB
C++
217 lines
7 KiB
C++
#include "AuthConfigFactory.h"
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "Netrc.h"
|
|
#include "prefs.h"
|
|
#include "Request.h"
|
|
#include "AuthConfig.h"
|
|
#include "Option.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class AuthConfigFactoryTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(AuthConfigFactoryTest);
|
|
CPPUNIT_TEST(testCreateAuthConfig_http);
|
|
CPPUNIT_TEST(testCreateAuthConfig_httpNoChallenge);
|
|
CPPUNIT_TEST(testCreateAuthConfig_ftp);
|
|
CPPUNIT_TEST(testUpdateBasicCred);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
|
|
public:
|
|
void testCreateAuthConfig_http();
|
|
void testCreateAuthConfig_httpNoChallenge();
|
|
void testCreateAuthConfig_ftp();
|
|
void testUpdateBasicCred();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( AuthConfigFactoryTest );
|
|
|
|
void AuthConfigFactoryTest::testCreateAuthConfig_http()
|
|
{
|
|
SharedHandle<Request> req(new Request());
|
|
req->setUrl("http://localhost/download/aria2-1.0.0.tar.bz2");
|
|
|
|
Option option;
|
|
option.put(PREF_NO_NETRC, V_FALSE);
|
|
option.put(PREF_HTTP_AUTH_CHALLENGE, V_TRUE);
|
|
|
|
AuthConfigFactory factory(&option);
|
|
|
|
// without auth info
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req).isNull());
|
|
|
|
// with Netrc
|
|
SharedHandle<Netrc> netrc(new Netrc());
|
|
netrc->addAuthenticator
|
|
(SharedHandle<Authenticator>(new Authenticator("localhost",
|
|
"localhostuser",
|
|
"localhostpass",
|
|
"localhostacct")));
|
|
netrc->addAuthenticator
|
|
(SharedHandle<Authenticator>(new DefaultAuthenticator("default", "defaultpassword", "defaultaccount")));
|
|
factory.setNetrc(netrc);
|
|
|
|
// not activated
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req).isNull());
|
|
|
|
CPPUNIT_ASSERT(factory.activateBasicCred("localhost", "/"));
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("localhostuser:localhostpass"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
// See default token in netrc is ignored.
|
|
req->setUrl("http://mirror/");
|
|
|
|
CPPUNIT_ASSERT(!factory.activateBasicCred("mirror", "/"));
|
|
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req).isNull());
|
|
|
|
// with Netrc + user defined
|
|
option.put(PREF_HTTP_USER, "userDefinedUser");
|
|
option.put(PREF_HTTP_PASSWD, "userDefinedPassword");
|
|
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req).isNull());
|
|
|
|
CPPUNIT_ASSERT(factory.activateBasicCred("mirror", "/"));
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("userDefinedUser:userDefinedPassword"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
// username and password in URI
|
|
req->setUrl("http://aria2user:aria2password@localhost/download/aria2-1.0.0.tar.bz2");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("aria2user:aria2password"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
}
|
|
|
|
void AuthConfigFactoryTest::testCreateAuthConfig_httpNoChallenge()
|
|
{
|
|
SharedHandle<Request> req(new Request());
|
|
req->setUrl("http://localhost/download/aria2-1.0.0.tar.bz2");
|
|
|
|
Option option;
|
|
option.put(PREF_NO_NETRC, V_FALSE);
|
|
|
|
AuthConfigFactory factory(&option);
|
|
|
|
// without auth info
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req).isNull());
|
|
|
|
// with Netrc
|
|
SharedHandle<Netrc> netrc(new Netrc());
|
|
netrc->addAuthenticator
|
|
(SharedHandle<Authenticator>(new Authenticator("localhost",
|
|
"localhostuser",
|
|
"localhostpass",
|
|
"localhostacct")));
|
|
netrc->addAuthenticator
|
|
(SharedHandle<Authenticator>(new DefaultAuthenticator("default", "defaultpassword", "defaultaccount")));
|
|
factory.setNetrc(netrc);
|
|
|
|
// not activated
|
|
CPPUNIT_ASSERT_EQUAL(std::string("localhostuser:localhostpass"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
// See default token in netrc is ignored.
|
|
req->setUrl("http://mirror/");
|
|
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req).isNull());
|
|
|
|
// with Netrc + user defined
|
|
option.put(PREF_HTTP_USER, "userDefinedUser");
|
|
option.put(PREF_HTTP_PASSWD, "userDefinedPassword");
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("userDefinedUser:userDefinedPassword"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
// username and password in URI
|
|
req->setUrl("http://aria2user:aria2password@localhost/download/aria2-1.0.0.tar.bz2");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("aria2user:aria2password"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
}
|
|
|
|
void AuthConfigFactoryTest::testCreateAuthConfig_ftp()
|
|
{
|
|
SharedHandle<Request> req(new Request());
|
|
req->setUrl("ftp://localhost/download/aria2-1.0.0.tar.bz2");
|
|
|
|
Option option;
|
|
option.put(PREF_NO_NETRC, V_FALSE);
|
|
|
|
AuthConfigFactory factory(&option);
|
|
|
|
// without auth info
|
|
CPPUNIT_ASSERT_EQUAL(std::string("anonymous:ARIA2USER@"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
// with Netrc
|
|
SharedHandle<Netrc> netrc(new Netrc());
|
|
netrc->addAuthenticator
|
|
(SharedHandle<Authenticator>(new DefaultAuthenticator("default", "defaultpassword", "defaultaccount")));
|
|
factory.setNetrc(netrc);
|
|
CPPUNIT_ASSERT_EQUAL(std::string("default:defaultpassword"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
// disable Netrc
|
|
option.put(PREF_NO_NETRC, V_TRUE);
|
|
CPPUNIT_ASSERT_EQUAL(std::string("anonymous:ARIA2USER@"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
// with Netrc + user defined
|
|
option.put(PREF_NO_NETRC, V_FALSE);
|
|
option.put(PREF_FTP_USER, "userDefinedUser");
|
|
option.put(PREF_FTP_PASSWD, "userDefinedPassword");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("userDefinedUser:userDefinedPassword"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
// username and password in URI
|
|
req->setUrl("ftp://aria2user:aria2password@localhost/download/aria2-1.0.0.tar.bz2");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("aria2user:aria2password"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
}
|
|
|
|
void AuthConfigFactoryTest::testUpdateBasicCred()
|
|
{
|
|
Option option;
|
|
option.put(PREF_NO_NETRC, V_FALSE);
|
|
option.put(PREF_HTTP_AUTH_CHALLENGE, V_TRUE);
|
|
|
|
AuthConfigFactory factory(&option);
|
|
|
|
factory.updateBasicCred
|
|
(AuthConfigFactory::BasicCred("myname", "mypass", "localhost", "/", true));
|
|
factory.updateBasicCred
|
|
(AuthConfigFactory::BasicCred("price","j38jdc","localhost","/download", true));
|
|
factory.updateBasicCred
|
|
(AuthConfigFactory::BasicCred("alice","ium8","localhost","/documents", true));
|
|
factory.updateBasicCred
|
|
(AuthConfigFactory::BasicCred("jack", "jackx","mirror", "/doc", true));
|
|
|
|
SharedHandle<Request> req(new Request());
|
|
req->setUrl("http://localhost/download/v2.6/Changelog");
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("price:j38jdc"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
req->setUrl("http://localhost/documents/reference.html");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("alice:ium8"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
req->setUrl("http://localhost/documents2/manual.html");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("myname:mypass"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
req->setUrl("http://localhost/doc/readme.txt");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("myname:mypass"),
|
|
factory.createAuthConfig(req)->getAuthText());
|
|
|
|
req->setUrl("http://local/");
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req).isNull());
|
|
|
|
req->setUrl("http://mirror/");
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req).isNull());
|
|
}
|
|
|
|
} // namespace aria2
|