suricata
output-json-krb5.c
Go to the documentation of this file.
1 /* Copyright (C) 2018 Open Information Security Foundation
2  *
3  * You can copy, redistribute or modify this Program under the terms of
4  * the GNU General Public License version 2 as published by the Free
5  * Software Foundation.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * version 2 along with this program; if not, write to the Free Software
14  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
15  * 02110-1301, USA.
16  */
17 
18 /**
19  * \file
20  *
21  * \author Pierre Chifflier <chifflier@wzdftpd.net>
22  *
23  * Implement JSON/eve logging app-layer KRB5.
24  */
25 
26 #include "suricata-common.h"
27 #include "debug.h"
28 #include "detect.h"
29 #include "pkt-var.h"
30 #include "conf.h"
31 
32 #include "threads.h"
33 #include "threadvars.h"
34 #include "tm-threads.h"
35 
36 #include "util-unittest.h"
37 #include "util-buffer.h"
38 #include "util-debug.h"
39 #include "util-byte.h"
40 
41 #include "output.h"
42 #include "output-json.h"
43 
44 #include "app-layer.h"
45 #include "app-layer-parser.h"
46 
47 #include "app-layer-krb5.h"
48 #include "output-json-krb5.h"
49 
50 #include "rust.h"
51 #include "rust-krb-log-gen.h"
52 
53 typedef struct LogKRB5FileCtx_ {
57 
58 typedef struct LogKRB5LogThread_ {
62 
63 static int JsonKRB5Logger(ThreadVars *tv, void *thread_data,
64  const Packet *p, Flow *f, void *state, void *tx, uint64_t tx_id)
65 {
66  KRB5Transaction *krb5tx = tx;
67  LogKRB5LogThread *thread = thread_data;
68  json_t *js, *krb5js;
69 
70  js = CreateJSONHeader(p, LOG_DIR_PACKET, "krb5");
71  if (unlikely(js == NULL)) {
72  return TM_ECODE_FAILED;
73  }
74 
75  JsonAddCommonOptions(&thread->krb5log_ctx->cfg, p, f, js);
76 
77  krb5js = rs_krb5_log_json_response(state, krb5tx);
78  if (unlikely(krb5js == NULL)) {
79  goto error;
80  }
81  json_object_set_new(js, "krb5", krb5js);
82 
83  MemBufferReset(thread->buffer);
84  OutputJSONBuffer(js, thread->krb5log_ctx->file_ctx, &thread->buffer);
85 
86  json_decref(js);
87  return TM_ECODE_OK;
88 
89 error:
90  json_decref(js);
91  return TM_ECODE_FAILED;
92 }
93 
94 static void OutputKRB5LogDeInitCtxSub(OutputCtx *output_ctx)
95 {
96  LogKRB5FileCtx *krb5log_ctx = (LogKRB5FileCtx *)output_ctx->data;
97  SCFree(krb5log_ctx);
98  SCFree(output_ctx);
99 }
100 
101 static OutputInitResult OutputKRB5LogInitSub(ConfNode *conf,
102  OutputCtx *parent_ctx)
103 {
104  OutputInitResult result = { NULL, false };
105  OutputJsonCtx *ajt = parent_ctx->data;
106 
107  LogKRB5FileCtx *krb5log_ctx = SCCalloc(1, sizeof(*krb5log_ctx));
108  if (unlikely(krb5log_ctx == NULL)) {
109  return result;
110  }
111  krb5log_ctx->file_ctx = ajt->file_ctx;
112  krb5log_ctx->cfg = ajt->cfg;
113 
114  OutputCtx *output_ctx = SCCalloc(1, sizeof(*output_ctx));
115  if (unlikely(output_ctx == NULL)) {
116  SCFree(krb5log_ctx);
117  return result;
118  }
119  output_ctx->data = krb5log_ctx;
120  output_ctx->DeInit = OutputKRB5LogDeInitCtxSub;
121 
122  SCLogDebug("KRB5 log sub-module initialized.");
123 
126 
127  result.ctx = output_ctx;
128  result.ok = true;
129  return result;
130 }
131 
132 static TmEcode JsonKRB5LogThreadInit(ThreadVars *t, const void *initdata, void **data)
133 {
134  LogKRB5LogThread *thread = SCCalloc(1, sizeof(*thread));
135  if (unlikely(thread == NULL)) {
136  return TM_ECODE_FAILED;
137  }
138 
139  if (initdata == NULL) {
140  SCLogDebug("Error getting context for EveLogKRB5. \"initdata\" is NULL.");
141  SCFree(thread);
142  return TM_ECODE_FAILED;
143  }
144 
146  if (unlikely(thread->buffer == NULL)) {
147  SCFree(thread);
148  return TM_ECODE_FAILED;
149  }
150 
151  thread->krb5log_ctx = ((OutputCtx *)initdata)->data;
152  *data = (void *)thread;
153 
154  return TM_ECODE_OK;
155 }
156 
157 static TmEcode JsonKRB5LogThreadDeinit(ThreadVars *t, void *data)
158 {
159  LogKRB5LogThread *thread = (LogKRB5LogThread *)data;
160  if (thread == NULL) {
161  return TM_ECODE_OK;
162  }
163  if (thread->buffer != NULL) {
164  MemBufferFree(thread->buffer);
165  }
166  SCFree(thread);
167  return TM_ECODE_OK;
168 }
169 
171 {
172  /* Register as an eve sub-module. */
173  OutputRegisterTxSubModule(LOGGER_JSON_KRB5, "eve-log", "JsonKRB5Log",
174  "eve-log.krb5", OutputKRB5LogInitSub, ALPROTO_KRB5,
175  JsonKRB5Logger, JsonKRB5LogThreadInit,
176  JsonKRB5LogThreadDeinit, NULL);
177 
178  SCLogDebug("KRB5 JSON logger registered.");
179 }
void JsonKRB5LogRegister(void)
MemBuffer * MemBufferCreateNew(uint32_t size)
Definition: util-buffer.c:32
#define JSON_OUTPUT_BUFFER_SIZE
Definition: output-json.h:44
json_t * CreateJSONHeader(const Packet *p, enum OutputJsonLogDirection dir, const char *event_type)
Definition: output-json.c:710
OutputJsonCommonSettings cfg
#define SCLogDebug(...)
Definition: util-debug.h:335
int OutputJSONBuffer(json_t *js, LogFileCtx *file_ctx, MemBuffer **buffer)
Definition: output-json.c:809
#define unlikely(expr)
Definition: util-optimize.h:35
LogKRB5FileCtx * krb5log_ctx
#define MemBufferReset(mem_buffer)
Reset the mem buffer.
Definition: util-buffer.h:42
void(* DeInit)(struct OutputCtx_ *)
Definition: tm-modules.h:84
void JsonAddCommonOptions(const OutputJsonCommonSettings *cfg, const Packet *p, const Flow *f, json_t *js)
Definition: output-json.c:390
struct LogKRB5LogThread_ LogKRB5LogThread
#define SCCalloc(nm, a)
Definition: util-mem.h:253
OutputJsonCommonSettings cfg
Definition: output-json.h:81
void AppLayerParserRegisterLogger(uint8_t ipproto, AppProto alproto)
LogFileCtx * file_ctx
Definition: output-json.h:79
LogFileCtx * file_ctx
struct LogKRB5FileCtx_ LogKRB5FileCtx
void OutputRegisterTxSubModule(LoggerId id, const char *parent_name, const char *name, const char *conf_name, OutputInitSubFunc InitFunc, AppProto alproto, TxLogger TxLogFunc, ThreadInitFunc ThreadInit, ThreadDeinitFunc ThreadDeinit, ThreadExitPrintStatsFunc ThreadExitPrintStats)
Definition: output.c:402
Definition: conf.h:32
OutputCtx * ctx
Definition: output.h:42
#define SCFree(a)
Definition: util-mem.h:322
uint16_t tx_id
void * data
Definition: tm-modules.h:81
Per thread variable structure.
Definition: threadvars.h:57
struct KRB5Transaction_ KRB5Transaction
Flow data structure.
Definition: flow.h:325
void MemBufferFree(MemBuffer *buffer)
Definition: util-buffer.c:82