Skip to content
connection.hh 1.97 KiB
Newer Older
/*
  This file is part of SLS.
  Copyright (C) 2008 Sebastien LUTTRINGER <contact@seblu.net>

  SLS 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; version 2 of the License.

  SLS 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 SLS; if not, write to the Free Software
  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

#ifndef CONNECTION_HH
# define CONNECTION_HH

# include <pthread.h>

class Connection {
  // public methods
public:
  Connection(int fd = -1);
  virtual ~Connection();

  void connect(const char *addr, int port);
  void listen(int port, int max);

  void disconnect();
  bool connected();

  Connection *accept();

  // send methods
  inline void send(const string &data);
  void send(const char* buf, size_t len);
  inline void sendln(const string &data);
  void sendln(const char* buf, size_t len);

  // recv methods
  string recvln();

  // info methods
  string getlocalip();
  int getlocalport();

  string getremoteip();
  int getremoteport();

  int getsocket();

  // protected methods
protected:
  void socket_();
  void disconnect_();

  inline void setallinfo_();
  void setlocalip_();
  void setlocalport_();
  void setremoteip_();
  void setremoteport_();

  // Protected datas
protected:
  int socket_fd_; // connection socket

  // storage of info about connection
  int local_port_;
  int remote_port_;

  string local_ip_;
  string remote_ip_;

  string rbuf_; // read buffer
  string wbuf_; // write buffer

  pthread_mutex_t c_mutex_; // connection mutex
  pthread_mutex_t r_mutex_; // read mutex
  pthread_mutex_t w_mutex_; // write mutex
};

# include "connection.hxx"

#endif