eef4f5f4f0
git-svn-id: https://beef.googlecode.com/svn/trunk@1279 b87d56ec-f9c0-11de-8c8a-61c5e9addfc9
96 lines
2.6 KiB
Ruby
96 lines
2.6 KiB
Ruby
#
|
|
# Copyright 2011 Wade Alcorn wade@bindshell.net
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
module BeEF
|
|
module Core
|
|
module NetworkStack
|
|
module Handlers
|
|
|
|
#
|
|
# Class defining BeEF assets
|
|
#
|
|
class AssetHandler
|
|
|
|
# call BeEF::Core::NetworkStack::Handlers::AssetHandler.instance
|
|
include Singleton
|
|
|
|
attr_reader :allocations, :root_dir
|
|
|
|
def initialize
|
|
@allocations = {}
|
|
@http_server = BeEF::Core::Server.instance
|
|
@root_dir = File.expand_path('../../../../', __FILE__)
|
|
end
|
|
|
|
#
|
|
# Binds a file to a mount point
|
|
#
|
|
def bind(file, path=nil, extension=nil, count=-1)
|
|
url = buildURL(path, extension)
|
|
@allocations[url] = {'file' => "#{root_dir}"+file, 'path' => path, 'extension' => extension, 'count' => count}
|
|
@http_server.mount(url, true, WEBrick::HTTPServlet::FileHandler, @allocations[url]['file'])
|
|
print_info "File [" + "#{root_dir}"+file + "] bound to url [" + url + "]"
|
|
return url
|
|
end
|
|
|
|
#
|
|
# Unbinds a file from a mount point
|
|
#
|
|
def unbind(url)
|
|
@allocations.delete(url)
|
|
@http_server.unmount(url, true)
|
|
end
|
|
|
|
#
|
|
# Builds a URL based on the path and extention, if neither are passed a random URL will be generated
|
|
#
|
|
def buildURL(path, extension, length=20)
|
|
url = (path == nil) ? '/'+rand(36**length).to_s(36) : path;
|
|
url += (extension == nil) ? '' : '.'+extension;
|
|
return url
|
|
end
|
|
|
|
#
|
|
# Checks if the file is allocated, if the file isn't return true to pass onto FileHandler.
|
|
#
|
|
def check(url)
|
|
if @allocations.has_key?(url)
|
|
count = @allocations[url]['count']
|
|
if count == -1
|
|
return true
|
|
end
|
|
if count > 0
|
|
if (count - 1) == 0
|
|
unbind(url)
|
|
else
|
|
@allocations[url]['count'] = count - 1
|
|
end
|
|
return true
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
private
|
|
@http_server
|
|
@allocations
|
|
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|
|
end
|