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.
52 lines
1.8 KiB
Python
52 lines
1.8 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.
|
|
|
|
|
|
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('socket', metavar='socket', nargs='?', help='socket file to connnect to', default=None)
|
|
args = parser.parse_args()
|
|
|
|
if args.socket != None:
|
|
SOCKET_PATH = "@e_localstatedir@/" + args.socket[0]
|
|
else:
|
|
SOCKET_PATH = "@e_localstatedir@/suricata-command.socket"
|
|
|
|
sc = SuricataSC(SOCKET_PATH, verbose=args.verbose)
|
|
try:
|
|
sc.connect()
|
|
except SuricataNetException, err:
|
|
print "Unable to connect to socket %s: %s" % (SOCKET_PATH, err)
|
|
sys.exit(1)
|
|
except SuricataReturnException, err:
|
|
print "Unable to negotiate version with server: %s" % (err)
|
|
sys.exit(1)
|
|
try:
|
|
sc.interactive()
|
|
except SuricataNetException, err:
|
|
print "Communication error: %s" % (err)
|
|
sys.exit(1)
|
|
except SuricataReturnException, err:
|
|
print "Invalid return from server: %s" % (err)
|
|
sys.exit(1)
|
|
|
|
print "[+] Quit command client"
|
|
|
|
sc.close()
|