- 论坛徽章:
- 0
|
libssh2_exec.c
- #include "libssh2_exec.h"
- //#include "libssh2_config.h"
- //#include <libssh2.h>
- static int waitsocket(int *socket_fd, LIBSSH2_SESSION *session)
- {
- struct timeval timeout;
- int rc;
- fd_set fd;
- fd_set *writefd = NULL;
- fd_set *readfd = NULL;
- int dir;
- timeout.tv_sec = 10;
- timeout.tv_usec = 0;
- FD_ZERO(&fd);
- FD_SET(*socket_fd, &fd);
- /* now make sure we wait in the correct direction */
- dir = libssh2_session_block_directions(session);
- if(dir & LIBSSH2_SESSION_BLOCK_INBOUND)
- readfd = &fd;
- if(dir & LIBSSH2_SESSION_BLOCK_OUTBOUND)
- writefd = &fd;
- rc = select(*socket_fd + 1, readfd, writefd, NULL, &timeout);
- return rc;
- }
- void close_channel(LIBSSH2_CHANNEL *channel)
- {
- libssh2_channel_free(channel);
- channel = NULL;
- }
- void shutdown_socket(int *sock, LIBSSH2_SESSION *session)
- {
- libssh2_session_disconnect(session,"Normal Shutdown, Thank you for playing");
- libssh2_session_free(session);
- close(*sock);
- }
- /***********************************************
- 函数名称:connect_sock
- 函数功能:建立ssh 连接
- 传入参数:
- 返回数据:
- ***********************************************/
- int connect_sock(int *sock, LIBSSH2_SESSION **session,LIBSSH2_CHANNEL **channel)
- {
- const char *hostname = "127.0.0.1";
- const char *username = "root";
- const char *password = "root";
- unsigned long hostaddr;
- struct sockaddr_in sin;
- const char *fingerprint;
- int rc;
- int exitcode;
- int bytecount = 0;
- size_t len;
- LIBSSH2_KNOWNHOSTS *nh;
- int type;
- hostaddr = inet_addr(hostname);
- /* Ultra basic "connect to port 22 on localhost"
- * Your code is responsible for creating the socket establishing the
- * connection
- */
- *sock = socket(AF_INET, SOCK_STREAM, 0); //建立ssh socket
- sin.sin_family = AF_INET;
- sin.sin_port = htons(22);
- sin.sin_addr.s_addr = hostaddr;
- if (connect(*sock, (struct sockaddr*)(&sin), sizeof(struct sockaddr_in)) != 0) {
- fprintf(stderr, "failed to connect!\n");
- return -1;
- }
- /* Create a *session instance */
- *session = libssh2_session_init();
- if (!*session)
- return -1;
- /* tell libssh2 we want it all done non-blocking */
- libssh2_session_set_blocking(*session, 0); //建立ssh 会话
- /* ... start it up. This will trade welcome banners, exchange keys,
- * and setup crypto, compression, and MAC layers
- */
- while ((rc = libssh2_session_startup(*session, *sock)) == LIBSSH2_ERROR_EAGAIN);
- if (rc) {
- fprintf(stderr, "Failure establishing SSH *session: %d\n", rc);
- return -1;
- }
- nh = libssh2_knownhost_init(*session);
- if(!nh) {
- /* eeek, do cleanup here */
- return 2;
- }
- /* read all hosts from here */
- libssh2_knownhost_readfile(nh, "known_hosts",
- LIBSSH2_KNOWNHOST_FILE_OPENSSH);
- /* store all known hosts to here */
- libssh2_knownhost_writefile(nh, "dumpfile", LIBSSH2_KNOWNHOST_FILE_OPENSSH);
- fingerprint = libssh2_session_hostkey(*session, &len, &type);
- if(fingerprint) {
- struct libssh2_knownhost *host;
- int check = libssh2_knownhost_check(nh, (char *)hostname,
- (char *)fingerprint, len,
- LIBSSH2_KNOWNHOST_TYPE_PLAIN|
- LIBSSH2_KNOWNHOST_KEYENC_RAW,
- &host);
- // fprintf(stderr, "Host check: %d, key: %s\n", check,
- // (check <= LIBSSH2_KNOWNHOST_CHECK_MISMATCH)?
- // host->key:"<none>");
- /*****
- * At this point, we could verify that 'check' tells us the key is
- * fine or bail out.
- *****/
- }
- else {
- /* eeek, do cleanup here */
- return 3;
- }
- libssh2_knownhost_free(nh);
- if ( strlen(password) != 0 ) {
- /* We could authenticate via password *///进行密码验证
- while ((rc = libssh2_userauth_password(*session, username, password)) ==
- LIBSSH2_ERROR_EAGAIN);
- if (rc) {
- fprintf(stderr, "Authentication by password failed.\n");
- shutdown_socket(sock, *session);
- return 1;
- }
- }
- else {
- /* Or by public key */
- while ((rc = libssh2_userauth_publickey_fromfile(*session, username,
- "/home/user/"
- ".ssh/id_rsa.pub",
- "/home/user/"
- ".ssh/id_rsa",
- password)) ==
- LIBSSH2_ERROR_EAGAIN);
- if (rc) {
- fprintf(stderr, "\tAuthentication by public key failed\n");
- shutdown_socket(sock, *session);
- }
- }
- /* Exec non-blocking on the remove host */
- return 0;
- }
- /***********************************************
- 函数名称:create_channel
- 函数功能:创建ssh通道
- 传入参数:
- 返回数据:
- ***********************************************/
- int create_channel(int *sock, LIBSSH2_SESSION **session,LIBSSH2_CHANNEL **channel)
- {
- while( (*channel = libssh2_channel_open_session(*session)) == NULL &&
- libssh2_session_last_error(*session,NULL,NULL,0) ==
- LIBSSH2_ERROR_EAGAIN )
- {
- waitsocket(sock, *session);
- }
- if( *channel == NULL )
- {
- fprintf(stderr,"Error\n");
- exit( 1 );
- }
- return 0;
- }
- /***********************************************
- 函数名称:cmd_request
- 函数功能:执行请求的命令
- 传入参数:
- 返回数据:
- ***********************************************/
- int cmd_request(int *sock, LIBSSH2_SESSION *session,LIBSSH2_CHANNEL *channel,char *cmdline)
- {
-
- int rc=0,exitcode=0,bytecount=0;
- while( (rc = libssh2_channel_exec(channel, cmdline)) == LIBSSH2_ERROR_EAGAIN )//执行请求的命令
- {
- waitsocket(sock, session);
- }
- if( rc != 0 )
- {
- fprintf(stderr,"Error\n");
- exit( 1 );
- }
- char buffer[2048];
- for( ;; )
- {
- // loop until we block
- int rc=0,exitcode=0;
- do
- {
-
- memset(&buffer,0,sizeof(buffer));
- rc = libssh2_channel_read( channel, buffer, sizeof(buffer)-1 );//分批接收命令执行结果
- buffer[rc] = '\0';
- printf("\n%s\n",buffer);
- }
- while( rc > 0 );
- // this is due to blocking that would occur otherwise so we loop on
- // this condition
- if( rc == LIBSSH2_ERROR_EAGAIN )
- {
- waitsocket(sock, session);
- }
- else
- break;
- }
- exitcode = 127;
- while( (rc = libssh2_channel_close(channel)) == LIBSSH2_ERROR_EAGAIN )
- waitsocket(sock, session);
-
- if( rc == 0 )
- {
- exitcode = libssh2_channel_get_exit_status( channel );
- }
- if(exitcode == 127)
- {
- }
-
- return 0;
- }
复制代码
[ 本帖最后由 diyself 于 2010-1-13 13:12 编辑 ] |
|