diff options
author | Alex Crichton <alex@alexcrichton.com> | 2015-04-29 09:52:15 -0700 |
---|---|---|
committer | Alex Crichton <alex@alexcrichton.com> | 2015-04-29 09:52:24 -0700 |
commit | 40ca99b2d6bc6d4046d2b74438dc9ea769d28c5b (patch) | |
tree | 19ab2f7adf3723fe200d1f3ed8362a9367fff598 /libssh2-sys/libssh2-1.5.0/example/sftpdir_nonblock.c | |
parent | cce73bddc574715c0eb77ec5450a76840d43df81 (diff) | |
download | ssh2-rs-40ca99b2d6bc6d4046d2b74438dc9ea769d28c5b.zip |
Update libssh2 to 1.5.0
Diffstat (limited to 'libssh2-sys/libssh2-1.5.0/example/sftpdir_nonblock.c')
-rw-r--r-- | libssh2-sys/libssh2-1.5.0/example/sftpdir_nonblock.c | 252 |
1 files changed, 252 insertions, 0 deletions
diff --git a/libssh2-sys/libssh2-1.5.0/example/sftpdir_nonblock.c b/libssh2-sys/libssh2-1.5.0/example/sftpdir_nonblock.c new file mode 100644 index 0000000..1950e67 --- /dev/null +++ b/libssh2-sys/libssh2-1.5.0/example/sftpdir_nonblock.c @@ -0,0 +1,252 @@ +/* + * Sample doing an SFTP directory listing. + * + * The sample code has default values for host name, user name, password and + * path, but you can specify them on the command line like: + * + * "sftpdir 192.168.0.1 user password /tmp/secretdir" + */ + +#include "libssh2_config.h" +#include <libssh2.h> +#include <libssh2_sftp.h> + +#ifdef HAVE_WINSOCK2_H +# include <winsock2.h> +#endif +#ifdef HAVE_SYS_SOCKET_H +# include <sys/socket.h> +#endif +#ifdef HAVE_NETINET_IN_H +# include <netinet/in.h> +#endif +# ifdef HAVE_UNISTD_H +#include <unistd.h> +#endif +#ifdef HAVE_ARPA_INET_H +# include <arpa/inet.h> +#endif +#ifdef HAVE_INTTYPES_H +# include <inttypes.h> +#endif + +#include <sys/types.h> +#include <fcntl.h> +#include <errno.h> +#include <stdio.h> +#include <ctype.h> + +/* last resort for systems not defining PRIu64 in inttypes.h */ +#ifndef __PRI64_PREFIX +#ifdef WIN32 +#define __PRI64_PREFIX "I64" +#else +#if __WORDSIZE == 64 +#define __PRI64_PREFIX "l" +#else +#define __PRI64_PREFIX "ll" +#endif /* __WORDSIZE */ +#endif /* WIN32 */ +#endif /* !__PRI64_PREFIX */ +#ifndef PRIu64 +#define PRIu64 __PRI64_PREFIX "u" +#endif /* PRIu64 */ + +int main(int argc, char *argv[]) +{ + unsigned long hostaddr; + int sock, i, auth_pw = 1; + struct sockaddr_in sin; + const char *fingerprint; + LIBSSH2_SESSION *session; + const char *username="username"; + const char *password="password"; + const char *sftppath="/tmp/secretdir"; + int rc; + LIBSSH2_SFTP *sftp_session; + LIBSSH2_SFTP_HANDLE *sftp_handle; + +#ifdef WIN32 + WSADATA wsadata; + int err; + + err = WSAStartup(MAKEWORD(2,0), &wsadata); + if (err != 0) { + fprintf(stderr, "WSAStartup failed with error: %d\n", err); + return 1; + } +#endif + + if (argc > 1) { + hostaddr = inet_addr(argv[1]); + } else { + hostaddr = htonl(0x7F000001); + } + + if(argc > 2) { + username = argv[2]; + } + if(argc > 3) { + password = argv[3]; + } + if(argc > 4) { + sftppath = argv[4]; + } + + rc = libssh2_init (0); + if (rc != 0) { + fprintf (stderr, "libssh2 initialization failed (%d)\n", rc); + return 1; + } + + /* + * The application code is responsible for creating the socket + * and establishing the connection + */ + sock = socket(AF_INET, SOCK_STREAM, 0); + + 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; + + /* Since we have set non-blocking, tell libssh2 we are non-blocking */ + libssh2_session_set_blocking(session, 0); + + /* ... start it up. This will trade welcome banners, exchange keys, + * and setup crypto, compression, and MAC layers + */ + while ((rc = libssh2_session_handshake(session, sock)) == + LIBSSH2_ERROR_EAGAIN); + if(rc) { + fprintf(stderr, "Failure establishing SSH session: %d\n", rc); + return -1; + } + + /* At this point we havn't yet authenticated. The first thing to do + * is check the hostkey's fingerprint against our known hosts Your app + * may have it hard coded, may go to a file, may present it to the + * user, that's your call + */ + fingerprint = libssh2_hostkey_hash(session, LIBSSH2_HOSTKEY_HASH_SHA1); + fprintf(stderr, "Fingerprint: "); + for(i = 0; i < 20; i++) { + fprintf(stderr, "%02X ", (unsigned char)fingerprint[i]); + } + fprintf(stderr, "\n"); + + if (auth_pw) { + /* 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"); + goto shutdown; + } + } else { + /* Or by public key */ + while ((rc = libssh2_userauth_publickey_fromfile(session, username, + "/home/username/.ssh/id_rsa.pub", + "/home/username/.ssh/id_rsa", + password)) == LIBSSH2_ERROR_EAGAIN); + if (rc) { + fprintf(stderr, "\tAuthentication by public key failed\n"); + goto shutdown; + } + } + + fprintf(stderr, "libssh2_sftp_init()!\n"); + do { + sftp_session = libssh2_sftp_init(session); + + if ((!sftp_session) && (libssh2_session_last_errno(session) != + LIBSSH2_ERROR_EAGAIN)) { + fprintf(stderr, "Unable to init SFTP session\n"); + goto shutdown; + } + } while (!sftp_session); + + fprintf(stderr, "libssh2_sftp_opendir()!\n"); + /* Request a dir listing via SFTP */ + do { + sftp_handle = libssh2_sftp_opendir(sftp_session, sftppath); + + if ((!sftp_handle) && (libssh2_session_last_errno(session) != + LIBSSH2_ERROR_EAGAIN)) { + fprintf(stderr, "Unable to open dir with SFTP\n"); + goto shutdown; + } + } while (!sftp_handle); + + fprintf(stderr, "libssh2_sftp_opendir() is done, now receive listing!\n"); + do { + char mem[512]; + LIBSSH2_SFTP_ATTRIBUTES attrs; + + /* loop until we fail */ + while ((rc = libssh2_sftp_readdir(sftp_handle, mem, sizeof(mem), + &attrs)) == LIBSSH2_ERROR_EAGAIN) { + ; + } + if(rc > 0) { + /* rc is the length of the file name in the mem + buffer */ + + if(attrs.flags & LIBSSH2_SFTP_ATTR_PERMISSIONS) { + /* this should check what permissions it + is and print the output accordingly */ + printf("--fix----- "); + } else { + printf("---------- "); + } + + if(attrs.flags & LIBSSH2_SFTP_ATTR_UIDGID) { + printf("%4ld %4ld ", attrs.uid, attrs.gid); + } else { + printf(" - - "); + } + + if(attrs.flags & LIBSSH2_SFTP_ATTR_SIZE) { + printf("%8" PRIu64 " ", attrs.filesize); + } + + printf("%s\n", mem); + } + else if (rc == LIBSSH2_ERROR_EAGAIN) { + /* blocking */ + fprintf(stderr, "Blocking\n"); + } else { + break; + } + + } while (1); + + libssh2_sftp_closedir(sftp_handle); + libssh2_sftp_shutdown(sftp_session); + + shutdown: + + libssh2_session_disconnect(session, "Normal Shutdown, Thank you for playing"); + libssh2_session_free(session); + +#ifdef WIN32 + closesocket(sock); +#else + close(sock); +#endif + fprintf(stderr, "all done\n"); + + libssh2_exit(); + + return 0; +} |