sems/apps/diameter_client/ServerConnection.h

/*
 * Copyright (C) 2007-2009 IPTEGO GmbH
 *
 * This file is part of SEMS, a free SIP media server.
 *
 * SEMS 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 2 of the License, or
 * (at your option) any later version. This program is released under
 * the GPL with the additional exemption that compiling, linking,
 * and/or using OpenSSL is allowed.
 *
 * For a license to use the SEMS software under conditions
 * other than those described here, or to purchase support for this
 * software, please contact iptel.org by e-mail at the following addresses:
 *    info@iptel.org
 *
 * SEMS 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, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#ifndef _DIAMETER_SERVER_CONNECTION_H
#define _DIAMETER_SERVER_CONNECTION_H
#include <stdint.h>

#include "diameter_client.h"

#include "AmThread.h"
#include "AmEventQueue.h"
#include "AmArg.h"

#include <string>
#include <vector>
#include <map>
#include <utility>
#include <stdint.h>
using std::string;
using std::vector;
using std::map;
using std::pair;

#define AF_IP4 1 // http://www.iana.org/assignments/address-family-numbers

enum {
  AAA_ERROR_NOTINIT    = -1,
  AAA_ERROR_NOCONN     = -2,
  AAA_ERROR_NOFREECONN = -3,
  AAA_ERROR_TIMEOUT    = -4,
  AAA_ERROR_MESSAGE    = -5,
  AAA_ERROR_COMM       = -6
};

struct DiameterRequestEvent 
  : public AmEvent
{
  int command_code;
  int app_id;
  AmArg val;
  string sess_link;

  enum { ID_NewRequest = 0 }; 
  DiameterRequestEvent(int command_code,
		       int app_id,
		       AmArg val,
		       string sess_link)
    : AmEvent(ID_NewRequest), command_code(command_code),
    app_id(app_id), val(val), sess_link(sess_link)
  {
  }
};

struct DiameterServerConnection {
  bool in_use;
  dia_tcp_conn* dia_conn; // transport connection
  rd_buf_t rb;

  string origin_host;

  AAAMsgIdentifier h2h;
  AAAMsgIdentifier e2e;
  void terminate(bool tls_shutdown = false);

  void setIDs(AAAMessage* msg);

  DiameterServerConnection();
  ~DiameterServerConnection() {}
};

class ServerConnection 
: public AmEventQueue,
  public AmThread,
  public AmEventHandler
{
  struct timeval connect_ts;
  bool open;
  int timeout_check_cntr;
  
  string server_name;
  int server_port;

  // openssl
  string ca_file;
  string cert_file;

  string origin_host;
  string origin_realm;
  string origin_ip;
  AAAApplicationId app_id;

  int request_timeout; // millisec

  char origin_ip_address[2+4];// AF and address

  //  the client
  string product_name;
  uint32_t vendorID;
  
  DiameterServerConnection conn;

  typedef map<unsigned int, pair<string, struct timeval> > DReqMap;
  DReqMap req_map;
  AmMutex req_map_mut;

  void openConnection();
  void closeConnection(bool tls_shutdown = false);

  int addOrigin(AAAMessage* msg);

  int handleReply(AAAMessage* rep);
  int handleRequest(AAAMessage* req);
  AAAMessage* ReqEvent2AAAMessage(DiameterRequestEvent* re);
  AmArg AAAMessageAVPs2AmArg(AAAMessage* rep);
  int AAAMessageGetReplyCode(AAAMessage* rep);

  static int addStringAVP(AAAMessage* msg, AAA_AVPCode avp_code, string& val, bool attail = false);
  static int addDataAVP(AAAMessage* msg, AAA_AVPCode avp_code, char* val, unsigned int len);
  static int addResultCodeAVP(AAAMessage* msg, AAAResultCode code);
  static int addGroupedAVP(AAA_AVP *avp, AAA_AVPCode avp_code, char* val, unsigned int len);

  // send request and get response  
  int sendRequest(AAAMessage* req, unsigned int& exe);
  void process(AmEvent*);
  void receive();
  void setRetryConnectLater();
  void checkTimeouts();
  void shutdownConnection();

 public:
  ServerConnection();
  ~ServerConnection();
  int init(const string& _server_name, 
	   int _server_port,
	   const string& _ca_file,
	   const string& _cert_file,
	   const string& _origin_host, 
	   const string& _origin_realm,
	   const string& _origin_ip,
	   AAAApplicationId _app_id,
	   unsigned int _vendorID,
	   const string& _product_name,
	   int _request_timeout); // millisec

  bool is_open() { return open; }
  void run();
  void on_stop();
};

#endif