1 #include "config.h"
2
3 #define LIBSSH_STATIC
4
5 #include "torture.h"
6 #include "sftp.c"
7
8 #include <sys/types.h>
9 #include <pwd.h>
10 #include <errno.h>
11
sshd_setup(void ** state)12 static int sshd_setup(void **state)
13 {
14 torture_setup_sshd_server(state, false);
15
16 return 0;
17 }
18
sshd_teardown(void ** state)19 static int sshd_teardown(void **state) {
20 torture_teardown_sshd_server(state);
21
22 return 0;
23 }
24
session_setup(void ** state)25 static int session_setup(void **state)
26 {
27 struct torture_state *s = *state;
28 struct passwd *pwd;
29 int rc;
30
31 pwd = getpwnam("bob");
32 assert_non_null(pwd);
33
34 rc = setuid(pwd->pw_uid);
35 assert_return_code(rc, errno);
36
37 s->ssh.session = torture_ssh_session(s,
38 TORTURE_SSH_SERVER,
39 NULL,
40 TORTURE_SSH_USER_ALICE,
41 NULL);
42 assert_non_null(s->ssh.session);
43
44 s->ssh.tsftp = torture_sftp_session(s->ssh.session);
45 assert_non_null(s->ssh.tsftp);
46
47 return 0;
48 }
49
session_teardown(void ** state)50 static int session_teardown(void **state)
51 {
52 struct torture_state *s = *state;
53
54 torture_rmdirs(s->ssh.tsftp->testdir);
55 torture_sftp_close(s->ssh.tsftp);
56 ssh_disconnect(s->ssh.session);
57 ssh_free(s->ssh.session);
58
59 return 0;
60 }
61
torture_sftp_mkdir(void ** state)62 static void torture_sftp_mkdir(void **state) {
63 struct torture_state *s = *state;
64
65 struct torture_sftp *t = s->ssh.tsftp;
66 char tmpdir[128] = {0};
67 int rc;
68
69 assert_non_null(t);
70
71 snprintf(tmpdir, sizeof(tmpdir) - 1, "%s/mkdir_test", t->testdir);
72
73 rc = sftp_mkdir(t->sftp, tmpdir, 0755);
74 if(rc != SSH_OK)
75 fprintf(stderr,"error:%s\n",ssh_get_error(t->sftp->session));
76 assert_true(rc == 0);
77
78 /* check if it really has been created */
79 assert_true(torture_isdir(tmpdir));
80
81 rc = sftp_rmdir(t->sftp, tmpdir);
82 assert_true(rc == 0);
83
84 /* check if it has been deleted */
85 assert_false(torture_isdir(tmpdir));
86 }
87
torture_run_tests(void)88 int torture_run_tests(void) {
89 int rc;
90 struct CMUnitTest tests[] = {
91 cmocka_unit_test_setup_teardown(torture_sftp_mkdir,
92 session_setup,
93 session_teardown)
94 };
95
96 ssh_init();
97
98 torture_filter_tests(tests);
99 rc = cmocka_run_group_tests(tests, sshd_setup, sshd_teardown);
100
101 ssh_finalize();
102
103 return rc;
104 }
105