krb: convert to jsonbuilder

Closes redmine ticket 3754.
pull/5188/head
Shivani Bhardwaj 5 years ago committed by Victor Julien
parent 72dab0a8b7
commit 67e7be633c

@ -17,20 +17,18 @@
// written by Pierre Chifflier <chifflier@wzdftpd.net> // written by Pierre Chifflier <chifflier@wzdftpd.net>
use crate::json::*; use crate::jsonbuilder::{JsonBuilder, JsonError};
use crate::krb::krb5::{KRB5State,KRB5Transaction,test_weak_encryption}; use crate::krb::krb5::{KRB5State,KRB5Transaction,test_weak_encryption};
#[no_mangle] fn krb5_log_response(jsb: &mut JsonBuilder, tx: &mut KRB5Transaction) -> Result<(), JsonError>
pub extern "C" fn rs_krb5_log_json_response(_state: &mut KRB5State, tx: &mut KRB5Transaction) -> *mut JsonT
{ {
let js = Json::object();
match tx.error_code { match tx.error_code {
Some(c) => { Some(c) => {
js.set_string("msg_type", "KRB_ERROR"); jsb.set_string("msg_type", "KRB_ERROR")?;
js.set_string("failed_request", &format!("{:?}", tx.msg_type)); jsb.set_string("failed_request", &format!("{:?}", tx.msg_type))?;
js.set_string("error_code", &format!("{:?}", c)); jsb.set_string("error_code", &format!("{:?}", c))?;
}, },
None => { js.set_string("msg_type", &format!("{:?}", tx.msg_type)); }, None => { jsb.set_string("msg_type", &format!("{:?}", tx.msg_type))?; },
} }
let cname = match tx.cname { let cname = match tx.cname {
Some(ref x) => format!("{}", x), Some(ref x) => format!("{}", x),
@ -48,11 +46,17 @@ pub extern "C" fn rs_krb5_log_json_response(_state: &mut KRB5State, tx: &mut KRB
Some(ref x) => format!("{:?}", x), Some(ref x) => format!("{:?}", x),
None => "<none>".to_owned(), None => "<none>".to_owned(),
}; };
js.set_string("cname", &cname); jsb.set_string("cname", &cname)?;
js.set_string("realm", &realm); jsb.set_string("realm", &realm)?;
js.set_string("sname", &sname); jsb.set_string("sname", &sname)?;
js.set_string("encryption", &encryption); jsb.set_string("encryption", &encryption)?;
js.set_boolean("weak_encryption", tx.etype.map_or(false,test_weak_encryption)); jsb.set_bool("weak_encryption", tx.etype.map_or(false,test_weak_encryption))?;
return js.unwrap();
return Ok(());
} }
#[no_mangle]
pub extern "C" fn rs_krb5_log_json_response(jsb: &mut JsonBuilder, _state: &mut KRB5State, tx: &mut KRB5Transaction) -> bool
{
krb5_log_response(jsb, tx).is_ok()
}

@ -64,29 +64,28 @@ static int JsonKRB5Logger(ThreadVars *tv, void *thread_data,
{ {
KRB5Transaction *krb5tx = tx; KRB5Transaction *krb5tx = tx;
LogKRB5LogThread *thread = thread_data; LogKRB5LogThread *thread = thread_data;
json_t *js, *krb5js;
js = CreateJSONHeader(p, LOG_DIR_PACKET, "krb5", NULL); JsonBuilder *jb = CreateEveHeader(p, LOG_DIR_PACKET, "krb5", NULL);
if (unlikely(js == NULL)) { if (unlikely(jb == NULL)) {
return TM_ECODE_FAILED; return TM_ECODE_FAILED;
} }
JsonAddCommonOptions(&thread->krb5log_ctx->cfg, p, f, js); EveAddCommonOptions(&thread->krb5log_ctx->cfg, p, f, jb);
krb5js = rs_krb5_log_json_response(state, krb5tx); jb_open_object(jb, "krb5");
if (unlikely(krb5js == NULL)) { if (!rs_krb5_log_json_response(jb, state, krb5tx)) {
goto error; goto error;
} }
json_object_set_new(js, "krb5", krb5js); jb_close(jb);
MemBufferReset(thread->buffer); MemBufferReset(thread->buffer);
OutputJSONBuffer(js, thread->krb5log_ctx->file_ctx, &thread->buffer); OutputJsonBuilderBuffer(jb, thread->krb5log_ctx->file_ctx, &thread->buffer);
json_decref(js); jb_free(jb);
return TM_ECODE_OK; return TM_ECODE_OK;
error: error:
json_decref(js); jb_free(jb);
return TM_ECODE_FAILED; return TM_ECODE_FAILED;
} }

Loading…
Cancel
Save