shinys000114/libssh
1
0
Fork 0
mirror of https://git.libssh.org/projects/libssh.git synced 2026-02-11 18:50:28 +09:00
Code Issues Packages Projects Releases Wiki Activity

pcap: Reformat

Browse Source 
Signed-off-by: Jakub Jelen <jjelen@redhat.com>
Reviewed-by: Andreas Schneider <asn@cryptomilk.org>
Reviewed-by: Norbert Pocs <npocs@redhat.com>
This commit is contained in:
Jakub Jelen
2023-06-27 14:54:18 +02:00
parent 19404bf509
commit 96faaeea03
1 changed files with 144 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
src/pcap.c
View File

@@ -122,10 +122,11 @@ struct ssh_pcap_file_struct {
/** /**
* @brief create a new ssh_pcap_file object * @brief create a new ssh_pcap_file object
*/ */
ssh_pcap_file ssh_pcap_file_new(void) { ssh_pcap_file ssh_pcap_file_new(void)
struct ssh_pcap_file_struct *pcap; {
struct ssh_pcap_file_struct *pcap = NULL;
pcap = (struct ssh_pcap_file_struct *) malloc(sizeof(struct ssh_pcap_file_struct)); pcap = malloc(sizeof(struct ssh_pcap_file_struct));
if (pcap == NULL) { if (pcap == NULL) {
return NULL; return NULL;
} }
@@ -137,29 +138,36 @@ ssh_pcap_file ssh_pcap_file_new(void) {
/** @internal /** @internal
* @brief writes a packet on file * @brief writes a packet on file
*/ */
static int ssh_pcap_file_write(ssh_pcap_file pcap, ssh_buffer packet){ static int ssh_pcap_file_write(ssh_pcap_file pcap, ssh_buffer packet)
{
int err; int err;
uint32_t len; uint32_t len;
if(pcap == NULL || pcap->output==NULL) if (pcap == NULL || pcap->output == NULL) {
return SSH_ERROR; return SSH_ERROR;
}
len = ssh_buffer_get_len(packet); len = ssh_buffer_get_len(packet);
err = fwrite(ssh_buffer_get(packet), len, 1, pcap->output); err = fwrite(ssh_buffer_get(packet), len, 1, pcap->output);
if(err<0) if (err < 0) {
return SSH_ERROR; return SSH_ERROR;
else } else {
return SSH_OK; return SSH_OK;
} }
}
/** @internal /** @internal
* @brief prepends a packet with the pcap header and writes packet * @brief prepends a packet with the pcap header and writes packet
* on file * on file
*/ */
int ssh_pcap_file_write_packet(ssh_pcap_file pcap, ssh_buffer packet, uint32_t original_len){ int ssh_pcap_file_write_packet(ssh_pcap_file pcap, ssh_buffer packet, uint32_t original_len)
{
ssh_buffer header = ssh_buffer_new(); ssh_buffer header = ssh_buffer_new();
struct timeval now; struct timeval now;
int err; int err;
if(header == NULL)
if (header == NULL) {
return SSH_ERROR; return SSH_ERROR;
}
gettimeofday(&now, NULL); gettimeofday(&now, NULL);
err = ssh_buffer_allocate_size(header, err = ssh_buffer_allocate_size(header,
sizeof(uint32_t) * 4 + sizeof(uint32_t) * 4 +
@@ -196,21 +204,26 @@ error:
/** /**
* @brief opens a new pcap file and creates header * @brief opens a new pcap file and creates header
*/ */
int ssh_pcap_file_open(ssh_pcap_file pcap, const char *filename){ int ssh_pcap_file_open(ssh_pcap_file pcap, const char *filename)
ssh_buffer header; {
ssh_buffer header = NULL;
int err; int err;
if(pcap == NULL)
if (pcap == NULL) {
return SSH_ERROR; return SSH_ERROR;
}
if (pcap->output) { if (pcap->output) {
fclose(pcap->output); fclose(pcap->output);
pcap->output = NULL; pcap->output = NULL;
} }
pcap->output = fopen(filename, "wb"); pcap->output = fopen(filename, "wb");
if(pcap->output==NULL) if (pcap->output == NULL) {
return SSH_ERROR; return SSH_ERROR;
}
header = ssh_buffer_new(); header = ssh_buffer_new();
if(header==NULL) if (header == NULL) {
return SSH_ERROR; return SSH_ERROR;
}
err = ssh_buffer_allocate_size(header, err = ssh_buffer_allocate_size(header,
sizeof(uint32_t) * 5 + sizeof(uint32_t) * 5 +
sizeof(uint16_t) * 2); sizeof(uint16_t) * 2);
@@ -255,19 +268,24 @@ error:
return err; return err;
} }
int ssh_pcap_file_close(ssh_pcap_file pcap){ int ssh_pcap_file_close(ssh_pcap_file pcap)
{
int err; int err;
if(pcap ==NULL || pcap->output==NULL)
if (pcap == NULL || pcap->output == NULL) {
return SSH_ERROR; return SSH_ERROR;
}
err = fclose(pcap->output); err = fclose(pcap->output);
pcap->output = NULL; pcap->output = NULL;
if(err != 0) if (err != 0) {
return SSH_ERROR; return SSH_ERROR;
else } else {
return SSH_OK; return SSH_OK;
} }
}
void ssh_pcap_file_free(ssh_pcap_file pcap){ void ssh_pcap_file_free(ssh_pcap_file pcap)
{
ssh_pcap_file_close(pcap); ssh_pcap_file_close(pcap);
SAFE_FREE(pcap); SAFE_FREE(pcap);
} }
@@ -276,7 +294,8 @@ void ssh_pcap_file_free(ssh_pcap_file pcap){
/** @internal /** @internal
* @brief allocates a new ssh_pcap_context object * @brief allocates a new ssh_pcap_context object
*/ */
ssh_pcap_context ssh_pcap_context_new(ssh_session session){ ssh_pcap_context ssh_pcap_context_new(ssh_session session)
{
ssh_pcap_context ctx = (struct ssh_pcap_context_struct *)malloc(sizeof(struct ssh_pcap_context_struct)); ssh_pcap_context ctx = (struct ssh_pcap_context_struct *)malloc(sizeof(struct ssh_pcap_context_struct));
if (ctx == NULL) { if (ctx == NULL) {
ssh_set_error_oom(session); ssh_set_error_oom(session);
@@ -287,11 +306,13 @@ ssh_pcap_context ssh_pcap_context_new(ssh_session session){
return ctx; return ctx;
} }
void ssh_pcap_context_free(ssh_pcap_context ctx){ void ssh_pcap_context_free(ssh_pcap_context ctx)
{
SAFE_FREE(ctx); SAFE_FREE(ctx);
} }
void ssh_pcap_context_set_file(ssh_pcap_context ctx, ssh_pcap_file pcap){ void ssh_pcap_context_set_file(ssh_pcap_context ctx, ssh_pcap_file pcap)
{
ctx->file = pcap; ctx->file = pcap;
} }
@@ -506,15 +527,17 @@ error:
* sessions. * sessions.
* @returns SSH_ERROR in case of error, SSH_OK otherwise. * @returns SSH_ERROR in case of error, SSH_OK otherwise.
*/ */
int ssh_set_pcap_file(ssh_session session, ssh_pcap_file pcap){ int ssh_set_pcap_file(ssh_session session, ssh_pcap_file pcap)
{
ssh_pcap_context ctx = ssh_pcap_context_new(session); ssh_pcap_context ctx = ssh_pcap_context_new(session);
if (ctx == NULL) { if (ctx == NULL) {
ssh_set_error_oom(session); ssh_set_error_oom(session);
return SSH_ERROR; return SSH_ERROR;
} }
ctx->file = pcap; ctx->file = pcap;
if(session->pcap_ctx) if (session->pcap_ctx) {
ssh_pcap_context_free(session->pcap_ctx); ssh_pcap_context_free(session->pcap_ctx);
}
session->pcap_ctx = ctx; session->pcap_ctx = ctx;
return SSH_OK; return SSH_OK;
} }
@@ -527,26 +550,34 @@ int ssh_set_pcap_file(ssh_session session, ssh_pcap_file pcap){
#include "libssh/libssh.h" #include "libssh/libssh.h"
#include "libssh/priv.h" #include "libssh/priv.h"
int ssh_pcap_file_close(ssh_pcap_file pcap){ int ssh_pcap_file_close(ssh_pcap_file pcap)
{
(void)pcap; (void)pcap;
return SSH_ERROR; return SSH_ERROR;
} }
void ssh_pcap_file_free(ssh_pcap_file pcap){ void ssh_pcap_file_free(ssh_pcap_file pcap)
{
(void)pcap; (void)pcap;
} }
ssh_pcap_file ssh_pcap_file_new(void){ ssh_pcap_file ssh_pcap_file_new(void)
{
return NULL; return NULL;
} }
int ssh_pcap_file_open(ssh_pcap_file pcap, const char *filename){ int ssh_pcap_file_open(ssh_pcap_file pcap, const char *filename)
{
(void)pcap; (void)pcap;
(void)filename; (void)filename;
return SSH_ERROR; return SSH_ERROR;
} }
int ssh_set_pcap_file(ssh_session session, ssh_pcap_file pcapfile){ int ssh_set_pcap_file(ssh_session session, ssh_pcap_file pcapfile)
{
(void)pcapfile; (void)pcapfile;
ssh_set_error(session, SSH_REQUEST_DENIED, "Pcap support not compiled in"); ssh_set_error(session, SSH_REQUEST_DENIED, "Pcap support not compiled in");
return SSH_ERROR; return SSH_ERROR;
} }