Software Público Brasileiro

/*
 * "Software pw3270, desenvolvido com base nos códigos fontes do WC3270  e X3270
 * (Paul Mattes Paul.Mattes@usa.net), de emulação de terminal 3270 para acesso a
 * aplicativos mainframe. Registro no INPI sob o nome G3270.
 *
 * Copyright (C) <2008> <Banco do Brasil S.A.>
 *
 * Este programa é software livre. Você pode redistribuí-lo e/ou modificá-lo sob
 * os termos da GPL v.2 - Licença Pública Geral  GNU,  conforme  publicado  pela
 * Free Software Foundation.
 *
 * Este programa é distribuído na expectativa de  ser  útil,  mas  SEM  QUALQUER
 * GARANTIA; sem mesmo a garantia implícita de COMERCIALIZAÇÃO ou  de  ADEQUAÇÃO
 * A QUALQUER PROPÓSITO EM PARTICULAR. Consulte a Licença Pública Geral GNU para
 * obter mais detalhes.
 *
 * Você deve ter recebido uma cópia da Licença Pública Geral GNU junto com este
 * programa; se não, escreva para a Free Software Foundation, Inc., 51 Franklin
 * St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * Este programa está nomeado como - e possui - linhas de código.
 *
 * Contatos:
 *
 * perry.werneck@gmail.com	(Alexandre Perry de Souza Werneck)
 * erico.mendonca@gmail.com	(Erico Mascarenhas Mendonça)
 *
 */

/**
 * @file src/session/local/session.cc
 *
 * @brief Implement lib3270 direct access layout (NO IPC).
 *
 * @author perry.werneck@gmail.com
 *
 */

 #ifdef _WIN32

	#include <winsock2.h>
	#include <windows.h>
	#include <lmcons.h>
	#include <libloaderapi.h>

 #endif // _WIN32

 #include <ipc-client-internals.h>
 #include <lib3270/actions.h>
 #include <lib3270/properties.h>
 #include <lib3270/toggle.h>
 #include <iostream>
 #include <cstring>

 extern "C" {
	 #include <lib3270/session.h>


 }

 using std::string;

