mirror of https://github.com/OISF/suricata
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
#!/usr/bin/python
|
|
# Copyright(C) 2013 Open Information Security Foundation
|
|
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, version 2 of the License.
|
|
#
|
|
# 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
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
|
|
|
|
from __future__ import print_function
|
|
import sys
|
|
import argparse
|
|
from suricatasc import *
|
|
|
|
parser = argparse.ArgumentParser(prog='suricatasc', description='Client for Suricata unix socket')
|
|
parser.add_argument('-v', '--verbose', action='store_const', const=True, help='verbose output (including JSON dump)')
|
|
parser.add_argument('-c', '--command', default=None, help='execute on single command and return JSON')
|
|
parser.add_argument('socket', metavar='socket', nargs='?', help='socket file to connnect to', default=None)
|
|
args = parser.parse_args()
|
|
|
|
if args.socket != None:
|
|
SOCKET_PATH = args.socket
|
|
else:
|
|
SOCKET_PATH = "@e_localstatedir@/suricata-command.socket"
|
|
|
|
sc = SuricataSC(SOCKET_PATH, verbose=args.verbose)
|
|
try:
|
|
sc.connect()
|
|
except SuricataNetException as err:
|
|
print("Unable to connect to socket %s: %s" % (SOCKET_PATH, err), file=sys.stderr)
|
|
sys.exit(1)
|
|
except SuricataReturnException as err:
|
|
print("Unable to negotiate version with server: %s" % (err), file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
if args.command:
|
|
(command, arguments) = sc.parse_command(args.command)
|
|
res = sc.send_command(command, arguments)
|
|
print(json.dumps(res))
|
|
sc.close()
|
|
if res['return'] == 'OK':
|
|
sys.exit(0)
|
|
else:
|
|
sys.exit(1)
|
|
|
|
try:
|
|
sc.interactive()
|
|
except SuricataNetException as err:
|
|
print("Communication error: %s" % (err))
|
|
sys.exit(1)
|
|
except SuricataReturnException as err:
|
|
print("Invalid return from server: %s" % (err))
|
|
sys.exit(1)
|
|
|
|
print("[+] Quit command client")
|
|
|
|
sc.close()
|