/******************************************************************************
* Copyright (c) 2000-2019 Ericsson Telecom AB
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v2.0
* which accompanies this distribution, and is available at
* https://www.eclipse.org/org/documents/epl-2.0/EPL-2.0.html
*
* Contributors:
* Tibor Csondes - initial implementation and initial documentation
* Akos Cserveni
* Elemer Lelik
* Gabor Szalai
* Gabor Tatarka
* Gergely Futo
* Istvan Sandor
* Peter Dimitrov
* Peter Kremer
* Tibor Bende
* Zoltan Medve
******************************************************************************/
//
// File: TELNETasp_PT.hh
// Description: TELNET testport header file
// Rev: R9A
// Prodnr: CNL 113 320
//
#ifndef TELNETasp__PT_HH
#define TELNETasp__PT_HH
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <errno.h>
#include <unistd.h>
#include <netdb.h>
#include <string.h>
#include <netinet/in.h>
#include <stdlib.h>
#include <string.h>
#include <regex.h>
#include <fcntl.h>
#include <memory.h>
#include "TELNETasp_PortType.hh"
#define BUFFER_SIZE (100*1024)
#define READMODE_UNBUFFERED 0
#define READMODE_BUFFERED 1
#define WINDOW_SIZE_LENGTH 9 // IAC SB NAWS x x y y IAC SE
namespace TELNETasp__PortType {
//#define DEBUG_ProcessCmd 1
//extern INTEGER FindString(const CHARSTRING& msg_ch, const CHARSTRING& pattern_ch);
typedef enum {
WSS_INIT,
WSS_WILL_SENT,
WSS_AGREED,
WSS_REFUSED,
WSS_NOT_SET
} WindowSize_state_t;
class Window_Size {
WindowSize_state_t my_state;
int width, height;
char *port_name;
bool ws_debugAllowed;
public:
Window_Size(const char *p_port_name);
~Window_Size();
inline void reset() { my_state = WSS_NOT_SET; width = -1; height = -1; ws_debugAllowed = false; }
void set_state(WindowSize_state_t new_state);
inline WindowSize_state_t get_state() const { return my_state; }
void set_width(int w);
inline int get_width() const { return width; }
void set_height(int h);
inline int get_height() const { return height; }
const unsigned char *encode() const;
inline bool is_set() const { return width > 0 && height > 0; }
void log_debug(const char *fmt, ...);
void set_debug(bool debug_enabled);
};
typedef struct {
bool match;
unsigned int start;
unsigned int end;
} Regex_Prompt_MatchResult;
class Regex_Prompt {
regex_t posix_regexp;
bool rp_debugAllowed;
public:
Regex_Prompt(const char *p_pattern, bool debug_enabled, bool raw_prompt);
~Regex_Prompt();
Regex_Prompt_MatchResult match(const char *msg);
void log_debug(const char *fmt, ...);
};
class Prompt_List {
typedef struct {
unsigned int id;
bool is_regex;
union {
char *prompt;
Regex_Prompt *regex_prompt;
};
} prompt_elem;
prompt_elem **elems;
size_t n_elems;
bool id_to_index(unsigned int p_id, size_t &index) const;
bool pl_debugAllowed;
public:
Prompt_List() : elems(NULL), n_elems(0), pl_debugAllowed(false) { }
~Prompt_List();
void set_prompt(unsigned int p_id, const char *p_prompt, bool p_is_regex, bool raw_prompt = false);
void check(const char *warning_prefix);
void clear();
size_t nof_prompts() const { return n_elems; }
int findPrompt(size_t &pos, const unsigned char *bufptr, size_t buflen) const;
void log_debug(const char *fmt, ...);
void set_debug(bool debug_enabled);
};
class TELNETasp__PT : public TELNETasp__PT_BASE {
public:
TELNETasp__PT(const char *par_port_name = NULL);
~TELNETasp__PT();
void set_parameter(const char *parameter_name,
const char *parameter_value);
void Handle_Fd_Event_Readable(int fd);
/* void Event_Handler(const fd_set *read_fds,
const fd_set *write_fds, const fd_set *error_fds,
double time_since_last_call);
*/
protected:
void user_map(const char *system_port);
void user_unmap(const char *system_port);
void user_start();
void user_stop();
void outgoing_send(const CHARSTRING& send_par);
void outgoing_send(const ASP__TelnetPortParameters& send_par);
void outgoing_send(const ASP__TelnetDynamicConfig& send_par);
void outgoing_send(const ASP__TelnetConnection& send_par);
void outgoing_send(const ASP__TelnetClose& send_par);
void outgoing_send(const ASP__Get__portnum&);
void InitStrPar(char *&par, const char *name, const char *val);
int ProcessCmd(unsigned char * buf, int buflen);
int ProcessNoCmd(unsigned char * buf, int buflen);
void LoginCtrl( const char* buf, const char* nam, const char* pw );
int isPrompt(size_t& pos); // returns the length of the found prompt
int RecvClrMsg();
void log_buffer(const char * logmsg,
const unsigned char * buf, size_t buflen);
void log_debug(const char *fmt, ...);
bool buf_strcmp(const char * s1, const unsigned char * s2,
size_t s2_len, size_t& pos);
int send_nonblock(int sockfd, const char *buf, size_t len, int flags);
// Class variables and constants
int nvtsock;
bool isConnected;
bool isLoginSent;
bool isPasswordSent;
bool isDomainSent;
bool isRawRegexpPrompt;
CHARSTRING echobuf;
TTCN_Buffer ttcn_buf;
struct hostent *he;
bool config_finished;
int map_poll_timeout;
// parameters
char* ctrl_hostname;
int ctrl_portnum;
char* ctrl_username;
char* ctrl_password;
char* ctrl_domain;
int ctrl_readmode;
bool ctrl_login_skipped;
bool ctrl_detect_server_disconnected;
bool ctrl_detect_connection_establishment_result;
bool ctrl_client_cleanup_linefeed;
char* ctrl_terminal_type;
bool ctrl_echo;
bool ctrl_CRLF;
Prompt_List prompt_list;
Window_Size window_size;
int dont_close_fd;
char* ctrl_username_client;
char* ctrl_password_client;
char* ctrl_server_prompt;
char* ctrl_password_prompt;
char* ctrl_loginname_prompt;
bool server_mode_def;
bool ctrl_server_attach_prompt;
bool ctrl_detect_client_disconnected;
bool ctrl_server_failsafe_sending;
bool server_mode;
bool port_mapped;
int fd_server;
bool isClientLoggedIn;
bool pass_prompt_send;
bool login_incorrect;
bool debugAllowed;
bool emptyEcho;
bool suppressed;
void recv_msg_from_client(int& fd);
void close_connection(int& fd);
bool check(const char * s1, const unsigned char * s2, int s2_len);
// fd_set readfds;
ASP__TelnetPortParameters *asp_params;
void set_asp_params();
void reset_configuration();
};
}
#endif