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