71 lines
2.0 KiB
OCaml
71 lines
2.0 KiB
OCaml
open Core
|
|
|
|
let default_socket_path = "/run/systemd/journal/socket"
|
|
|
|
type t = {
|
|
mutex : Mutex.t;
|
|
sock_fd : Unix.file_descr;
|
|
dest : Unix.sockaddr;
|
|
buf : Buffer.t;
|
|
}
|
|
|
|
let should_upgrade () =
|
|
let stderr = Unix.fstat Unix.stderr in
|
|
let dev_ino = Printf.sprintf "%d:%d" stderr.st_dev stderr.st_ino in
|
|
Sys.getenv_opt "JOURNAL_STREAM" = Some dev_ino
|
|
|
|
let make ?(path = default_socket_path) () = {
|
|
mutex = Mutex.create ();
|
|
sock_fd = Unix.socket PF_UNIX SOCK_DGRAM 0 ~cloexec:true;
|
|
dest = ADDR_UNIX path;
|
|
buf = Buffer.create 256;
|
|
}
|
|
|
|
let add_field dgram key value =
|
|
if String.contains value '\n' then
|
|
begin
|
|
Buffer.add_string dgram key;
|
|
Buffer.add_char dgram '\n';
|
|
Buffer.add_int64_le dgram (Int64.of_int (String.length value));
|
|
Buffer.add_string dgram value;
|
|
Buffer.add_char dgram '\n';
|
|
end
|
|
else
|
|
Printf.bprintf dgram "%s=%s\n" key value
|
|
|
|
let syslog_priority = function
|
|
| TRACE
|
|
| DEBUG -> "7" (* LOG_DEBUG *)
|
|
| INFO -> "6" (* LOG_INFO *)
|
|
| WARN -> "4" (* LOG_WARNING *)
|
|
| ERROR -> "3" (* LOG_ERR *)
|
|
|
|
let writer t ~ts ~ns ~filename ~lineno ~func ~errno ~exn ~lvl msg =
|
|
Mutex.lock t.mutex;
|
|
let dgram =
|
|
Buffer.clear t.buf;
|
|
ignore ts;
|
|
let maybe_exn = match exn with
|
|
| Some (exn, bt) ->
|
|
Printf.sprintf "\nException: %s\n%s" (Printexc.to_string exn)
|
|
(Printexc.raw_backtrace_to_string bt)
|
|
| None -> ""
|
|
in
|
|
add_field t.buf "MESSAGE" (Printf.sprintf "%s: %s%s" ns msg maybe_exn);
|
|
add_field t.buf "PRIORITY" (syslog_priority lvl);
|
|
begin if not (filename = "") then
|
|
add_field t.buf "CODE_FILE" filename;
|
|
add_field t.buf "CODE_LINE" (Printf.sprintf "%d" lineno)
|
|
end;
|
|
begin if not (func = "") then
|
|
add_field t.buf "CODE_FUNC" func
|
|
end;
|
|
let errno = Util.get_errno errno exn in
|
|
begin if errno != 0 then
|
|
add_field t.buf "ERRNO" (Printf.sprintf "%d" errno)
|
|
end;
|
|
Buffer.to_bytes t.buf
|
|
in
|
|
Mutex.unlock t.mutex;
|
|
Unix.sendto t.sock_fd dgram 0 (Bytes.length dgram) [] t.dest |> ignore
|