libssh c++

ssh.h

/* ssh_t Using libssh*/#ifndef __LIBSSH_CLASS_TYPE_H__#define __LIBSSH_CLASS_TYPE_H_#include <libssh/libssh.h>#include <libssh/ssh2.h>#include <libssh/sftp.h>#include <libssh/callbacks.h>#include <sys/types.h>#include <sys/stat.h>#include <sys/time.h>#include <dirent.h>#include <unistd.h>#include <stdlib.h>#include <stdint.h>#include <fcntl.h>#include <errno.h>#include <pthread.h>#include <string>#include <fstream>class ssh_t{public: ssh_t(); virtual ~ssh_t(); /** ssh_options_set */ int set_host(const std::string& host){return ssh_options_set(m_session, SSH_OPTIONS_HOST, host.c_str());} int set_port(int port){return ssh_options_set(m_session, SSH_OPTIONS_PORT, &port);} int set_user(const std::string& user){return ssh_options_set(m_session, SSH_OPTIONS_USER, user.c_str());} int set_path(const std::string& path){return ssh_options_set(m_session, SSH_OPTIONS_SSH_DIR, path.c_str());} int set_timeout(int t){return ssh_options_set(m_session, SSH_OPTIONS_TIMEOUT, &t);} int set_ssh2(int on){return ssh_options_set(m_session, SSH_OPTIONS_SSH2, &on);} int set_ssh1(int on){return ssh_options_set(m_session, SSH_OPTIONS_SSH1, &on);} /** session */ int connect(const std::string& host, int port, const std::string& user, const std::string& password) {set_host(host); set_user(user); set_port(port); if(ssh_connect(m_session))return -1; return authenticate(password);} void disconnect() {if(m_session){::ssh_disconnect(m_session);}} int get_status()const {return ssh_get_status(m_session);} const char* get_error()const {return ssh_get_error(m_session);} int get_error_code()const {return ssh_get_error_code(m_session);} /** new functions */ int exec(const std::string& cmdline, int fd_stdout=-1, int fd_stderr=-1); int exec(const std::string& cmdline, std::ostream* strm_stdout=NULL, std::ostream* strm_stderr=NULL); int upload(const std::string& local, const std::string& remote_path, std::string& remote_name); int download(const std::string& remote, const std::string& local_path, std::string& local_name); /** scp2 not supposed! */ int scp_upload(const std::string& local, const std::string& remote_path, const std::string& remote_name="");private: int authenticate(const std::string& password); int authenticate_kbdint(const std::string& password); /** channel */ int duplicate_from_channel(ssh_channel channel, int fd, int is_stderr); int duplicate_from_channel(ssh_channel channel, std::ostream& strm, int is_stderr); /** sftp */ int upload_file(sftp_session sftp, const std::string& local, struct stat& st, const std::string& remote); int upload_symlink(sftp_session sftp, const std::string& local, struct stat& st, const std::string& remote); int upload_dir(sftp_session sftp, const std::string& local, struct stat& st, const std::string& remote); int download_file(sftp_session sftp, const std::string& remote, sftp_attributes attrs, const std::string& local); int download_symlink(sftp_session sftp, const std::string& remote, sftp_attributes attrs, const std::string& local); int download_dir(sftp_session sftp, const std::string& remote, sftp_attributes attrs, const std::string& local); /** scp */private: ssh_session m_session; sftp_session m_sftp_session; char* m_buf; const static uint32_t m_buflen = 1024*16;};#endif //__LIBSSH_CLASS_TYPE_H_ 

ssh.cpp

