/*
* Sample showing how to do SFTP write transfers.
*
* The sample code has default values for host name, user name, password
* and path to copy, but you can specify them on the command line like:
*
* "sftp 192.168.0.1 user password sftp_write.c /tmp/secrets"
*/
#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
#include <sys/types.h>
#include <fcntl.h>
#include <errno.h>
#include <stdio.h>
#include <ctype.h>
#include <string.h>
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 *loclfile = "sftp_write.c";
const char *sftppath = "/tmp/TEST";
int rc;
FILE *local;
LIBSSH2_SFTP *sftp_session;
LIBSSH2_SFTP_HANDLE *sftp_handle;
char mem[1024*100];
size_t nread;
char *ptr;
char dir[256];
char* pchar;
char* pchar2;
LIBSSH2_SFTP_HANDLE *sftp_dir_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) {
loclfile = argv[4];
}
if(argc > 5) {
sftppath = argv[5];
}
rc = libssh2_init(0);
if(rc != 0) {
fprintf(stderr, "libssh2 initialization failed (%d)\n", rc);
return 1;
}
local = fopen(loclfile, "rb");
if(!local) {
fprintf(stderr, "Can't open local file %s\n", loclfile);
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 blocking */
libssh2_session_set_blocking(session, 1);
/* ... start it up. This will trade welcome banners, exchange keys,
* and setup crypto, compression, and MAC layers
*/
rc = libssh2_session_handshake(session, sock);
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 */
if(libssh2_userauth_password(session, username, password)) {
fprintf(stderr, "Authentication by password failed.\n");
goto shutdown;
}
}
else {
/* Or by public key */
const char *pubkey = "/home/username/.ssh/id_rsa.pub";
const char *privkey = "/home/username/.ssh/id_rsa.pub";
if(libssh2_userauth_publickey_fromfile(session, username,
pubkey, privkey,
password)) {
fprintf(stderr, "\tAuthentication by public key failed\n");
goto shutdown;
}
}
fprintf(stderr, "libssh2_sftp_init()!\n");
sftp_session = libssh2_sftp_init(session);
if(!sftp_session) {
fprintf(stderr, "Unable to init SFTP session\n");
goto shutdown;
}
#if 1
if( *(sftppath + strlen(sftppath) - 1) == '/') {
fprintf(stderr, "Please specify file name for SFTP server\n");
goto shutdown;
}
/* Loop Create DIR */
memset(dir, 0, sizeof(dir));
pchar = strchr(sftppath, '/');
if( pchar == sftppath ) {
while( strlen(pchar) > 0 && (pchar2 = strchr((char *)pchar + 1, '/')) ) {
// example: /tmp/1.txt or /tmp/1.txt/ or /tmp/test/files/1.txt or /tmp//test/files/1.txt all will deal
if(strlen(pchar2) < 1)
break;
if((pchar - pchar2) == 1) {
pchar = pchar2;
continue;
}
strncpy(dir + strlen(dir), pchar, pchar2 - pchar);
fprintf(stderr, "DIR is %s\n", dir);
fprintf(stderr, "libssh2_sftp_opendir()!\n");
/* Request a dir listing via SFTP */
sftp_dir_handle = libssh2_sftp_opendir(sftp_session, dir);
if(!sftp_dir_handle) {
fprintf(stderr, "Unable to open dir with SFTP, create it\n");
/* Make a directory via SFTP */
rc = libssh2_sftp_mkdir(sftp_session, dir,
LIBSSH2_SFTP_S_IRWXU|
LIBSSH2_SFTP_S_IRGRP|LIBSSH2_SFTP_S_IXGRP|
LIBSSH2_SFTP_S_IROTH|LIBSSH2_SFTP_S_IXOTH);
if(rc) {
fprintf(stderr, "libssh2_sftp_mkdir failed: %d\n", rc);
break;
}
}else {
libssh2_sftp_closedir(sftp_dir_handle);
}
pchar = pchar2;
}
}
#endif
fprintf(stderr, "libssh2_sftp_open()!\n");
/* Request a file via SFTP */
sftp_handle =
libssh2_sftp_open(sftp_session, sftppath,
LIBSSH2_FXF_WRITE|LIBSSH2_FXF_CREAT|LIBSSH2_FXF_TRUNC,
LIBSSH2_SFTP_S_IRUSR|LIBSSH2_SFTP_S_IWUSR|
LIBSSH2_SFTP_S_IRGRP|LIBSSH2_SFTP_S_IROTH);
if(!sftp_handle) {
fprintf(stderr, "Unable to open file with SFTP\n");
goto shutdown;
}
fprintf(stderr, "libssh2_sftp_open() is done, now send data!\n");
do {
nread = fread(mem, 1, sizeof(mem), local);
if(nread <= 0) {
/* end of file */
break;
}
ptr = mem;
do {
/* write data in a loop until we block */
rc = libssh2_sftp_write(sftp_handle, ptr, nread);
if(rc < 0)
break;
ptr += rc;
nread -= rc;
} while(nread);
} while(rc > 0);
libssh2_sftp_close(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
if(local)
fclose(local);
fprintf(stderr, "all done\n");
libssh2_exit();
return 0;
}
《openssl和libssh2在linux環境的編譯過程》
openssl-1.0.2u編譯
下載地址:https://www.openssl.org/source/
1. 配置
./config --prefix=/home/work/openssl-1.0.2u/build
2. 動態編譯需要修改makefile
SHARED_LIBS=libcrypto.so libssl.so
CFLAG加入 "-DOPENSSL_NO_STATIC_ENGINE -fPIC"
3. 編譯中還會報如下錯誤:
make[2]: Entering directory `/home/work/openssl-1.0.2u/test'
rc4test.o: In function `main':
rc4test.c:(.text.startup+0x3b): undefined reference to `OPENSSL_cpuid_setup'
collect2: error: ld returned 1 exit status
只要在test/rc4test.c將報錯的地方註釋掉即可(沒關係!只是測試代碼)
4. 編譯
make
5. 安裝
make install
libssh2-1.9.0
下載地址:https://www.libssh2.org/
./configure --prefix=`pwd`/build --with-openssl CPPFLAGS="-I/home/work/openssl-1.0.2u/build/include/openssl" LDFLAGS="-ldl -L/home/work/openssl-1.0.2u/build/lib" --enable-static=no
make
make install
把上面的源代碼,放在libssh2-1.9.0/example/目錄下,然後在該目錄下執行make進行編譯。
執行make V=1可以輸出如下完整信息(其實上面的源碼就是ssh2的示例程序,不過我在上面增加了創建文件目錄的功能):
gcc -DHAVE_CONFIG_H -I../include -I../example -I/home/work/openssl-1.0.2u/build/include/openssl -I/home/work/openssl-1.0.2u/build/include -g -O2 -MT sftp_write.o -MD -MP -MF .deps/sftp_write.Tpo -c -o sftp_write.o sftp_write.c
mv -f .deps/sftp_write.Tpo .deps/sftp_write.Po
/bin/sh ../libtool --tag=CC --mode=link gcc -g -O2 -ldl -L/home/work/openssl-1.0.2u/build/lib -o sftp_write sftp_write.o ../src/libssh2.la
libtool: link: gcc -g -O2 -o .libs/sftp_write sftp_write.o -ldl -L/home/work/openssl-1.0.2u/build/lib ../src/.libs/libssh2.so -Wl,-rpath -Wl,/home/work/libssh2-1.9.0/build/lib
運行測試用例(把本地文件readme.txt,上傳到sftp服務器):
# sftp服務器ip:192.168.3.79 用戶:test 用戶密碼:123456 本地文件:readme.txt 目標文件:/tmp/readme.txt
./sftp_write 192.168.3.79 test 123456 readme.txt /tmp/readme.txt