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(const std::string & host, uint16_t port, uint16_t lp, const std::string & pass);
  45     ~STG_CLIENT();
  46
  47     int Start();
  48     int Stop();
  49
  50     std::string GetUserPassword() const;
  51
  52     int Authorize(const std::string & login, const std::string & svc);
  53     int Authenticate(const std::string & login, const std::string & svc);
  54     int PostAuthenticate(const std::string & login, const std::string & svc);
  55     int Account(const std::string & type, const std::string & login, const std::string & svc, const std::string & sessid);
  56
  57     uint32_t GetFramedIP() const;
  58
  59     const std::string & GetError() const { return errorStr; };
  60
  61 private:
  62     uint16_t localPort;
  63     std::string password;
  64     int sock;
  65     std::string errorStr;
  66
  67     struct sockaddr_in outerAddr;
  68
  69     std::string userPassword;
  70
  71     uint32_t framedIP;
  72
  73     BLOWFISH_CTX ctx;
  74
  75     int PrepareNet();
  76
  77     void InitEncrypt();
  78     void Encrypt(char * dst, const char * src, int len8);
  79     void Decrypt(char * dst, const char * src, int len8);
  80
  81     int Request(RAD_PACKET * packet, const std::string & login, const std::string & svc, uint8_t packetType);
  82
  83     int RecvData(RAD_PACKET * packet);
  84     int Send(const RAD_PACKET & packet);
  85 };
  86
  87 #endif