#include "ssh.h"static pthread_once_t once_ssh_init = PTHREAD_ONCE_INIT;void do_ssh_init(){ssh_init();}ssh_t::ssh_t():m_session(ssh_new()), m_buf(new char[m_buflen]){ pthread_once(&once_ssh_init, do_ssh_init); /** default options */ set_port(36000); set_ssh2(1); set_ssh1(1); set_timeout(30);}ssh_t::~ssh_t(){ disconnect(); if(m_session) { ssh_free(m_session); m_session = NULL; }}int ssh_t::authenticate_kbdint(const std::string& password){ int err=ssh_userauth_kbdint(m_session,NULL,NULL); const char *name, *instruction, *prompt; int i,n; char echo; while (err==SSH_AUTH_INFO) { name=ssh_userauth_kbdint_getname(m_session); instruction=ssh_userauth_kbdint_getinstruction(m_session); n=ssh_userauth_kbdint_getnprompts(m_session); for(i=0;i<n;++i) { prompt=ssh_userauth_kbdint_getprompt(m_session,i,&echo); if(echo && strstr(prompt, "(yes/no)?")) { if (ssh_userauth_kbdint_setanswer(m_session,i,"yes") < 0) { return SSH_AUTH_ERROR; } } else { if (ssh_userauth_kbdint_setanswer(m_session,i,password.c_str()) < 0) { return SSH_AUTH_ERROR; } } } err=ssh_userauth_kbdint(m_session,NULL,NULL); } return err;}int ssh_t::authenticate(const std::string& password){ // Try to authenticate int rc = ::ssh_userauth_none(m_session, NULL); if (rc == SSH_AUTH_ERROR) { return rc; } int method = ssh_auth_list(m_session); while (rc != SSH_AUTH_SUCCESS) { // Try to authenticate with password if (method & SSH_AUTH_METHOD_PASSWORD) { rc = ssh_userauth_password(m_session, NULL, password.c_str()); if (rc == SSH_AUTH_ERROR) { return rc; } else if (rc == SSH_AUTH_SUCCESS) { break; } } // Try to authenticate with keyboard interactive"; if (method & SSH_AUTH_METHOD_INTERACTIVE) { rc = authenticate_kbdint(password); if (rc == SSH_AUTH_ERROR) { return rc; } else if (rc == SSH_AUTH_SUCCESS) { break; } } // Try to authenticate with public key first if (method & SSH_AUTH_METHOD_PUBLICKEY) { rc = ssh_userauth_autopubkey(m_session, NULL); if (rc == SSH_AUTH_ERROR) { return rc; } else if (rc == SSH_AUTH_SUCCESS) { break; } } } return rc;}int ssh_t::duplicate_from_channel(ssh_channel channel, int fd, int is_stderr){ int dup_cnt = 0; while (0==::channel_is_eof(channel)) { int len = ::channel_read(channel, m_buf, m_buflen, is_stderr); if (len<=0 || SSH_ERROR==len) { break; } while (len>0) { int wn = ::write(fd, m_buf, len); if (wn>0) { dup_cnt += wn; len -= wn; } else if (-1==wn && EINTR!=errno && EAGAIN!=errno) { return -1; } } if (len>0) { break; } } return dup_cnt;}int ssh_t::duplicate_from_channel(ssh_channel channel, std::ostream& strm, int is_stderr){ int dup_cnt = 0; while (0==::channel_is_eof(channel)) { int len = ::channel_read(channel, m_buf, m_buflen, is_stderr); if (len<=0 || SSH_ERROR==len) { break; } if (!strm.write(m_buf, len)) { return -1; } } return dup_cnt;}int ssh_t::exec(const std::string& cmdline, int fd_stdout/*=-1*/, int fd_stderr/*=-1*/){ int ret = 0; /** build channel */ ssh_channel channel = channel_new(m_session); if (NULL==channel) return -1; if(channel_open_session(channel)) { channel_free(channel); return -1; } /** request exec */ if (0!=channel_request_exec(channel, cmdline.c_str())) { ret = -1; } else { ::channel_send_eof(channel); /** get exit code */ ret = channel_get_exit_status(channel); /** dulicate channel data */ if (fd_stdout>0) { duplicate_from_channel(channel, fd_stdout, 0); } if (fd_stderr>0) { duplicate_from_channel(channel, fd_stderr, 1); } } channel_close(channel); channel_free(channel); channel = NULL; return ret;}int ssh_t::exec(const std::string& cmdline, std::ostream* pstrm_stdout/*=NULL*/, std::ostream* pstrm_stderr/*=NULL*/){ int ret = 0; /** build channel */ ssh_channel channel = channel_new(m_session); if (NULL==channel) return -1; if(channel_open_session(channel)) { channel_free(channel); return -1; } /** request exec */ if (0!=channel_request_exec(channel, cmdline.c_str())) { ret = -1; } else { ::channel_send_eof(channel); /** get exit code */ ret = channel_get_exit_status(channel); /** dulicate channel data */ if (pstrm_stdout && *pstrm_stdout) { duplicate_from_channel(channel, *pstrm_stdout, 0); } if (pstrm_stderr && *pstrm_stderr) { duplicate_from_channel(channel, *pstrm_stderr, 1); } } channel_close(channel); channel_free(channel); channel = NULL; return ret;}int ssh_t::upload(const std::string& local, const std::string& remote_path, std::string& remote_name){ sftp_session sftp = sftp_new(m_session); if (NULL==sftp) { return -1; } if (::sftp_init(sftp)) { sftp_free(sftp); return -1; } struct stat st; if (::stat(local.c_str(), &st)) /* top target, call stat(but not lstat) */ { sftp_free(sftp); return -1; } /* rename or keep local name */ std::string remote_full_path = remote_path; if (remote_name.size()) { remote_full_path = remote_path + "/" + remote_name; } else { char* base = ssh_basename(local.c_str()); if (base) { remote_full_path = remote_path + "/" + base; remote_name = base; free(base); } } int ret = 0; switch (st.st_mode & S_IFMT) { case S_IFLNK: case S_IFREG: ret = upload_file(sftp, local, st, remote_full_path); break; case S_IFDIR: ret = upload_dir(sftp, local, st, remote_full_path); break; } sftp_free(sftp); return ret;}int ssh_t::download(const std::string& remote, const std::string& local_path, std::string& local_name){ sftp_session sftp = sftp_new(m_session); if (NULL==sftp) { return -1; } if (::sftp_init(sftp)) { sftp_free(sftp); return -1; } sftp_attributes attrs = ::sftp_stat(sftp, remote.c_str()); /* top target, call stat(but not lstat) */ if (NULL==attrs) { sftp_free(sftp); return -1; } /* rename or keep local name */ std::string local_full_path = local_path; if (local_name.size()) { local_full_path = local_path + "/" + local_name; } else { char* base = ssh_basename(remote.c_str()); if (base) { local_full_path = local_path + "/" + base; local_name = base; free(base); } } int ret = 0; switch (attrs->type) { case SSH_FILEXFER_TYPE_REGULAR: download_file(sftp, remote, attrs, local_full_path); break; case SSH_FILEXFER_TYPE_SYMLINK: download_symlink(sftp, remote, attrs, local_full_path); break; case SSH_FILEXFER_TYPE_DIRECTORY: download_dir(sftp, remote, attrs, local_full_path); break; default: break; } sftp_free(sftp); sftp_attributes_free(attrs); return ret;}int ssh_t::upload_file(sftp_session sftp, const std::string& local, struct stat& st, const std::string& remote){ int ret = 0; /* open remote for write */ sftp_file file = sftp_open(sftp, remote.c_str(), O_WRONLY | O_CREAT | O_TRUNC, st.st_mode&0777); if (NULL==file) { return -1; } /* duplicate */ off_t duplen = 0; std::ifstream istrm; istrm.open(local.c_str()); while (istrm && !istrm.eof()) { int len = istrm.read(m_buf, m_buflen).gcount(); if (len>0) { int wn = sftp_write(file, m_buf, len); if (wn!=len) { break; } duplen += wn; } } istrm.close(); sftp_close(file); if (duplen!=st.st_size) { ret = -1; } struct timeval times[2]; times[0].tv_sec = st.st_atime; times[0].tv_usec = 0; times[1].tv_sec = st.st_ctime; times[1].tv_usec = 0; ::sftp_utimes(sftp, remote.c_str(), times); return ret;}int ssh_t::download_file(sftp_session sftp, const std::string& remote, sftp_attributes attrs, const std::string& local){ int ret = 0; /* open remote for write */ sftp_file file = sftp_open(sftp, remote.c_str(), O_RDONLY, 0); if (NULL==file) { return -1; } /* duplicate */ off_t duplen = 0; std::ofstream ostrm; ostrm.open(local.c_str()); if (!ostrm) { sftp_close(file); return -1; } while (1) { int len = ::sftp_read(file, m_buf, m_buflen); if (len>0) { if (!ostrm.write(m_buf, len)) { break; } duplen += len; } else { break; } } ostrm.close(); sftp_close(file); if (duplen!=(off_t)attrs->size) { ret = -1; } struct timeval times[2]; times[0].tv_sec = attrs->atime; times[0].tv_usec = 0; times[1].tv_sec = attrs->mtime; times[1].tv_usec = 0; ::utimes(local.c_str(), times); ::chmod(local.c_str(), attrs->permissions&0777); return ret;}int ssh_t::upload_symlink(sftp_session sftp, const std::string& local, struct stat& st, const std::string& remote){ char lnk[PATH_MAX] = {0}; ::readlink(local.c_str(), lnk, PATH_MAX); int ret = ::sftp_symlink(sftp, lnk, remote.c_str()); if (ret) { sftp_attributes attrs = ::sftp_stat(sftp, remote.c_str()); if (attrs && SSH_FILEXFER_TYPE_SYMLINK==attrs->type) { ret = 0; } if (attrs) { sftp_attributes_free(attrs); } } return ret;}int ssh_t::download_symlink(sftp_session sftp, const std::string& remote, sftp_attributes attrs, const std::string& local){ char* lnk = ::sftp_readlink(sftp, remote.c_str()); int ret = ::symlink(local.c_str(), lnk); free(lnk); return ret;}int ssh_t::upload_dir(sftp_session sftp, const std::string& local, struct stat& st, const std::string& remote){ if (::sftp_mkdir(sftp, remote.c_str(), st.st_mode&0777) && SSH_FX_FILE_ALREADY_EXISTS!=sftp_get_error(sftp)) { return -1; } DIR* dir = ::opendir(local.c_str()); if (NULL==dir) { return -1; } int ret = 0; struct dirent entry, *pentry=NULL; while (readdir_r(dir, &entry, &pentry)==0 && pentry && 0==ret) { if ((entry.d_name[0]=='.'&&entry.d_name[1]=='/0') || (entry.d_name[0]=='.'&&entry.d_name[1]=='.'&&entry.d_name[2]=='/0')) { continue; } std::string sub_local = local + "/" + entry.d_name; std::string sub_remote = remote + "/" + entry.d_name; struct stat sub_st; ::lstat(sub_local.c_str(), &sub_st); /* do lstat here */ switch (sub_st.st_mode & S_IFMT) { case S_IFREG: ret = upload_file(sftp, sub_local, sub_st, sub_remote); break; case S_IFLNK: ret = upload_symlink(sftp, sub_local, sub_st, sub_remote); break; case S_IFDIR: ret = upload_dir(sftp, sub_local, sub_st, sub_remote); break; default: ret = -1; break; } } ::closedir(dir); return ret;}int ssh_t::download_dir(sftp_session sftp, const std::string& remote, sftp_attributes attrs, const std::string& local){ if (::mkdir(local.c_str(), attrs->permissions&0777) && EEXIST!=errno) { return -1; } sftp_dir dir = ::sftp_opendir(sftp, remote.c_str()); if (NULL==dir) { return -1; } int ret = 0; while (!sftp_dir_eof(dir) && 0==ret) { sftp_attributes sub_attrs = sftp_readdir(sftp, dir); if (NULL==sub_attrs) { ret = -1; break; } if (('.'==sub_attrs->name[0]&&0==sub_attrs->name[1]) || ('.'==sub_attrs->name[0]&&'.'==sub_attrs->name[1]&&0==sub_attrs->name[2])) { sftp_attributes_free(sub_attrs); continue; } std::string sub_local = local + "/" + sub_attrs->name; std::string sub_remote = remote + "/" + sub_attrs->name; switch (sub_attrs->type) { case SSH_FILEXFER_TYPE_REGULAR: ret = download_file(sftp, sub_remote, sub_attrs, sub_local); break; case SSH_FILEXFER_TYPE_SYMLINK: ret = download_symlink(sftp, sub_remote, sub_attrs, sub_local); break; case SSH_FILEXFER_TYPE_DIRECTORY: ret = download_dir(sftp, sub_remote, sub_attrs, sub_local); break; default: ret = -1; break; } sftp_attributes_free(sub_attrs); } ::sftp_closedir(dir); return ret;}int ssh_t::scp_upload(const std::string& local, const std::string& remote_path, const std::string& remote_name/*=""*/){ struct stat st; if (::stat(local.c_str(), &st)) /* top target, call stat(but not lstat) */ { return -1; } /* rename or keep local name */ std::string remote_full_path = remote_path; if (remote_name.size()) { remote_full_path = remote_path + "/" + remote_name; } else { char* base = ssh_basename(local.c_str()); remote_full_path = remote_path + "/" + base; free(base); } ssh_scp scp = ssh_scp_new(m_session, SSH_SCP_WRITE, remote_full_path.c_str()); if (NULL==scp) { return -1; } if (::ssh_scp_init(scp)) { ssh_scp_free(scp); return -1; } int ret = -1; switch (st.st_mode & S_IFMT) { case S_IFREG: ret = SSH_OK==ssh_scp_push_file(scp, local.c_str(), st.st_size, st.st_mode&077)?0:-1; break; case S_IFDIR: ret = SSH_OK==ssh_scp_push_directory(scp, local.c_str(), st.st_mode&077)?0:-1; break; default: ret = -1; break; } if (scp) { ssh_scp_close(scp); ssh_scp_free(scp); } return ret;}