mirror of https://github.com/OISF/suricata
Minor cosmetic changes to the cuda code.
Moved a couple of functions to more cuda relevant files; Re-structured some data types.pull/400/head
parent
c9f076def3
commit
602c91ed41
@ -0,0 +1,74 @@
|
||||
/* Copyright (C) 2007-2010 Open Information Security Foundation
|
||||
*
|
||||
* You can copy, redistribute or modify this Program under the terms of
|
||||
* the GNU General Public License version 2 as published by the Free
|
||||
* Software Foundation.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* version 2 along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
||||
* 02110-1301, USA.
|
||||
*/
|
||||
|
||||
/**
|
||||
* \file
|
||||
*
|
||||
* \author Anoop Saldanha <anoopsaldanha@gmail.com>
|
||||
*/
|
||||
|
||||
#ifdef __SC_CUDA_SUPPORT__
|
||||
|
||||
#include "suricata.h"
|
||||
#include "util-mpm.h"
|
||||
#include "util-cuda-handlers.h"
|
||||
#include "util-cuda-vars.h"
|
||||
#include "detect-engine-mpm.h"
|
||||
#include "util-debug.h"
|
||||
#include "util-mpm-ac.h"
|
||||
|
||||
static DetectEngineCtx *cuda_de_ctx = NULL;
|
||||
|
||||
void CudaVarsSetDeCtx(DetectEngineCtx *de_ctx)
|
||||
{
|
||||
if (cuda_de_ctx != NULL) {
|
||||
SCLogError(SC_ERR_FATAL, "CudaVarsSetDeCtx() called more than once. "
|
||||
"This function should be called only once during the "
|
||||
"lifetime of the engine.");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
cuda_de_ctx = de_ctx;
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
int CudaThreadVarsInit(CudaThreadVars *ctv)
|
||||
{
|
||||
if (PatternMatchDefaultMatcher() != MPM_AC_CUDA)
|
||||
return 0;
|
||||
|
||||
MpmCudaConf *conf = CudaHandlerGetCudaProfile("mpm");
|
||||
if (conf == NULL) {
|
||||
SCLogError(SC_ERR_AC_CUDA_ERROR, "Error obtaining cuda mpm profile.");
|
||||
return -1;
|
||||
}
|
||||
|
||||
ctv->mpm_is_cuda = 1;
|
||||
ctv->cuda_ac_cb = CudaHandlerModuleGetData(MPM_AC_CUDA_MODULE_NAME, MPM_AC_CUDA_MODULE_CUDA_BUFFER_NAME);
|
||||
ctv->data_buffer_size_max_limit = conf->data_buffer_size_max_limit;
|
||||
ctv->data_buffer_size_min_limit = conf->data_buffer_size_min_limit;
|
||||
ctv->mpm_proto_tcp_ctx_ts = MpmFactoryGetMpmCtxForProfile(cuda_de_ctx, cuda_de_ctx->sgh_mpm_context_proto_tcp_packet, 0);
|
||||
ctv->mpm_proto_tcp_ctx_tc = MpmFactoryGetMpmCtxForProfile(cuda_de_ctx, cuda_de_ctx->sgh_mpm_context_proto_tcp_packet, 1);
|
||||
ctv->mpm_proto_udp_ctx_ts = MpmFactoryGetMpmCtxForProfile(cuda_de_ctx, cuda_de_ctx->sgh_mpm_context_proto_udp_packet, 0);
|
||||
ctv->mpm_proto_udp_ctx_tc = MpmFactoryGetMpmCtxForProfile(cuda_de_ctx, cuda_de_ctx->sgh_mpm_context_proto_udp_packet, 1);
|
||||
ctv->mpm_proto_other_ctx = MpmFactoryGetMpmCtxForProfile(cuda_de_ctx, cuda_de_ctx->sgh_mpm_context_proto_other_packet, 0);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
#endif
|
@ -0,0 +1,65 @@
|
||||
/* Copyright (C) 2007-2010 Open Information Security Foundation
|
||||
*
|
||||
* You can copy, redistribute or modify this Program under the terms of
|
||||
* the GNU General Public License version 2 as published by the Free
|
||||
* Software Foundation.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* version 2 along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
||||
* 02110-1301, USA.
|
||||
*/
|
||||
|
||||
/**
|
||||
* \file
|
||||
*
|
||||
* \author Anoop Saldanha <anoopsaldanha@gmail.com>
|
||||
*/
|
||||
|
||||
#ifdef __SC_CUDA_SUPPORT__
|
||||
|
||||
#ifndef __UTIL_CUDA_VARS__H__
|
||||
#define __UTIL_CUDA_VARS__H__
|
||||
|
||||
#include "util-cuda-buffer.h"
|
||||
#include "util-mpm.h"
|
||||
#include "threads.h"
|
||||
|
||||
typedef struct CudaThreadVars_ {
|
||||
/* cb - CudaBuffer */
|
||||
CudaBufferData *cuda_ac_cb;
|
||||
|
||||
MpmCtx *mpm_proto_other_ctx;
|
||||
|
||||
MpmCtx *mpm_proto_tcp_ctx_ts;
|
||||
MpmCtx *mpm_proto_udp_ctx_ts;
|
||||
|
||||
MpmCtx *mpm_proto_tcp_ctx_tc;
|
||||
MpmCtx *mpm_proto_udp_ctx_tc;
|
||||
|
||||
uint16_t data_buffer_size_max_limit;
|
||||
uint16_t data_buffer_size_min_limit;
|
||||
|
||||
uint8_t mpm_is_cuda;
|
||||
} CudaThreadVars;
|
||||
|
||||
typedef struct CudaPacketVars_ {
|
||||
uint8_t cuda_mpm_enabled;
|
||||
uint8_t cuda_done;
|
||||
uint16_t cuda_gpu_matches;
|
||||
SCMutex cuda_mutex;
|
||||
SCCondT cuda_cond;
|
||||
uint32_t cuda_results[(UTIL_MPM_CUDA_DATA_BUFFER_SIZE_MAX_LIMIT_DEFAULT * 2) + 1];
|
||||
} CudaPacketVars;
|
||||
|
||||
void CudaVarsSetDeCtx(struct DetectEngineCtx_ *de_ctx);
|
||||
int CudaThreadVarsInit(CudaThreadVars *ctv);
|
||||
|
||||
#endif /* __UTIL_CUDA_VARS__H__ */
|
||||
|
||||
#endif /* __SC_CUDA_SUPPORT__ */
|
Loading…
Reference in New Issue