mirror of
https://github.com/tmate-io/tmate-ssh-server.git
synced 2020-11-18 19:53:51 -08:00
Forgot some libssh files
This commit is contained in:
Nicolas Viennot
parent
1d16931564
commit
6ec03881a3
61
libssh/cmake/Modules/FindNaCl.cmake
Normal file
61
libssh/cmake/Modules/FindNaCl.cmake
Normal file
@ -0,0 +1,61 @@
|
|||||||
|
# - Try to find NaCl
|
||||||
|
# Once done this will define
|
||||||
|
#
|
||||||
|
# NACL_FOUND - system has NaCl
|
||||||
|
# NACL_INCLUDE_DIRS - the NaCl include directory
|
||||||
|
# NACL_LIBRARIES - Link these to use NaCl
|
||||||
|
# NACL_DEFINITIONS - Compiler switches required for using NaCl
|
||||||
|
#
|
||||||
|
# Copyright (c) 2010 Andreas Schneider <asn@cynapses.org>
|
||||||
|
# Copyright (c) 2013 Aris Adamantiadis <aris@badcode.be>
|
||||||
|
#
|
||||||
|
# Redistribution and use is allowed according to the terms of the New
|
||||||
|
# BSD license.
|
||||||
|
# For details see the accompanying COPYING-CMAKE-SCRIPTS file.
|
||||||
|
#
|
||||||
|
|
||||||
|
|
||||||
|
if (NACL_LIBRARIES AND NACL_INCLUDE_DIRS)
|
||||||
|
# in cache already
|
||||||
|
set(NACL_FOUND TRUE)
|
||||||
|
else (NACL_LIBRARIES AND NACL_INCLUDE_DIRS)
|
||||||
|
|
||||||
|
find_path(NACL_INCLUDE_DIR
|
||||||
|
NAMES
|
||||||
|
nacl/crypto_box_curve25519xsalsa20poly1305.h
|
||||||
|
PATHS
|
||||||
|
/usr/include
|
||||||
|
/usr/local/include
|
||||||
|
/opt/local/include
|
||||||
|
/sw/include
|
||||||
|
)
|
||||||
|
|
||||||
|
find_library(NACL_LIBRARY
|
||||||
|
NAMES
|
||||||
|
nacl
|
||||||
|
PATHS
|
||||||
|
/usr/lib
|
||||||
|
/usr/local/lib
|
||||||
|
/opt/local/lib
|
||||||
|
/sw/lib
|
||||||
|
)
|
||||||
|
|
||||||
|
set(NACL_INCLUDE_DIRS
|
||||||
|
${NACL_INCLUDE_DIR}
|
||||||
|
)
|
||||||
|
|
||||||
|
if (NACL_LIBRARY)
|
||||||
|
set(NACL_LIBRARIES
|
||||||
|
${NACL_LIBRARIES}
|
||||||
|
${NACL_LIBRARY}
|
||||||
|
)
|
||||||
|
endif (NACL_LIBRARY)
|
||||||
|
|
||||||
|
include(FindPackageHandleStandardArgs)
|
||||||
|
find_package_handle_standard_args(NaCl DEFAULT_MSG NACL_LIBRARIES NACL_INCLUDE_DIRS)
|
||||||
|
|
||||||
|
# show the NACL_INCLUDE_DIRS and NACL_LIBRARIES variables only in the advanced view
|
||||||
|
mark_as_advanced(NACL_INCLUDE_DIRS NACL_LIBRARIES)
|
||||||
|
|
||||||
|
endif (NACL_LIBRARIES AND NACL_INCLUDE_DIRS)
|
||||||
|
|
||||||
272
libssh/src/curve25519_ref.c
Normal file
272
libssh/src/curve25519_ref.c
Normal file
@ -0,0 +1,272 @@
|
|||||||
|
/*
|
||||||
|
version 20081011
|
||||||
|
Matthew Dempsky
|
||||||
|
Public domain.
|
||||||
|
Derived from public domain code by D. J. Bernstein.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "libssh/curve25519.h"
|
||||||
|
static const unsigned char base[32] = {9};
|
||||||
|
|
||||||
|
int crypto_scalarmult_base(unsigned char *q,
|
||||||
|
const unsigned char *n)
|
||||||
|
{
|
||||||
|
return crypto_scalarmult(q,n,base);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void add(unsigned int out[32],const unsigned int a[32],const unsigned int b[32])
|
||||||
|
{
|
||||||
|
unsigned int j;
|
||||||
|
unsigned int u;
|
||||||
|
u = 0;
|
||||||
|
for (j = 0;j < 31;++j) { u += a[j] + b[j]; out[j] = u & 255; u >>= 8; }
|
||||||
|
u += a[31] + b[31]; out[31] = u;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void sub(unsigned int out[32],const unsigned int a[32],const unsigned int b[32])
|
||||||
|
{
|
||||||
|
unsigned int j;
|
||||||
|
unsigned int u;
|
||||||
|
u = 218;
|
||||||
|
for (j = 0;j < 31;++j) {
|
||||||
|
u += a[j] + 65280 - b[j];
|
||||||
|
out[j] = u & 255;
|
||||||
|
u >>= 8;
|
||||||
|
}
|
||||||
|
u += a[31] - b[31];
|
||||||
|
out[31] = u;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void squeeze(unsigned int a[32])
|
||||||
|
{
|
||||||
|
unsigned int j;
|
||||||
|
unsigned int u;
|
||||||
|
u = 0;
|
||||||
|
for (j = 0;j < 31;++j) { u += a[j]; a[j] = u & 255; u >>= 8; }
|
||||||
|
u += a[31]; a[31] = u & 127;
|
||||||
|
u = 19 * (u >> 7);
|
||||||
|
for (j = 0;j < 31;++j) { u += a[j]; a[j] = u & 255; u >>= 8; }
|
||||||
|
u += a[31]; a[31] = u;
|
||||||
|
}
|
||||||
|
|
||||||
|
static const unsigned int minusp[32] = {
|
||||||
|
19, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 128
|
||||||
|
} ;
|
||||||
|
|
||||||
|
static void freeze(unsigned int a[32])
|
||||||
|
{
|
||||||
|
unsigned int aorig[32];
|
||||||
|
unsigned int j;
|
||||||
|
unsigned int negative;
|
||||||
|
|
||||||
|
for (j = 0;j < 32;++j) aorig[j] = a[j];
|
||||||
|
add(a,a,minusp);
|
||||||
|
negative = -((a[31] >> 7) & 1);
|
||||||
|
for (j = 0;j < 32;++j) a[j] ^= negative & (aorig[j] ^ a[j]);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void mult(unsigned int out[32],const unsigned int a[32],const unsigned int b[32])
|
||||||
|
{
|
||||||
|
unsigned int i;
|
||||||
|
unsigned int j;
|
||||||
|
unsigned int u;
|
||||||
|
|
||||||
|
for (i = 0;i < 32;++i) {
|
||||||
|
u = 0;
|
||||||
|
for (j = 0;j <= i;++j) u += a[j] * b[i - j];
|
||||||
|
for (j = i + 1;j < 32;++j) u += 38 * a[j] * b[i + 32 - j];
|
||||||
|
out[i] = u;
|
||||||
|
}
|
||||||
|
squeeze(out);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void mult121665(unsigned int out[32],const unsigned int a[32])
|
||||||
|
{
|
||||||
|
unsigned int j;
|
||||||
|
unsigned int u;
|
||||||
|
|
||||||
|
u = 0;
|
||||||
|
for (j = 0;j < 31;++j) { u += 121665 * a[j]; out[j] = u & 255; u >>= 8; }
|
||||||
|
u += 121665 * a[31]; out[31] = u & 127;
|
||||||
|
u = 19 * (u >> 7);
|
||||||
|
for (j = 0;j < 31;++j) { u += out[j]; out[j] = u & 255; u >>= 8; }
|
||||||
|
u += out[j]; out[j] = u;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void square(unsigned int out[32],const unsigned int a[32])
|
||||||
|
{
|
||||||
|
unsigned int i;
|
||||||
|
unsigned int j;
|
||||||
|
unsigned int u;
|
||||||
|
|
||||||
|
for (i = 0;i < 32;++i) {
|
||||||
|
u = 0;
|
||||||
|
for (j = 0;j < i - j;++j) u += a[j] * a[i - j];
|
||||||
|
for (j = i + 1;j < i + 32 - j;++j) u += 38 * a[j] * a[i + 32 - j];
|
||||||
|
u *= 2;
|
||||||
|
if ((i & 1) == 0) {
|
||||||
|
u += a[i / 2] * a[i / 2];
|
||||||
|
u += 38 * a[i / 2 + 16] * a[i / 2 + 16];
|
||||||
|
}
|
||||||
|
out[i] = u;
|
||||||
|
}
|
||||||
|
squeeze(out);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void c_select(unsigned int p[64],unsigned int q[64],const unsigned int r[64],const unsigned int s[64],unsigned int b)
|
||||||
|
{
|
||||||
|
unsigned int j;
|
||||||
|
unsigned int t;
|
||||||
|
unsigned int bminus1;
|
||||||
|
|
||||||
|
bminus1 = b - 1;
|
||||||
|
for (j = 0;j < 64;++j) {
|
||||||
|
t = bminus1 & (r[j] ^ s[j]);
|
||||||
|
p[j] = s[j] ^ t;
|
||||||
|
q[j] = r[j] ^ t;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void mainloop(unsigned int work[64],const unsigned char e[32])
|
||||||
|
{
|
||||||
|
unsigned int xzm1[64];
|
||||||
|
unsigned int xzm[64];
|
||||||
|
unsigned int xzmb[64];
|
||||||
|
unsigned int xzm1b[64];
|
||||||
|
unsigned int xznb[64];
|
||||||
|
unsigned int xzn1b[64];
|
||||||
|
unsigned int a0[64];
|
||||||
|
unsigned int a1[64];
|
||||||
|
unsigned int b0[64];
|
||||||
|
unsigned int b1[64];
|
||||||
|
unsigned int c1[64];
|
||||||
|
unsigned int r[32];
|
||||||
|
unsigned int s[32];
|
||||||
|
unsigned int t[32];
|
||||||
|
unsigned int u[32];
|
||||||
|
unsigned int j;
|
||||||
|
unsigned int b;
|
||||||
|
int pos;
|
||||||
|
|
||||||
|
for (j = 0;j < 32;++j) xzm1[j] = work[j];
|
||||||
|
xzm1[32] = 1;
|
||||||
|
for (j = 33;j < 64;++j) xzm1[j] = 0;
|
||||||
|
|
||||||
|
xzm[0] = 1;
|
||||||
|
for (j = 1;j < 64;++j) xzm[j] = 0;
|
||||||
|
|
||||||
|
for (pos = 254;pos >= 0;--pos) {
|
||||||
|
b = e[pos / 8] >> (pos & 7);
|
||||||
|
b &= 1;
|
||||||
|
c_select(xzmb,xzm1b,xzm,xzm1,b);
|
||||||
|
add(a0,xzmb,xzmb + 32);
|
||||||
|
sub(a0 + 32,xzmb,xzmb + 32);
|
||||||
|
add(a1,xzm1b,xzm1b + 32);
|
||||||
|
sub(a1 + 32,xzm1b,xzm1b + 32);
|
||||||
|
square(b0,a0);
|
||||||
|
square(b0 + 32,a0 + 32);
|
||||||
|
mult(b1,a1,a0 + 32);
|
||||||
|
mult(b1 + 32,a1 + 32,a0);
|
||||||
|
add(c1,b1,b1 + 32);
|
||||||
|
sub(c1 + 32,b1,b1 + 32);
|
||||||
|
square(r,c1 + 32);
|
||||||
|
sub(s,b0,b0 + 32);
|
||||||
|
mult121665(t,s);
|
||||||
|
add(u,t,b0);
|
||||||
|
mult(xznb,b0,b0 + 32);
|
||||||
|
mult(xznb + 32,s,u);
|
||||||
|
square(xzn1b,c1);
|
||||||
|
mult(xzn1b + 32,r,work);
|
||||||
|
c_select(xzm,xzm1,xznb,xzn1b,b);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (j = 0;j < 64;++j) work[j] = xzm[j];
|
||||||
|
}
|
||||||
|
|
||||||
|
static void recip(unsigned int out[32],const unsigned int z[32])
|
||||||
|
{
|
||||||
|
unsigned int z2[32];
|
||||||
|
unsigned int z9[32];
|
||||||
|
unsigned int z11[32];
|
||||||
|
unsigned int z2_5_0[32];
|
||||||
|
unsigned int z2_10_0[32];
|
||||||
|
unsigned int z2_20_0[32];
|
||||||
|
unsigned int z2_50_0[32];
|
||||||
|
unsigned int z2_100_0[32];
|
||||||
|
unsigned int t0[32];
|
||||||
|
unsigned int t1[32];
|
||||||
|
int i;
|
||||||
|
|
||||||
|
/* 2 */ square(z2,z);
|
||||||
|
/* 4 */ square(t1,z2);
|
||||||
|
/* 8 */ square(t0,t1);
|
||||||
|
/* 9 */ mult(z9,t0,z);
|
||||||
|
/* 11 */ mult(z11,z9,z2);
|
||||||
|
/* 22 */ square(t0,z11);
|
||||||
|
/* 2^5 - 2^0 = 31 */ mult(z2_5_0,t0,z9);
|
||||||
|
|
||||||
|
/* 2^6 - 2^1 */ square(t0,z2_5_0);
|
||||||
|
/* 2^7 - 2^2 */ square(t1,t0);
|
||||||
|
/* 2^8 - 2^3 */ square(t0,t1);
|
||||||
|
/* 2^9 - 2^4 */ square(t1,t0);
|
||||||
|
/* 2^10 - 2^5 */ square(t0,t1);
|
||||||
|
/* 2^10 - 2^0 */ mult(z2_10_0,t0,z2_5_0);
|
||||||
|
|
||||||
|
/* 2^11 - 2^1 */ square(t0,z2_10_0);
|
||||||
|
/* 2^12 - 2^2 */ square(t1,t0);
|
||||||
|
/* 2^20 - 2^10 */ for (i = 2;i < 10;i += 2) { square(t0,t1); square(t1,t0); }
|
||||||
|
/* 2^20 - 2^0 */ mult(z2_20_0,t1,z2_10_0);
|
||||||
|
|
||||||
|
/* 2^21 - 2^1 */ square(t0,z2_20_0);
|
||||||
|
/* 2^22 - 2^2 */ square(t1,t0);
|
||||||
|
/* 2^40 - 2^20 */ for (i = 2;i < 20;i += 2) { square(t0,t1); square(t1,t0); }
|
||||||
|
/* 2^40 - 2^0 */ mult(t0,t1,z2_20_0);
|
||||||
|
|
||||||
|
/* 2^41 - 2^1 */ square(t1,t0);
|
||||||
|
/* 2^42 - 2^2 */ square(t0,t1);
|
||||||
|
/* 2^50 - 2^10 */ for (i = 2;i < 10;i += 2) { square(t1,t0); square(t0,t1); }
|
||||||
|
/* 2^50 - 2^0 */ mult(z2_50_0,t0,z2_10_0);
|
||||||
|
|
||||||
|
/* 2^51 - 2^1 */ square(t0,z2_50_0);
|
||||||
|
/* 2^52 - 2^2 */ square(t1,t0);
|
||||||
|
/* 2^100 - 2^50 */ for (i = 2;i < 50;i += 2) { square(t0,t1); square(t1,t0); }
|
||||||
|
/* 2^100 - 2^0 */ mult(z2_100_0,t1,z2_50_0);
|
||||||
|
|
||||||
|
/* 2^101 - 2^1 */ square(t1,z2_100_0);
|
||||||
|
/* 2^102 - 2^2 */ square(t0,t1);
|
||||||
|
/* 2^200 - 2^100 */ for (i = 2;i < 100;i += 2) { square(t1,t0); square(t0,t1); }
|
||||||
|
/* 2^200 - 2^0 */ mult(t1,t0,z2_100_0);
|
||||||
|
|
||||||
|
/* 2^201 - 2^1 */ square(t0,t1);
|
||||||
|
/* 2^202 - 2^2 */ square(t1,t0);
|
||||||
|
/* 2^250 - 2^50 */ for (i = 2;i < 50;i += 2) { square(t0,t1); square(t1,t0); }
|
||||||
|
/* 2^250 - 2^0 */ mult(t0,t1,z2_50_0);
|
||||||
|
|
||||||
|
/* 2^251 - 2^1 */ square(t1,t0);
|
||||||
|
/* 2^252 - 2^2 */ square(t0,t1);
|
||||||
|
/* 2^253 - 2^3 */ square(t1,t0);
|
||||||
|
/* 2^254 - 2^4 */ square(t0,t1);
|
||||||
|
/* 2^255 - 2^5 */ square(t1,t0);
|
||||||
|
/* 2^255 - 21 */ mult(out,t1,z11);
|
||||||
|
}
|
||||||
|
|
||||||
|
int crypto_scalarmult(unsigned char *q,
|
||||||
|
const unsigned char *n,
|
||||||
|
const unsigned char *p)
|
||||||
|
{
|
||||||
|
unsigned int work[96];
|
||||||
|
unsigned char e[32];
|
||||||
|
unsigned int i;
|
||||||
|
for (i = 0;i < 32;++i) e[i] = n[i];
|
||||||
|
e[0] &= 248;
|
||||||
|
e[31] &= 127;
|
||||||
|
e[31] |= 64;
|
||||||
|
for (i = 0;i < 32;++i) work[i] = p[i];
|
||||||
|
mainloop(work,e);
|
||||||
|
recip(work + 32,work + 32);
|
||||||
|
mult(work + 64,work,work + 32);
|
||||||
|
freeze(work + 64);
|
||||||
|
for (i = 0;i < 32;++i) q[i] = work[64 + i];
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
94
libssh/tests/client/torture_forward.c
Normal file
94
libssh/tests/client/torture_forward.c
Normal file
@ -0,0 +1,94 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the SSH Library
|
||||||
|
*
|
||||||
|
* Copyright (c) 2013 by Andreas Schneider <asn@cryptomilk.org>
|
||||||
|
*
|
||||||
|
* The SSH Library is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU Lesser General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2.1 of the License, or (at your
|
||||||
|
* option) any later version.
|
||||||
|
*
|
||||||
|
* The SSH Library is distributed in the hope that it will be useful, but
|
||||||
|
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
||||||
|
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
|
||||||
|
* License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Lesser General Public License
|
||||||
|
* along with the SSH Library; see the file COPYING. If not, write to
|
||||||
|
* the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
|
||||||
|
* MA 02111-1307, USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#define LIBSSH_STATIC
|
||||||
|
|
||||||
|
#include "torture.h"
|
||||||
|
#include <libssh/libssh.h>
|
||||||
|
|
||||||
|
static void setup(void **state)
|
||||||
|
{
|
||||||
|
ssh_session session;
|
||||||
|
const char *host;
|
||||||
|
const char *user;
|
||||||
|
const char *password;
|
||||||
|
|
||||||
|
host = getenv("TORTURE_HOST");
|
||||||
|
if (host == NULL) {
|
||||||
|
host = "localhost";
|
||||||
|
}
|
||||||
|
|
||||||
|
user = getenv("TORTURE_USER");
|
||||||
|
password = getenv("TORTURE_PASSWORD");
|
||||||
|
|
||||||
|
session = torture_ssh_session(host, user, password);
|
||||||
|
|
||||||
|
assert_false(session == NULL);
|
||||||
|
*state = session;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void teardown(void **state)
|
||||||
|
{
|
||||||
|
ssh_session session = *state;
|
||||||
|
|
||||||
|
assert_false(session == NULL);
|
||||||
|
|
||||||
|
if (ssh_is_connected(session)) {
|
||||||
|
ssh_disconnect(session);
|
||||||
|
}
|
||||||
|
ssh_free(session);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void torture_ssh_forward(void **state)
|
||||||
|
{
|
||||||
|
ssh_session session = *state;
|
||||||
|
#if 0
|
||||||
|
ssh_channel c;
|
||||||
|
#endif
|
||||||
|
int bound_port;
|
||||||
|
int rc;
|
||||||
|
|
||||||
|
rc = ssh_forward_listen(session, "127.0.0.1", 8080, &bound_port);
|
||||||
|
assert_int_equal(rc, SSH_OK);
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
c = ssh_forward_accept(session, 60000);
|
||||||
|
assert_non_null(c);
|
||||||
|
|
||||||
|
ssh_channel_send_eof(c);
|
||||||
|
ssh_channel_close(c);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
int torture_run_tests(void) {
|
||||||
|
int rc;
|
||||||
|
|
||||||
|
const UnitTest tests[] = {
|
||||||
|
unit_test_setup_teardown(torture_ssh_forward, setup, teardown),
|
||||||
|
};
|
||||||
|
|
||||||
|
ssh_init();
|
||||||
|
|
||||||
|
rc = run_tests(tests);
|
||||||
|
|
||||||
|
ssh_finalize();
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
114
libssh/tests/client/torture_request_env.c
Normal file
114
libssh/tests/client/torture_request_env.c
Normal file
@ -0,0 +1,114 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of the SSH Library
|
||||||
|
*
|
||||||
|
* Copyright (c) 2013 by Andreas Schneider <asn@cryptomilk.org>
|
||||||
|
*
|
||||||
|
* The SSH Library is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU Lesser General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2.1 of the License, or (at your
|
||||||
|
* option) any later version.
|
||||||
|
*
|
||||||
|
* The SSH Library is distributed in the hope that it will be useful, but
|
||||||
|
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
||||||
|
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
|
||||||
|
* License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Lesser General Public License
|
||||||
|
* along with the SSH Library; see the file COPYING. If not, write to
|
||||||
|
* the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
|
||||||
|
* MA 02111-1307, USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#define LIBSSH_STATIC
|
||||||
|
|
||||||
|
#include "torture.h"
|
||||||
|
#include <libssh/libssh.h>
|
||||||
|
|
||||||
|
static void setup(void **state)
|
||||||
|
{
|
||||||
|
ssh_session session;
|
||||||
|
const char *host;
|
||||||
|
const char *user;
|
||||||
|
const char *password;
|
||||||
|
|
||||||
|
host = getenv("TORTURE_HOST");
|
||||||
|
if (host == NULL) {
|
||||||
|
host = "localhost";
|
||||||
|
}
|
||||||
|
|
||||||
|
user = getenv("TORTURE_USER");
|
||||||
|
password = getenv("TORTURE_PASSWORD");
|
||||||
|
|
||||||
|
session = torture_ssh_session(host, user, password);
|
||||||
|
|
||||||
|
assert_false(session == NULL);
|
||||||
|
*state = session;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void teardown(void **state)
|
||||||
|
{
|
||||||
|
ssh_session session = *state;
|
||||||
|
|
||||||
|
assert_false(session == NULL);
|
||||||
|
|
||||||
|
if (ssh_is_connected(session)) {
|
||||||
|
ssh_disconnect(session);
|
||||||
|
}
|
||||||
|
ssh_free(session);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void torture_request_env(void **state)
|
||||||
|
{
|
||||||
|
ssh_session session = *state;
|
||||||
|
ssh_channel c;
|
||||||
|
char buffer[4096];
|
||||||
|
int nbytes;
|
||||||
|
int rc;
|
||||||
|
int lang_found = 0;
|
||||||
|
|
||||||
|
c = ssh_channel_new(session);
|
||||||
|
assert_non_null(c);
|
||||||
|
|
||||||
|
rc = ssh_channel_open_session(c);
|
||||||
|
assert_int_equal(rc, SSH_OK);
|
||||||
|
|
||||||
|
rc = ssh_channel_request_env(c, "LANG", "LIBSSH");
|
||||||
|
assert_int_equal(rc, SSH_OK);
|
||||||
|
|
||||||
|
rc = ssh_channel_request_exec(c, "bash -c export");
|
||||||
|
assert_int_equal(rc, SSH_OK);
|
||||||
|
|
||||||
|
nbytes = ssh_channel_read(c, buffer, sizeof(buffer), 0);
|
||||||
|
while (nbytes > 0) {
|
||||||
|
#if 0
|
||||||
|
rc = fwrite(buffer, 1, nbytes, stdout);
|
||||||
|
assert_int_equal(rc, nbytes);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
if (strstr(buffer, "LANG=\"LIBSSH\"")) {
|
||||||
|
lang_found = 1;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
nbytes = ssh_channel_read(c, buffer, sizeof(buffer), 0);
|
||||||
|
}
|
||||||
|
assert_int_equal(lang_found, 1);
|
||||||
|
|
||||||
|
ssh_channel_close(c);
|
||||||
|
}
|
||||||
|
|
||||||
|
int torture_run_tests(void) {
|
||||||
|
int rc;
|
||||||
|
|
||||||
|
const UnitTest tests[] = {
|
||||||
|
unit_test_setup_teardown(torture_request_env, setup, teardown),
|
||||||
|
};
|
||||||
|
|
||||||
|
ssh_init();
|
||||||
|
|
||||||
|
rc = run_tests(tests);
|
||||||
|
|
||||||
|
ssh_finalize();
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
|
|
||||||
83
libssh/tests/client/torture_sftp_read.c
Normal file
83
libssh/tests/client/torture_sftp_read.c
Normal file
@ -0,0 +1,83 @@
|
|||||||
|
#define LIBSSH_STATIC
|
||||||
|
|
||||||
|
#include "torture.h"
|
||||||
|
#include "sftp.c"
|
||||||
|
|
||||||
|
#define MAX_XFER_BUF_SIZE 16384
|
||||||
|
|
||||||
|
static void setup(void **state) {
|
||||||
|
ssh_session session;
|
||||||
|
struct torture_sftp *t;
|
||||||
|
const char *host;
|
||||||
|
const char *user;
|
||||||
|
const char *password;
|
||||||
|
|
||||||
|
host = getenv("TORTURE_HOST");
|
||||||
|
if (host == NULL) {
|
||||||
|
host = "localhost";
|
||||||
|
}
|
||||||
|
|
||||||
|
user = getenv("TORTURE_USER");
|
||||||
|
password = getenv("TORTURE_PASSWORD");
|
||||||
|
|
||||||
|
session = torture_ssh_session(host, user, password);
|
||||||
|
assert_false(session == NULL);
|
||||||
|
t = torture_sftp_session(session);
|
||||||
|
assert_false(t == NULL);
|
||||||
|
|
||||||
|
*state = t;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void teardown(void **state) {
|
||||||
|
struct torture_sftp *t = *state;
|
||||||
|
|
||||||
|
assert_false(t == NULL);
|
||||||
|
|
||||||
|
torture_rmdirs(t->testdir);
|
||||||
|
torture_sftp_close(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void torture_sftp_read_blocking(void **state) {
|
||||||
|
struct torture_sftp *t = *state;
|
||||||
|
char libssh_tmp_file[] = "/tmp/libssh_sftp_test_XXXXXX";
|
||||||
|
char buf[MAX_XFER_BUF_SIZE];
|
||||||
|
ssize_t bytesread;
|
||||||
|
ssize_t byteswritten;
|
||||||
|
int fd;
|
||||||
|
sftp_file file;
|
||||||
|
|
||||||
|
|
||||||
|
file = sftp_open(t->sftp, "/usr/bin/ssh", O_RDONLY, 0);
|
||||||
|
assert_non_null(file);
|
||||||
|
|
||||||
|
fd = mkstemp(libssh_tmp_file);
|
||||||
|
unlink(libssh_tmp_file);
|
||||||
|
|
||||||
|
for (;;) {
|
||||||
|
bytesread = sftp_read(file, buf, MAX_XFER_BUF_SIZE);
|
||||||
|
if (bytesread == 0) {
|
||||||
|
break; /* EOF */
|
||||||
|
}
|
||||||
|
assert_false(bytesread < 0);
|
||||||
|
|
||||||
|
byteswritten = write(fd, buf, bytesread);
|
||||||
|
assert_int_equal(byteswritten, bytesread);
|
||||||
|
}
|
||||||
|
|
||||||
|
close(fd);
|
||||||
|
sftp_close(file);
|
||||||
|
}
|
||||||
|
|
||||||
|
int torture_run_tests(void) {
|
||||||
|
int rc;
|
||||||
|
const UnitTest tests[] = {
|
||||||
|
unit_test_setup_teardown(torture_sftp_read_blocking, setup, teardown)
|
||||||
|
};
|
||||||
|
|
||||||
|
ssh_init();
|
||||||
|
|
||||||
|
rc = run_tests(tests);
|
||||||
|
ssh_finalize();
|
||||||
|
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
48
libssh/tests/unittests/torture_channel.c
Normal file
48
libssh/tests/unittests/torture_channel.c
Normal file
@ -0,0 +1,48 @@
|
|||||||
|
#define LIBSSH_STATIC
|
||||||
|
#include <libssh/priv.h>
|
||||||
|
|
||||||
|
#include <sys/types.h>
|
||||||
|
#include <sys/stat.h>
|
||||||
|
#include <fcntl.h>
|
||||||
|
|
||||||
|
#include "torture.h"
|
||||||
|
#include "channels.c"
|
||||||
|
|
||||||
|
static void torture_channel_select(void **state)
|
||||||
|
{
|
||||||
|
fd_set readfds;
|
||||||
|
int fd;
|
||||||
|
int rc;
|
||||||
|
int i;
|
||||||
|
|
||||||
|
(void)state; /* unused */
|
||||||
|
|
||||||
|
fd = open("/dev/null", 0);
|
||||||
|
assert_true(fd > 2);
|
||||||
|
|
||||||
|
FD_SET(fd, &readfds);
|
||||||
|
|
||||||
|
for (i = 0; i < 10; i++) {
|
||||||
|
ssh_channel cin[1] = { NULL, };
|
||||||
|
ssh_channel cout[1] = { NULL, };
|
||||||
|
struct timeval tv = { .tv_sec = 0, .tv_usec = 1000 };
|
||||||
|
|
||||||
|
rc = ssh_select(cin, cout, fd + 1, &readfds, &tv);
|
||||||
|
assert_int_equal(rc, SSH_OK);
|
||||||
|
}
|
||||||
|
|
||||||
|
close(fd);
|
||||||
|
}
|
||||||
|
|
||||||
|
int torture_run_tests(void) {
|
||||||
|
int rc;
|
||||||
|
const UnitTest tests[] = {
|
||||||
|
unit_test(torture_channel_select),
|
||||||
|
};
|
||||||
|
|
||||||
|
ssh_init();
|
||||||
|
rc = run_tests(tests);
|
||||||
|
ssh_finalize();
|
||||||
|
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user