mirror of
https://github.com/inspec/inspec
synced 2024-11-23 05:03:07 +00:00
2857d07151
The opening and closing mechanic varied between all the various resources. This changes them all to use a HEREDOC with a tilde to remove leading whitespace. This removes the need for the special method to trim the `#print_example` method from shell. Signed-off-by: Franklin Webber <franklin.webber@gmail.com>
86 lines
2.5 KiB
Ruby
86 lines
2.5 KiB
Ruby
# encoding: utf-8
|
|
# copyright: 2015, Vulcano Security GmbH
|
|
|
|
require 'shellwords'
|
|
|
|
module Inspec::Resources
|
|
class MysqlSession < Inspec.resource(1)
|
|
name 'mysql_session'
|
|
supports platform: 'unix'
|
|
supports platform: 'windows'
|
|
desc 'Use the mysql_session InSpec audit resource to test SQL commands run against a MySQL database.'
|
|
example <<~EXAMPLE
|
|
sql = mysql_session('my_user','password','host')
|
|
describe sql.query('show databases like \'test\';') do
|
|
its('stdout') { should_not match(/test/) }
|
|
end
|
|
EXAMPLE
|
|
|
|
def initialize(user = nil, pass = nil, host = 'localhost', port = nil, socket = nil)
|
|
@user = user
|
|
@pass = pass
|
|
@host = host
|
|
@port = port
|
|
@socket = socket
|
|
init_fallback if user.nil? or pass.nil?
|
|
skip_resource("Can't run MySQL SQL checks without authentication") if @user.nil? or @pass.nil?
|
|
end
|
|
|
|
def query(q, db = '')
|
|
mysql_cmd = create_mysql_cmd(q, db)
|
|
cmd = inspec.command(mysql_cmd)
|
|
out = cmd.stdout + "\n" + cmd.stderr
|
|
if out =~ /Can't connect to .* MySQL server/ || out.downcase =~ /^error /
|
|
# skip this test if the server can't run the query
|
|
warn("Can't connect to MySQL instance for SQL checks.")
|
|
end
|
|
|
|
# return the raw command output
|
|
cmd
|
|
end
|
|
|
|
def to_s
|
|
'MySQL Session'
|
|
end
|
|
|
|
private
|
|
|
|
def escape_string(query)
|
|
Shellwords.escape(query)
|
|
end
|
|
|
|
def create_mysql_cmd(q, db = '')
|
|
# TODO: simple escape, must be handled by a library
|
|
# that does this securely
|
|
escaped_query = q.gsub(/\\/, '\\\\').gsub(/"/, '\\"').gsub(/\$/, '\\$')
|
|
|
|
# construct the query
|
|
command = 'mysql'
|
|
command += " -u#{escape_string(@user)}" unless @user.nil?
|
|
command += " -p#{escape_string(@pass)}" unless @pass.nil?
|
|
|
|
if !@socket.nil?
|
|
command += " -S #{@socket}"
|
|
else
|
|
command += " -h #{@host}"
|
|
end
|
|
command += " --port #{@port}" unless @port.nil?
|
|
command += " #{db}" unless db.empty?
|
|
command += %{ -s -e "#{escaped_query}"}
|
|
command
|
|
end
|
|
|
|
def init_fallback
|
|
# support debian mysql administration login
|
|
return if inspec.platform.in_family?('windows')
|
|
debian = inspec.command('test -f /etc/mysql/debian.cnf && cat /etc/mysql/debian.cnf').stdout
|
|
return if debian.empty?
|
|
|
|
user = debian.match(/^\s*user\s*=\s*([^ ]*)\s*$/)
|
|
pass = debian.match(/^\s*password\s*=\s*([^ ]*)\s*$/)
|
|
return if user.nil? or pass.nil?
|
|
@user = user[1]
|
|
@pass = pass[1]
|
|
end
|
|
end
|
|
end
|