aboutsummaryrefslogtreecommitdiff
path: root/src/server_ssh.c
blob: 9f246572079c78e50d80940187d76874cad0a2de (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#include <stdlib.h>
#include <assert.h>
#include <libssh/callbacks.h>
#include <libssh/server.h>

#include "server_ssh.h"
#include "server.h"
#include "log.h"

struct server_callbacks potd_ssh_callbacks = {
    .on_listen = ssh_on_listen,
    .on_shutdown = ssh_on_shutdown
};

static int set_default_keys(ssh_bind sshbind, int rsa_already_set,
                            int dsa_already_set, int ecdsa_already_set);


int ssh_init_cb(struct server_ctx *ctx)
{
    ctx->server_cbs = potd_ssh_callbacks;

    if (ssh_init())
        return 1;
    ssh_data *d = (ssh_data *) calloc(1, sizeof(*d));
    assert(d);
    d->sshbind = ssh_bind_new();
    d->session = ssh_new();
    ctx->server_dat.data = d;
    if (!d->sshbind || !d->session)
        return 1;
    if (set_default_keys(d->sshbind, 0, 0, 0))
        return 1;

    return 0;
}

int ssh_on_listen(struct server_data *data)
{
    ssh_data *d = (ssh_data *) data->data;

    if (ssh_bind_listen(d->sshbind) < 0) {
        E("Error listening to SSH socket: %s", ssh_get_error(d->sshbind));
    }
    return 0;
}

int ssh_on_shutdown(struct server_data *data)
{
    return 0;
}

static int set_default_keys(ssh_bind sshbind, int rsa_already_set,
                             int dsa_already_set, int ecdsa_already_set)
{
    if (!rsa_already_set) {
        if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_RSAKEY,
                                 "./ssh_host_rsa_key"))
            return 1;
    }
    if (!dsa_already_set) {
        if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_DSAKEY,
                                 "./ssh_host_dsa_key"))
            return 1;
    }
    if (!ecdsa_already_set) {
        if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_ECDSAKEY,
                                 "./ssh_host_ecdsa_key"))
            return 1;
    }
    return 0;
}