|
Packit Service |
a2489d |
/*
|
|
Packit Service |
a2489d |
* lftp - file transfer program
|
|
Packit Service |
a2489d |
*
|
|
Packit Service |
a2489d |
* Copyright (c) 1996-2016 by Alexander V. Lukyanov (lav@yars.free.net)
|
|
Packit Service |
a2489d |
*
|
|
Packit Service |
a2489d |
* This program is free software; you can redistribute it and/or modify
|
|
Packit Service |
a2489d |
* it under the terms of the GNU General Public License as published by
|
|
Packit Service |
a2489d |
* the Free Software Foundation; either version 3 of the License, or
|
|
Packit Service |
a2489d |
* (at your option) any later version.
|
|
Packit Service |
a2489d |
*
|
|
Packit Service |
a2489d |
* This program is distributed in the hope that it will be useful,
|
|
Packit Service |
a2489d |
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit Service |
a2489d |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
Packit Service |
a2489d |
* GNU General Public License for more details.
|
|
Packit Service |
a2489d |
*
|
|
Packit Service |
a2489d |
* You should have received a copy of the GNU General Public License
|
|
Packit Service |
a2489d |
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
Packit Service |
a2489d |
*/
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
#ifndef NETACCESS_H
|
|
Packit Service |
a2489d |
#define NETACCESS_H
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
#include "FileAccess.h"
|
|
Packit Service |
a2489d |
#include "Resolver.h"
|
|
Packit Service |
a2489d |
#include "LsCache.h"
|
|
Packit Service |
a2489d |
#include "RateLimit.h"
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
class NetAccess : public FileAccess, public Networker
|
|
Packit Service |
a2489d |
{
|
|
Packit Service |
a2489d |
protected:
|
|
Packit Service |
a2489d |
class SiteData
|
|
Packit Service |
a2489d |
{
|
|
Packit Service |
a2489d |
int current_connection_limit;
|
|
Packit Service |
a2489d |
int connection_limit;
|
|
Packit Service |
a2489d |
Timer connection_limit_timer;
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
public:
|
|
Packit Service |
a2489d |
SiteData(const xstring &site)
|
|
Packit Service |
a2489d |
: current_connection_limit(0), connection_limit(0),
|
|
Packit Service |
a2489d |
connection_limit_timer("net:connection-limit-timer",site) {}
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
void SetConnectionLimit(int L) {
|
|
Packit Service |
a2489d |
connection_limit=L;
|
|
Packit Service |
a2489d |
if(L && current_connection_limit>=L) {
|
|
Packit Service |
a2489d |
current_connection_limit=L;
|
|
Packit Service |
a2489d |
connection_limit_timer.Stop();
|
|
Packit Service |
a2489d |
}
|
|
Packit Service |
a2489d |
}
|
|
Packit Service |
a2489d |
int GetConnectionLimit() {
|
|
Packit Service |
a2489d |
if(current_connection_limit
|
|
Packit Service |
a2489d |
&& (!connection_limit || connection_limit>current_connection_limit)
|
|
Packit Service |
a2489d |
&& connection_limit_timer.Stopped()) {
|
|
Packit Service |
a2489d |
current_connection_limit++;
|
|
Packit Service |
a2489d |
if(!connection_limit || connection_limit>current_connection_limit)
|
|
Packit Service |
a2489d |
connection_limit_timer.Reset();
|
|
Packit Service |
a2489d |
}
|
|
Packit Service |
a2489d |
return current_connection_limit;
|
|
Packit Service |
a2489d |
}
|
|
Packit Service |
a2489d |
void DecreaseConnectionLimit() {
|
|
Packit Service |
a2489d |
if(current_connection_limit>1) {
|
|
Packit Service |
a2489d |
current_connection_limit--;
|
|
Packit Service |
a2489d |
connection_limit_timer.Reset();
|
|
Packit Service |
a2489d |
}
|
|
Packit Service |
a2489d |
}
|
|
Packit Service |
a2489d |
};
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
static xmap_p<NetAccess::SiteData> site_data;
|
|
Packit Service |
a2489d |
SiteData *GetSiteData() const {
|
|
Packit Service |
a2489d |
const xstring& key=GetConnectURL(NO_PATH);
|
|
Packit Service |
a2489d |
SiteData *data=site_data.lookup(key);
|
|
Packit Service |
a2489d |
if(!data) {
|
|
Packit Service |
a2489d |
data=new SiteData(key);
|
|
Packit Service |
a2489d |
site_data.add(key,data);
|
|
Packit Service |
a2489d |
}
|
|
Packit Service |
a2489d |
data->SetConnectionLimit(connection_limit);
|
|
Packit Service |
a2489d |
return data;
|
|
Packit Service |
a2489d |
}
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
int GetConnectionLimit() {
|
|
Packit Service |
a2489d |
return GetSiteData()->GetConnectionLimit();
|
|
Packit Service |
a2489d |
}
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
SMTaskRef<Resolver> resolver;
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
xarray<sockaddr_u> peer;
|
|
Packit Service |
a2489d |
int peer_curr;
|
|
Packit Service |
a2489d |
void ClearPeer();
|
|
Packit Service |
a2489d |
void NextPeer();
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
int max_persist_retries;
|
|
Packit Service |
a2489d |
int persist_retries;
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
Timer idle_timer;
|
|
Packit Service |
a2489d |
Timer timeout_timer;
|
|
Packit Service |
a2489d |
bool CheckTimeout();
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
int reconnect_interval;
|
|
Packit Service |
a2489d |
float reconnect_interval_current;
|
|
Packit Service |
a2489d |
float reconnect_interval_multiplier;
|
|
Packit Service |
a2489d |
int reconnect_interval_max;
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
int connection_limit;
|
|
Packit Service |
a2489d |
bool connection_takeover;
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
Ref<RateLimit> rate_limit;
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
int socket_buffer;
|
|
Packit Service |
a2489d |
int socket_maxseg;
|
|
Packit Service |
a2489d |
void SetSocketBuffer(int sock) { Networker::SetSocketBuffer(sock,socket_buffer); }
|
|
Packit Service |
a2489d |
void SetSocketMaxseg(int sock) { Networker::SetSocketMaxseg(sock,socket_maxseg); }
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
int SocketCreate(int af,int type,int proto) { return Networker::SocketCreate(af,type,proto,hostname); }
|
|
Packit Service |
a2489d |
int SocketCreateTCP(int af) { return Networker::SocketCreateTCP(af,hostname); }
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
int Poll(int fd,int ev,const char **err);
|
|
Packit Service |
a2489d |
const char *CheckHangup(const struct pollfd *pfd,int num);
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
xstring_c proxy;
|
|
Packit Service |
a2489d |
xstring_c proxy_port;
|
|
Packit Service |
a2489d |
xstring_c proxy_user;
|
|
Packit Service |
a2489d |
xstring_c proxy_pass;
|
|
Packit Service |
a2489d |
xstring_c proxy_proto;
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
xstring_c home_auto;
|
|
Packit Service |
a2489d |
void PropagateHomeAuto();
|
|
Packit Service |
a2489d |
const char *FindHomeAuto();
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
void SayConnectingTo();
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
void SetProxy(const char *);
|
|
Packit Service |
a2489d |
static bool NoProxy(const char *);
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
int Resolve(const char *defp,const char *ser,const char *pr);
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
const char *DelayingMessage();
|
|
Packit Service |
a2489d |
bool ReconnectAllowed();
|
|
Packit Service |
a2489d |
bool CheckRetries(); // returns false if max-retries exceeded.
|
|
Packit Service |
a2489d |
bool NextTry(); // increments retries; does CheckRetries().
|
|
Packit Service |
a2489d |
void TrySuccess(); // reset retry counters.
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
virtual void HandleTimeout();
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
public:
|
|
Packit Service |
a2489d |
void Init();
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
NetAccess();
|
|
Packit Service |
a2489d |
NetAccess(const NetAccess *);
|
|
Packit Service |
a2489d |
~NetAccess();
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
void Reconfig(const char *name=0);
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
void Open(const char *fn,int mode,off_t offs);
|
|
Packit Service |
a2489d |
void ResetLocationData();
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
void Close();
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
void Cleanup();
|
|
Packit Service |
a2489d |
void CleanupThis();
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
int CountConnections();
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
static void ClassInit();
|
|
Packit Service |
a2489d |
static void ClassCleanup() {
|
|
Packit Service |
a2489d |
site_data.empty();
|
|
Packit Service |
a2489d |
}
|
|
Packit Service |
a2489d |
};
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
class GenericParseListInfo : public ListInfo
|
|
Packit Service |
a2489d |
{
|
|
Packit Service |
a2489d |
bool redir_resolution;
|
|
Packit Service |
a2489d |
int redir_count;
|
|
Packit Service |
a2489d |
int max_redir;
|
|
Packit Service |
a2489d |
FileAccessRef redir_session;
|
|
Packit Service |
a2489d |
Ref<FileSet> redir_fs;
|
|
Packit Service |
a2489d |
bool ResolveRedirect(const FileInfo *fi);
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
protected:
|
|
Packit Service |
a2489d |
int mode;
|
|
Packit Service |
a2489d |
SMTaskRef<IOBuffer> ubuf;
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
bool get_time_for_dirs;
|
|
Packit Service |
a2489d |
bool can_get_prec_time;
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
virtual FileSet *Parse(const char *buf,int len)
|
|
Packit Service |
a2489d |
{ return session->ParseLongList(buf,len); }
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
public:
|
|
Packit Service |
a2489d |
GenericParseListInfo(FileAccess *session,const char *path);
|
|
Packit Service |
a2489d |
int Do();
|
|
Packit Service |
a2489d |
const char *Status();
|
|
Packit Service |
a2489d |
};
|
|
Packit Service |
a2489d |
|
|
Packit Service |
a2489d |
#endif//NETACCESS_H
|