projects/rlm_stg/stg_client.h

   1 /*
   2  *    This program is free software; you can redistribute it and/or modify
   3  *    it under the terms of the GNU General Public License as published by
   4  *    the Free Software Foundation; either version 2 of the License, or
   5  *    (at your option) any later version.
   6  *
   7  *    This program is distributed in the hope that it will be useful,
   8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
   9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  10  *    GNU General Public License for more details.
  11  *
  12  *    You should have received a copy of the GNU General Public License
  13  *    along with this program; if not, write to the Free Software
  14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  15  */
  16
  17 /*
  18  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
  19  */
  20
  21 /*
  22  *  Header file for client part of data access via Stargazer for RADIUS
  23  *
  24  *  $Revision: 1.4 $
  25  *  $Date: 2010/04/16 12:30:02 $
  26  *
  27  */
  28
  29 #ifndef STG_CLIENT_H
  30 #define STG_CLIENT_H
  31
  32 #include <string>
  33
  34 #include <netinet/in.h>
  35 #include <arpa/inet.h>
  36 #include <sys/socket.h> // socklen_t
  37
  38 #include "blowfish.h"
  39 #include "rad_packets.h"
  40
  41 class STG_CLIENT
  42 {
  43 public:
  44     STG_CLIENT();
  45     ~STG_CLIENT();
  46
  47     void SetServer(const std::string & host);
  48     void SetPort(uint16_t port);
  49     void SetLocalPort(uint16_t port);
  50     void SetPassword(const std::string & password);
  51
  52     int Start();
  53     int Stop();
  54
  55     std::string GetUserPassword() const;
  56
  57     int Authorize(const std::string & login, const std::string & svc);
  58     int Authenticate(const std::string & login, const std::string & svc);
  59     int PostAuthenticate(const std::string & login, const std::string & svc);
  60     int Account(const std::string & type, const std::string & login, const std::string & svc, const std::string & sessid);
  61
  62     uint32_t GetFramedIP() const;
  63
  64
  65     const std::string & GetError() const { return errorStr; };
  66
  67 private:
  68     std::string host;
  69     uint16_t port;
  70     uint16_t localPort;
  71     std::string password;
  72     int sock;
  73     std::string errorStr;
  74
  75     struct sockaddr_in outerAddr;
  76     socklen_t outerAddrLen;
  77
  78     std::string userPassword;
  79
  80     uint32_t framedIP;
  81
  82     BLOWFISH_CTX ctx;
  83
  84     int PrepareNet();
  85     void FinalizeNet();
  86
  87     void InitEncrypt();
  88     void Encrypt(char * dst, const char * src, int len8);
  89     void Decrypt(char * dst, const char * src, int len8);
  90
  91     int Request(RAD_PACKET * packet, const std::string & login, const std::string & svc, uint8_t packetType);
  92
  93     int RecvData(RAD_PACKET * packet);
  94     int Send(const RAD_PACKET & packet);
  95
  96 };
  97
  98 #endif