120 lines
4.1 KiB
Ruby
120 lines
4.1 KiB
Ruby
# Ossec Agent provisioning recipe
|
|
# install the ossec-hids-agent package, push the global
|
|
# and role specific configuration for the node
|
|
# get a key from the ossec-server if there's one
|
|
|
|
if node['platform_family'] == "rhel"
|
|
include_recipe "yum-atomic"
|
|
elsif node['platform_family'] == "debian"
|
|
include_recipe "apt-atomic"
|
|
end
|
|
|
|
class Chef::Recipe
|
|
include OssecCore
|
|
end
|
|
|
|
|
|
# Run this recipe if the node is an agent. Since the ossec::agent recipe is
|
|
# added to the base role, ossec-servers will run it as well, making this check
|
|
# necessary
|
|
if not node["ossec"]["agent"]["enable"]
|
|
# return will exit this recipe
|
|
# and continue the chef provisioning
|
|
Chef::Log.info("OSSEC: agent is not enabled on this node")
|
|
return true
|
|
end
|
|
|
|
# Search for the ossec server, and do nothing if there's none
|
|
if Chef::Config[:solo]
|
|
Chef::Log.warn('This recipe uses search. Chef Solo does not support search')
|
|
else
|
|
ossec_server = search(:node,
|
|
"role:ossec-server " \
|
|
"AND chef_environment:#{node.chef_environment}"
|
|
).first
|
|
if ossec_server.nil?
|
|
Chef::Log.info("OSSEC: No ossec server available. Agent will not be provisionned")
|
|
return true
|
|
end
|
|
|
|
# install the agent package
|
|
package "ossec-hids-agent"
|
|
|
|
# define the agent parameters
|
|
agent_hash = ossec_agent_create_parameters(node, ossec_server)
|
|
|
|
# check for the agent configuration on the server. if the server has none, do
|
|
# not continue the provisioning. If the server has a configuration for this
|
|
# agent, store the parameters on the node and continue
|
|
if ossec_verify_agent(agent_hash, ossec_server)
|
|
node.normal["ossec"]["agents"][agent_hash[:id]] = ossec_server["ossec"]["agents"][agent_hash[:id]].to_hash
|
|
else
|
|
Chef::Log.info("OSSEC: this agent is unknown on the ossec server")
|
|
return true
|
|
end
|
|
|
|
# Make sure that the server prepared a key for us
|
|
unless ossec_agent_has_valid_key?(agent_hash, ossec_server)
|
|
Chef::Log.info("OSSEC: Server doesn't have a valid key for agent.")
|
|
return true
|
|
end
|
|
|
|
service "ossec-agent" do
|
|
#provider Chef::Provider::Service::Init
|
|
service_name node["ossec"]["client"]["service_name"]
|
|
supports :start => true, :stop => true, :restart => true, :status => true
|
|
action [ :start ]
|
|
only_if "test -e /var/ossec/etc/ossec.conf && test -e /var/ossec/etc/client.keys"
|
|
end
|
|
|
|
# Get the IP of the ossec server
|
|
ossec_server_ip = ossec_server[:network][:lanip] || ossec_server.ipaddress
|
|
|
|
# Expand the local flags from node attributes
|
|
ossec_set_filtered_flags!("command", "active-response", "syslog_files")
|
|
ossec_set_syscheck_flags!("ignore")
|
|
|
|
template "/var/ossec/etc/ossec.conf" do
|
|
source "ossec-agent.conf.erb"
|
|
owner "ossec"
|
|
group "ossec"
|
|
variables("ossec_server_ip" => ossec_server_ip )
|
|
manage_symlink_source true
|
|
notifies :restart, "service[ossec-agent]"
|
|
end
|
|
|
|
# If client.keys is modified, ask for a queue rid on the server
|
|
template "/var/ossec/etc/client.keys" do
|
|
mode 0440
|
|
owner "root"
|
|
group "ossec"
|
|
notifies :create, "ruby_block[set-rid-flag]"
|
|
notifies :restart, "service[ossec-agent]"
|
|
end
|
|
|
|
# "set-rid-flag" is not run by default, but called when the agent's key
|
|
# is modified (or created)
|
|
ruby_block "set-rid-flag" do
|
|
block do
|
|
# if the server side rid flag is not set to "done",
|
|
# request a queue rid by setting the agent side flag to "todo"
|
|
if ossec_server["ossec"]["agents"][agent_hash[:id]]["rid"].eql?("none")
|
|
node.normal["ossec"]["agents"][agent_hash[:id]]["rid"] = "todo"
|
|
Chef::Log.info "Setting Queue Rid Flag on"
|
|
end
|
|
end
|
|
action :nothing
|
|
end
|
|
|
|
# unset rid flag if necessary, check that at every run
|
|
ruby_block "unset rid flag" do
|
|
block do
|
|
node.normal["ossec"]["agents"][agent_hash[:id]]["rid"] = "none"
|
|
Chef::Log.info "Setting Queue Rid Flag off"
|
|
end
|
|
notifies :restart, "service[ossec-agent]"
|
|
only_if { node["ossec"]["agents"][agent_hash[:id]]["rid"].eql?("todo") \
|
|
and ossec_server["ossec"]["agents"][agent_hash[:id]]["rid"].eql?("done") }
|
|
end
|
|
end
|