/*
 * lftp - file transfer program
 *
 * Copyright (c) 1996-2016 by Alexander V. Lukyanov (lav@yars.free.net)
 *
 * 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; either version 3 of the License, or
 * (at your option) any later version.
 *
 * 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.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef NETACCESS_H
#define NETACCESS_H

#include "FileAccess.h"
#include "Resolver.h"
#include "LsCache.h"
#include "RateLimit.h"

class NetAccess : public FileAccess, public Networker
{
protected:
   class SiteData
   {
      int current_connection_limit;
      int connection_limit;
      Timer connection_limit_timer;

   public:
      SiteData(const xstring &site)
	 : current_connection_limit(0), connection_limit(0),
	   connection_limit_timer("net:connection-limit-timer",site) {}

      void SetConnectionLimit(int L) {
	 connection_limit=L;
	 if(L && current_connection_limit>=L) {
	    current_connection_limit=L;
	    connection_limit_timer.Stop();
	 }
      }
      int GetConnectionLimit() {
	 if(current_connection_limit
	 && (!connection_limit || connection_limit>current_connection_limit)
	 && connection_limit_timer.Stopped()) {
	    current_connection_limit++;
	    if(!connection_limit || connection_limit>current_connection_limit)
	       connection_limit_timer.Reset();
	 }
	 return current_connection_limit;
      }
      void DecreaseConnectionLimit() {
	 if(current_connection_limit>1) {
	    current_connection_limit--;
	    connection_limit_timer.Reset();
	 }
      }
   };

   static xmap_p<NetAccess::SiteData> site_data;
   SiteData *GetSiteData() const {
      const xstring& key=GetConnectURL(NO_PATH);
      SiteData *data=site_data.lookup(key);
      if(!data) {
	 data=new SiteData(key);
	 site_data.add(key,data);
      }
      data->SetConnectionLimit(connection_limit);
      return data;
   }

   int GetConnectionLimit() {
      return GetSiteData()->GetConnectionLimit();
   }

   SMTaskRef<Resolver> resolver;

   xarray<sockaddr_u> peer;
   int peer_curr;
   void	 ClearPeer();
   void	 NextPeer();

   int	 max_persist_retries;
   int	 persist_retries;

   Timer idle_timer;
   Timer timeout_timer;
   bool	 CheckTimeout();

   int	 reconnect_interval;
   float reconnect_interval_current;
   float reconnect_interval_multiplier;
   int   reconnect_interval_max;

   int	 connection_limit;
   bool	 connection_takeover;

   Ref<RateLimit> rate_limit;

   int	 socket_buffer;
   int	 socket_maxseg;
   void	 SetSocketBuffer(int sock) { Networker::SetSocketBuffer(sock,socket_buffer); }
   void	 SetSocketMaxseg(int sock) { Networker::SetSocketMaxseg(sock,socket_maxseg); }

   int SocketCreate(int af,int type,int proto) { return Networker::SocketCreate(af,type,proto,hostname); }
   int SocketCreateTCP(int af) { return Networker::SocketCreateTCP(af,hostname); }

   int Poll(int fd,int ev,const char **err);
   const char *CheckHangup(const struct pollfd *pfd,int num);

   xstring_c proxy;
   xstring_c proxy_port;
   xstring_c proxy_user;
   xstring_c proxy_pass;
   xstring_c proxy_proto;

   xstring_c home_auto;
   void	 PropagateHomeAuto();
   const char *FindHomeAuto();

   void SayConnectingTo();

   void SetProxy(const char *);
   static bool NoProxy(const char *);

   int Resolve(const char *defp,const char *ser,const char *pr);

   const char *DelayingMessage();
   bool ReconnectAllowed();
   bool CheckRetries();	// returns false if max-retries exceeded.
   bool NextTry();	// increments retries; does CheckRetries().
   void TrySuccess();	// reset retry counters.

   virtual void HandleTimeout();

public:
   void Init();

   NetAccess();
   NetAccess(const NetAccess *);
   ~NetAccess();

   void Reconfig(const char *name=0);

   void Open(const char *fn,int mode,off_t offs);
   void ResetLocationData();

   void Close();

   void Cleanup();
   void CleanupThis();

   int CountConnections();

   static void ClassInit();
   static void ClassCleanup() {
      site_data.empty();
   }
};

class GenericParseListInfo : public ListInfo
{
   bool redir_resolution;
   int redir_count;
   int max_redir;
   FileAccessRef redir_session;
   Ref<FileSet> redir_fs;
   bool ResolveRedirect(const FileInfo *fi);

protected:
   int mode;
   SMTaskRef<IOBuffer> ubuf;

   bool get_time_for_dirs;
   bool can_get_prec_time;

   virtual FileSet *Parse(const char *buf,int len)
      { return session->ParseLongList(buf,len); }

public:
   GenericParseListInfo(FileAccess *session,const char *path);
   int Do();
   const char *Status();
};

#endif//NETACCESS_H