working on adding a entity workflow
This commit is contained in:
280
source/ssh_server_client.c
Normal file
280
source/ssh_server_client.c
Normal file
@@ -0,0 +1,280 @@
|
||||
|
||||
/* #############################################################################
|
||||
* # sshGameServer
|
||||
* This is a test
|
||||
*
|
||||
* AUTHER: PreacherDHM
|
||||
* DATE: 07/03/26
|
||||
* #############################################################################
|
||||
*/
|
||||
|
||||
/* This is a sample implementation of a libssh based SSH server */
|
||||
|
||||
|
||||
#include "ssh_server_client.h"
|
||||
#include <libssh/libssh.h>
|
||||
#include <libssh/server.h>
|
||||
#include <libssh/callbacks.h>
|
||||
#include <pthread.h>
|
||||
#include <threads.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#ifdef HAVE_ARGP_H
|
||||
#include <argp.h>
|
||||
#endif
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <stdio.h>
|
||||
|
||||
|
||||
#ifndef BUF_SIZE
|
||||
#define BUF_SIZE 2049
|
||||
#endif
|
||||
|
||||
#ifndef KEYS_FOLDER
|
||||
#ifdef _WIN32
|
||||
#define KEYS_FOLDER
|
||||
#else
|
||||
#define KEYS_FOLDER "/etc/ssh/"
|
||||
#endif
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
|
||||
static int authenticated=0;
|
||||
static int tries = 0;
|
||||
static int error = 0;
|
||||
static ssh_channel chan[30] ={NULL};
|
||||
|
||||
typedef struct {
|
||||
ssh_session session;
|
||||
ServerConfig* config;
|
||||
} ud;
|
||||
|
||||
static int auth_none(ssh_session session,
|
||||
const char *user,
|
||||
void *userdata)
|
||||
{
|
||||
ssh_string banner = NULL;
|
||||
|
||||
(void)user; /* unused */
|
||||
(void)userdata; /* unused */
|
||||
|
||||
//ssh_set_auth_methods(session,
|
||||
// SSH_AUTH_METHOD_PASSWORD | SSH_AUTH_METHOD_GSSAPI_MIC);
|
||||
|
||||
banner = ssh_string_from_char("Loggin To your skelvetty acount!\n");
|
||||
if (banner != NULL) {
|
||||
ssh_send_issue_banner(session, banner);
|
||||
}
|
||||
ssh_string_free(banner);
|
||||
|
||||
authenticated ++;
|
||||
return SSH_AUTH_SUCCESS;
|
||||
}
|
||||
|
||||
static int auth_password(ssh_session session, const char *user,
|
||||
const char *password, void *userdata){
|
||||
//(void)userdata;
|
||||
//printf("Authenticating user %s pwd %s\n",user, password);
|
||||
//if(strcmp(user,USER) == 0 && strcmp(password, PASSWORD) == 0){
|
||||
// authenticated = 1;
|
||||
// printf("Authenticated\n");
|
||||
return SSH_AUTH_SUCCESS;
|
||||
//}
|
||||
//if (tries >= 3){
|
||||
// printf("Too many authentication tries\n");
|
||||
// ssh_disconnect(session);
|
||||
// error = 1;
|
||||
// return SSH_AUTH_DENIED;
|
||||
//}
|
||||
//tries++;
|
||||
//return SSH_AUTH_DENIED;
|
||||
}
|
||||
|
||||
#ifdef WITH_GSSAPI
|
||||
static int auth_gssapi_mic(ssh_session session, const char *user, const char *principal, void *userdata){
|
||||
ssh_gssapi_creds creds = ssh_gssapi_get_creds(session);
|
||||
(void)userdata;
|
||||
printf("Authenticating user %s with gssapi principal %s\n",user, principal);
|
||||
if (creds != NULL)
|
||||
printf("Received some gssapi credentials\n");
|
||||
else
|
||||
printf("Not received any forwardable creds\n");
|
||||
printf("authenticated\n");
|
||||
authenticated = 1;
|
||||
return SSH_AUTH_SUCCESS;
|
||||
}
|
||||
#endif
|
||||
|
||||
static int pty_request(ssh_session session, ssh_channel channel, const char *term,
|
||||
int x,int y, int px, int py, void *userdata){
|
||||
(void) session;
|
||||
(void) channel;
|
||||
(void) term;
|
||||
(void) x;
|
||||
(void) y;
|
||||
(void) px;
|
||||
(void) py;
|
||||
(void) userdata;
|
||||
printf("Allocated terminal\n");
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int shell_request(ssh_session session, ssh_channel channel, void *userdata){
|
||||
(void)session;
|
||||
(void)channel;
|
||||
(void)userdata;
|
||||
printf("Allocated shell\n");
|
||||
return 0;
|
||||
}
|
||||
struct ssh_channel_callbacks_struct channel_cb = {
|
||||
.channel_pty_request_function = pty_request,
|
||||
.channel_shell_request_function = shell_request
|
||||
};
|
||||
|
||||
static ssh_channel new_session_channel(ssh_session session, void *userdata){
|
||||
(void) session;
|
||||
(void) userdata;
|
||||
printf("Allocated session channel for id %d\n", *(int*)userdata);
|
||||
*(ssh_channel*)userdata = ssh_channel_new(session);
|
||||
ssh_callbacks_init(&channel_cb);
|
||||
ssh_set_channel_callbacks(*(ssh_channel*)userdata, &channel_cb);
|
||||
return *(ssh_channel*)userdata;
|
||||
}
|
||||
|
||||
|
||||
|
||||
void* Handel_Client(void* d) {
|
||||
ud data = *(ud*)d;
|
||||
char buf[BUF_SIZE];
|
||||
int r;
|
||||
int i;
|
||||
ssh_event mainloop;
|
||||
ssh_session session = data.session;
|
||||
ssh_channel c;
|
||||
|
||||
struct ssh_server_callbacks_struct cb = {
|
||||
.userdata = (void*)&c,
|
||||
.auth_none_function = auth_none,
|
||||
.auth_password_function = auth_password,
|
||||
.channel_open_request_session_function = new_session_channel,
|
||||
};
|
||||
|
||||
ssh_callbacks_init(&cb);
|
||||
ssh_set_server_callbacks(session, &cb);
|
||||
|
||||
if (ssh_handle_key_exchange(session)) {
|
||||
printf("ssh_handle_key_exchange: %s\n", ssh_get_error(session));
|
||||
return NULL;
|
||||
}
|
||||
|
||||
ssh_set_auth_methods(session, SSH_AUTH_METHOD_NONE);
|
||||
|
||||
mainloop = ssh_event_new();
|
||||
ssh_event_add_session(mainloop, session);
|
||||
|
||||
while (!(authenticated && c != NULL)){
|
||||
if(error)
|
||||
break;
|
||||
r = ssh_event_dopoll(mainloop, -1);
|
||||
if (r == SSH_ERROR){
|
||||
printf("Error : %s\n",ssh_get_error(session));
|
||||
ssh_disconnect(session);
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
char sendBuf[1024] = {0};
|
||||
int sendBuffSize = 1024;
|
||||
int counter = 0;
|
||||
if(error){
|
||||
printf("Error, exiting loop\n");
|
||||
} else
|
||||
printf("Authenticated and got a channel\n");
|
||||
data.config->cbs.ssh_init(data.config->cbs.init_var);
|
||||
do{
|
||||
i = 1;
|
||||
snprintf(sendBuf, 1024, "Counter: %d, id: %d\r", counter, data.config->id);
|
||||
char kittyBuffer[30] = {0};
|
||||
sprintf(kittyBuffer, "\e[>%d", 0b1000);
|
||||
if (ssh_channel_write(c, kittyBuffer, 30) == SSH_ERROR) {
|
||||
printf("error writing to channel\n");
|
||||
return NULL;
|
||||
}
|
||||
i = ssh_channel_read_nonblocking(c, buf, sizeof(buf)-1, 0);
|
||||
counter++;
|
||||
// code go here
|
||||
ssh_terminal_data db = {
|
||||
.inputBuffer = buf,
|
||||
.outputBuffer = sendBuf,
|
||||
.inputSize = 1024,
|
||||
.outputSize = &sendBuffSize,
|
||||
};
|
||||
data.config->cbs.ssh_run(&db,data.config->cbs.run_var);
|
||||
// code go here
|
||||
if (ssh_channel_write(c, sendBuf, 1024) == SSH_ERROR) {
|
||||
printf("error writing to channel\n");
|
||||
return NULL;
|
||||
}
|
||||
if(i>0) {
|
||||
|
||||
buf[i] = '\0';
|
||||
printf("%s", buf);
|
||||
fflush(stdout);
|
||||
if(buf[0] == '\x03') {
|
||||
ssh_disconnect(data.session);
|
||||
ssh_free(data.session);
|
||||
authenticated--;
|
||||
//ssh_channel_close(c);
|
||||
//ssh_channel_free(c);
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
} while (true);
|
||||
|
||||
data.config->cbs.ssh_stop(data.config->cbs.stop_var);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int ssh_start(ServerConfig* conf){
|
||||
ssh_session session;
|
||||
ssh_bind sshbind;
|
||||
|
||||
int port = conf->port;
|
||||
int r =0;
|
||||
|
||||
sshbind=ssh_bind_new();
|
||||
|
||||
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY, "./rsa.key");
|
||||
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDPORT, &port);
|
||||
|
||||
while(sshbind) {
|
||||
|
||||
if(ssh_bind_listen(sshbind)<0){
|
||||
printf("Error listening to socket: %s\n",ssh_get_error(sshbind));
|
||||
return 1;
|
||||
}
|
||||
session=ssh_new();
|
||||
r=ssh_bind_accept(sshbind,session);
|
||||
if(r==SSH_ERROR){
|
||||
printf("error accepting a connection : %s\n",ssh_get_error(sshbind));
|
||||
return 1;
|
||||
}
|
||||
ud d = {session, conf};
|
||||
|
||||
pthread_t thread_id;
|
||||
if(pthread_create(&thread_id, NULL, Handel_Client, (void*)&d) < 0) {
|
||||
ssh_disconnect(session);
|
||||
}
|
||||
}
|
||||
//ssh_disconnect(session);
|
||||
//ssh_bind_free(sshbind);
|
||||
//ssh_finalize();
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user