/*---[ Implement ]----------------------------------------------------------------------------------*/

 namespace TN3270 {

 #ifdef _WIN32
	static void write_log(const char *msg, int rc = (int) GetLastError()) {

		HANDLE hEventLog = RegisterEventSource(NULL, PACKAGE_NAME);

		debug(msg," rc=",rc);

		if(hEventLog) {

			char	username[UNLEN + 1];
			DWORD	szName = sizeof(username);

			memset(username,0,szName);

			if(!GetUserName(username, &szName)) {
				strncpy(username,"?",UNLEN);
			}

			char lasterror[1024];
			snprintf(lasterror,sizeof(lasterror),"The error code was %d",rc);

			const char *outMsg[] = {
				username,
				msg,
				lasterror
			};

			ReportEvent(
				hEventLog,
				EVENTLOG_ERROR_TYPE,
				1,
				0,
				NULL,
				3,
				0,
				outMsg,
				NULL
			);

			DeregisterEventSource(hEventLog);

		}

	}

	static HKEY openKey() {

		LSTATUS rc;
		HKEY hKey = 0;

		rc = RegOpenKeyEx(HKEY_LOCAL_MACHINE,"Software\\pw3270",0,KEY_QUERY_VALUE,&hKey);
		if(rc == ERROR_SUCCESS) {
			return hKey;
		}

		write_log("Can't open HKLM\\Software\\pw3270", (int) rc);

		rc = RegOpenKeyEx(HKEY_LOCAL_MACHINE,"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\pw3270",0,KEY_QUERY_VALUE,&hKey);
		if(rc == ERROR_SUCCESS) {
			return hKey;
		}

		write_log("Can't open HKLM\\Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\pw3270", (int) rc);

		return 0;
	}

 #endif // _WIN32

	Local::Session::Session() : Abstract::Session() {

		std::lock_guard<std::mutex> lock(sync);

#ifdef _WIN32
		{
			static bool initialized = false;

			if(!initialized) {

				// Get application DATADIR
				LSTATUS rc;

				char datadir[4096];
				HKEY hKey = openKey();
				unsigned long datalen = sizeof(datadir);

				memset(datadir,0,sizeof(datadir));

				if(hKey) {

					unsigned long datatype; // #defined in winnt.h (predefined types 0-11)

					rc = RegQueryValueExA(hKey,"InstallLocation",NULL,&datatype,(LPBYTE) datadir,&datalen);
					if(rc != ERROR_SUCCESS) {

						// Can't get DATADIR
						write_log("Can't get Install Location", (int) rc);

						*datadir = 0;
					}
					RegCloseKey(hKey);

				}

#ifdef DEBUG
				debug("Datadir=\"",datadir,"\"");
#endif // DEBUG

				if(*datadir) {

					wchar_t	*path = (wchar_t *) malloc(sizeof(datadir)*sizeof(wchar_t));
					mbstowcs(path, datadir, 4095);

					if(!SetDefaultDllDirectories(LOAD_LIBRARY_SEARCH_USER_DIRS)) {
						write_log("SetDefaultDllDirectories has failed");
					}
#ifdef DEBUG
					else {
						debug("SetDefaultDllDirectories has suceeded");
					}
#endif

					if(!AddDllDirectory(path)) {
						string msg = "Can't add ";
						msg += datadir;
						msg += " to directory path";
						write_log(msg.c_str());
					}
#ifdef DEBUG
					else {
						debug("AddDllDirectory has suceeded");
					}
#endif
					free(path);

				}

				initialized = true;

			}

		}
#endif // _WIN32

		this->hSession = lib3270_session_new("");

		lib3270_set_user_data(this->hSession,(void *) this);
		setCharSet(lib3270_get_display_charset(this->hSession));

		lib3270_set_popup_handler(this->hSession, popupHandler);

		// Setup callbacks
		struct lib3270_session_callbacks *cbk;

		cbk = lib3270_get_session_callbacks(this->hSession,sizeof(struct lib3270_session_callbacks));
		if(!cbk) {
			throw std::runtime_error( _("Invalid callback table, possible version mismatch in lib3270") );
		}

		cbk->update_connect	= connectHandler;


	}

	Local::Session::~Session() {

		std::lock_guard<std::mutex> lock(sync);

		lib3270_session_free(this->hSession);
		this->hSession = nullptr;
	}

	void Local::Session::chkResponse(int rc) {

		if(rc == 0)
			return;

#ifdef _WIN32
		if(rc == ENOTCONN)
			throw std::runtime_error("Not connected");
#endif // _WIN32

		throw std::system_error(rc, std::system_category());

	}

	void Local::Session::wait(time_t timeout) {

		std::lock_guard<std::mutex> lock(sync);
		chkResponse(lib3270_wait(this->hSession, timeout));

	}

	void Local::Session::connect(const char *url) {
		std::lock_guard<std::mutex> lock(sync);
		chkResponse(lib3270_connect_url(hSession,url,0));
    }

	void Local::Session::disconnect() {
		std::lock_guard<std::mutex> lock(sync);
		lib3270_disconnect(hSession);
	}

	// Wait for session state.
	void Local::Session::waitForReady(time_t timeout) throw() {
		std::lock_guard<std::mutex> lock(sync);
		chkResponse(lib3270_wait_for_ready(hSession,timeout));
	}

	std::string	Local::Session::toString(int baddr, size_t len, char lf) const {

		std::lock_guard<std::mutex> lock(const_cast<Local::Session *>(this)->sync);

        char * text = lib3270_get_string_at_address(hSession, baddr, len, lf);

        if(!text) {
            throw std::runtime_error( _("Can't get screen contents") );
        }

        string rc = convertFromHost(text);

        lib3270_free(text);

		return rc;

	}

	std::string	Local::Session::toString(int row, int col, size_t sz, char lf) const {

		std::lock_guard<std::mutex> lock(const_cast<Local::Session *>(this)->sync);

        char * text = lib3270_get_string_at(hSession, row, col, sz, lf);

        if(!text) {
            throw std::runtime_error( _("Can't get screen contents") );
        }

        string rc = convertFromHost(text);

        lib3270_free(text);

		return rc;
	}

	void Local::Session::getProperty(const char *name, int &value) const {

		const LIB3270_INT_PROPERTY * intprop = lib3270_get_int_properties_list();
		for(size_t ix = 0; intprop[ix].name; ix++) {

			if(!strcasecmp(name,intprop[ix].name)) {

				std::lock_guard<std::mutex> lock(const_cast<Local::Session *>(this)->sync);

				value = intprop[ix].get(hSession);

				if(value < 0 && errno != 0) {
					throw std::system_error(errno, std::system_category());
				}


			}

		}

		throw std::system_error(ENOENT, std::system_category());

	}

	void Local::Session::getProperty(const char *name, std::string &value) const {

		const LIB3270_STRING_PROPERTY * strprop = lib3270_get_string_properties_list();

		for(size_t ix = 0; strprop[ix].name; ix++) {

			if(!strcasecmp(name,strprop[ix].name)) {

				std::lock_guard<std::mutex> lock(const_cast<Local::Session *>(this)->sync);

				// Found it!
				const char * str = strprop[ix].get(hSession);

				if(str) {
					value.assign(str);
					return;
				}

				throw std::system_error(errno, std::system_category());

			}

		}

		throw std::system_error(ENOENT, std::system_category());
	}

	void Local::Session::getProperty(const char *name, bool &value) const {

		LIB3270_TOGGLE toggle = lib3270_get_toggle_id(name);
		if(toggle != (LIB3270_TOGGLE) -1) {

			// Is a Tn3270 toggle, get it!
			std::lock_guard<std::mutex> lock(const_cast<Local::Session *>(this)->sync);
			value = lib3270_get_toggle(hSession,toggle);

		}

		throw std::system_error(ENOENT, std::system_category());
	}

	ProgramMessage Local::Session::getProgramMessage() const {
		std::lock_guard<std::mutex> lock(const_cast<Local::Session *>(this)->sync);
		return (ProgramMessage) lib3270_get_program_message(this->hSession);
	}

	ConnectionState Local::Session::getConnectionState() const {
		std::lock_guard<std::mutex> lock(const_cast<Local::Session *>(this)->sync);
		return (ConnectionState) lib3270_get_connection_state(this->hSession);
	}

	/// @brief Set field at current position, jumps to next writable field.
	TN3270::Session & Local::Session::push(const char *text) {
		std::lock_guard<std::mutex> lock(sync);

		string converted = convertToHost(text);

		chkResponse(lib3270_input_string(this->hSession, (unsigned char *) converted.c_str(), converted.size()));

		return *this;
	}

	TN3270::Session & Local::Session::input(const char *text, size_t length) {
		std::lock_guard<std::mutex> lock(sync);

		string converted = convertToHost(text,length);

		chkResponse(lib3270_input_string(this->hSession, (unsigned char *) converted.c_str(), converted.size()));

		return *this;
	}

	TN3270::Session & Local::Session::push(int baddr, const char * text, int length) {
		std::lock_guard<std::mutex> lock(sync);

		string converted = convertToHost(text, length);

		if(lib3270_set_string_at_address(this->hSession,baddr,(unsigned char *) converted.c_str(),converted.length()) < 0) {
			throw std::system_error(errno, std::system_category());
		}

		return *this;
	}

	TN3270::Session & Local::Session::push(int row, int col, const char *text, int length) {
		std::lock_guard<std::mutex> lock(sync);

		string converted = convertToHost(text,length);

		if(lib3270_set_string_at(this->hSession,row,col,(unsigned char *) converted.c_str())) {
			throw std::system_error(errno, std::system_category());
		}

		return *this;
	}

	TN3270::Session & Local::Session::push(const Action action) {

		typedef int (*ActionCallback)(H3270 *);

		static const ActionCallback actions[] = {
            lib3270_enter,			// ENTER
            lib3270_erase,			// ERASE
            lib3270_eraseeof,		// ERASE_EOF
            lib3270_eraseeol,		// ERASE_EOL
            lib3270_eraseinput,		// ERASE_INPUT
            lib3270_kybdreset,		// KYBD_RESET
            lib3270_newline,		// NEWLINE
            lib3270_clear,			// CLEAR
            lib3270_select_field,	// SELECT_FIELD
            lib3270_select_all,		// SELECT_ALL
            lib3270_unselect,		// UNSELECT
            lib3270_reselect,		// RESELECT
            lib3270_delete,			// DELETE
            lib3270_dup,			// DUP
            lib3270_fieldmark,		// FIELDMARK
            lib3270_backspace,		// BACKSPACE
            lib3270_previousword,	// WORD_PREVIOUS
            lib3270_nextword,		// WORD_NEXT
            lib3270_fieldend,		// FIELD_END
            lib3270_firstfield,		// FIELD_FIRST
            lib3270_nextfield,		// FIELD_NEXT
            lib3270_previousfield,	// FIELD_PREVIOUS
            lib3270_attn,			// ATTN
            lib3270_break,			// BREAK
            lib3270_deleteword,		// WORD_DELETE
            lib3270_deletefield,	// FIELD_DELETE
            lib3270_sysreq,			// SYSREQ
		};

		if( ((size_t) action) > (sizeof(actions)/sizeof(actions[0]))) {
            throw std::system_error(EINVAL, std::system_category());
		}

		std::lock_guard<std::mutex> lock(sync);

		chkResponse(actions[(size_t) action](hSession));

		return *this;
	}

	TN3270::Session & Local::Session::pfkey(unsigned short value) {

		std::lock_guard<std::mutex> lock(sync);

		chkResponse(lib3270_pfkey(hSession,(int) value));

		return *this;
	}

	TN3270::Session & Local::Session::pakey(unsigned short value) {

		std::lock_guard<std::mutex> lock(sync);

		chkResponse(lib3270_pakey(hSession,(int) value));

		return *this;
	}


	TN3270::Session & Local::Session::pop(int baddr, std::string &text) {

		std::lock_guard<std::mutex> lock(sync);

		if(!lib3270_is_connected(hSession)) {
			throw std::system_error(ENOTCONN, std::system_category());
		}

		char *contents = lib3270_get_field_text_at(hSession, baddr);

		if(!contents) {
			throw std::system_error(errno, std::system_category());
		}

		text.assign(convertFromHost(contents).c_str());

		lib3270_free(contents);

		return *this;
	}

	TN3270::Session & Local::Session::pop(int row, int col, std::string &text) {
		return this->pop(lib3270_translate_to_address(hSession,row,col),text);
	}

	TN3270::Session & Local::Session::pop(std::string &text) {

		std::lock_guard<std::mutex> lock(sync);

		if(!lib3270_is_connected(hSession)) {
			throw std::system_error(ENOTCONN, std::system_category());
		}

		int baddr = lib3270_get_cursor_address(hSession);
		if(baddr < 0) {
			throw std::system_error(errno, std::system_category());
		}

		char *contents = lib3270_get_field_text_at(hSession, baddr);

		if(!contents) {
			throw std::system_error(errno, std::system_category());
		}

		text.assign(convertFromHost(contents).c_str());

		lib3270_free(contents);

		baddr = lib3270_get_next_unprotected(hSession,baddr);
		if(!baddr) {
			baddr = lib3270_get_next_unprotected(hSession,0);
		}

		if(lib3270_set_cursor_address(hSession,baddr)) {
			throw std::system_error(errno, std::system_category());
		}

		return *this;
	}

	/// @brief Set cursor address.
	///
	/// @param addr	Cursor address.
	TN3270::Session & Local::Session::setCursor(unsigned short addr) {

		if(lib3270_set_cursor_address(hSession,addr) < 0) {
			throw std::system_error(errno, std::system_category());
		}

		return *this;
	}

	/// @brief Get cursor address.
	///
	/// @return
	unsigned short Local::Session::getCursorAddress() {

		unsigned int position = lib3270_get_cursor_address(hSession);

		if(!position)
			throw std::system_error(errno, std::system_category());

		return position;
	}


	/// @brief Set cursor position.
	///
	/// @param row	New cursor row.
	/// @param col	New cursor column.
	TN3270::Session & Local::Session::setCursor(unsigned short row, unsigned short col) {

		if(lib3270_set_cursor_position(hSession,row,col)) {
			throw std::system_error(errno, std::system_category());
		}

		return *this;

	}

	// Get properties.
	std::string Local::Session::getVersion() const {
		return lib3270_get_version();
	}

	std::string Local::Session::getRevision() const {
		return lib3270_get_revision();
	}

	std::string Local::Session::getLUName() const {
		const char * luname =  lib3270_get_luname(hSession);
		if(luname)
			return luname;
		return "";
	}

	/// @brief Execute action by name.
	TN3270::Session & Local::Session::action(const char *action_name) {

		chkResponse(lib3270_action(hSession,action_name));

		return *this;
	}

	/// @brief Wait.
	TN3270::Session & Local::Session::wait(unsigned short seconds) {
		chkResponse(lib3270_wait(hSession,seconds));
		return *this;
	}

	/// @brief Wait for update.
	TN3270::Session & Local::Session::wait_for_update(unsigned short seconds) {
		throw std::system_error(ENOTSUP, std::system_category());
		return *this;
	}

	void Local::Session::setUnlockDelay(unsigned short delay) {
		lib3270_set_unlock_delay(hSession,delay);
	}

 }