mirror of
https://github.com/aria2/aria2.git
synced 2025-02-26 08:22:11 +00:00
14a47f43f8
If user name is embedded but password is missing in URI, first resolve password using .netrc. If password is found in .netrc, then use it as password. If not, use the password specified in --ftp-passwd. * src/AuthConfigFactory.cc * src/Request.cc * src/Request.h * test/AuthConfigFactoryTest.cc
241 lines
8.3 KiB
C++
241 lines
8.3 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;
|
|
|
|
// without auth info
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req, &option).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, &option).isNull());
|
|
|
|
CPPUNIT_ASSERT(factory.activateBasicCred("localhost", "/", &option));
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("localhostuser:localhostpass"),
|
|
factory.createAuthConfig(req, &option)->getAuthText());
|
|
|
|
// See default token in netrc is ignored.
|
|
req->setUrl("http://mirror/");
|
|
|
|
CPPUNIT_ASSERT(!factory.activateBasicCred("mirror", "/", &option));
|
|
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req, &option).isNull());
|
|
|
|
// with Netrc + user defined
|
|
option.put(PREF_HTTP_USER, "userDefinedUser");
|
|
option.put(PREF_HTTP_PASSWD, "userDefinedPassword");
|
|
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req, &option).isNull());
|
|
|
|
CPPUNIT_ASSERT(factory.activateBasicCred("mirror", "/", &option));
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("userDefinedUser:userDefinedPassword"),
|
|
factory.createAuthConfig(req, &option)->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, &option)->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;
|
|
|
|
// without auth info
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req, &option).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, &option)->getAuthText());
|
|
|
|
// See default token in netrc is ignored.
|
|
req->setUrl("http://mirror/");
|
|
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req, &option).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, &option)->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, &option)->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;
|
|
|
|
// without auth info
|
|
CPPUNIT_ASSERT_EQUAL(std::string("anonymous:ARIA2USER@"),
|
|
factory.createAuthConfig(req, &option)->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, &option)->getAuthText());
|
|
|
|
// disable Netrc
|
|
option.put(PREF_NO_NETRC, V_TRUE);
|
|
CPPUNIT_ASSERT_EQUAL(std::string("anonymous:ARIA2USER@"),
|
|
factory.createAuthConfig(req, &option)->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, &option)->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, &option)->getAuthText());
|
|
|
|
// username in URI, but no password. We have DefaultAuthenticator
|
|
// but username is not aria2user
|
|
req->setUrl("ftp://aria2user@localhost/download/aria2-1.0.0.tar.bz2");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("aria2user:userDefinedPassword"),
|
|
factory.createAuthConfig(req, &option)->getAuthText());
|
|
|
|
// Recreate netrc with entry for user aria2user
|
|
netrc.reset(new Netrc());
|
|
netrc->addAuthenticator
|
|
(SharedHandle<Authenticator>(new Authenticator("localhost",
|
|
"aria2user",
|
|
"netrcpass",
|
|
"netrcacct")));
|
|
factory.setNetrc(netrc);
|
|
// This time, we can find same username "aria2user" in netrc, so the
|
|
// password "netrcpass" is used, instead of "userDefinedPassword"
|
|
CPPUNIT_ASSERT_EQUAL(std::string("aria2user:netrcpass"),
|
|
factory.createAuthConfig(req, &option)->getAuthText());
|
|
// No netrc entry for host mirror, so "userDefinedPassword" is used.
|
|
req->setUrl("ftp://aria2user@mirror/download/aria2-1.0.0.tar.bz2");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("aria2user:userDefinedPassword"),
|
|
factory.createAuthConfig(req, &option)->getAuthText());
|
|
}
|
|
|
|
void AuthConfigFactoryTest::testUpdateBasicCred()
|
|
{
|
|
Option option;
|
|
option.put(PREF_NO_NETRC, V_FALSE);
|
|
option.put(PREF_HTTP_AUTH_CHALLENGE, V_TRUE);
|
|
|
|
AuthConfigFactory factory;
|
|
|
|
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, &option)->getAuthText());
|
|
|
|
req->setUrl("http://localhost/documents/reference.html");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("alice:ium8"),
|
|
factory.createAuthConfig(req, &option)->getAuthText());
|
|
|
|
req->setUrl("http://localhost/documents2/manual.html");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("myname:mypass"),
|
|
factory.createAuthConfig(req, &option)->getAuthText());
|
|
|
|
req->setUrl("http://localhost/doc/readme.txt");
|
|
CPPUNIT_ASSERT_EQUAL(std::string("myname:mypass"),
|
|
factory.createAuthConfig(req, &option)->getAuthText());
|
|
|
|
req->setUrl("http://local/");
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req, &option).isNull());
|
|
|
|
req->setUrl("http://mirror/");
|
|
CPPUNIT_ASSERT(factory.createAuthConfig(req, &option).isNull());
|
|
}
|
|
|
|
} // namespace aria2
|