--- server/lib/sinatra/sinatra_verbose.rb | 8 ++++++++ 1 files changed, 8 insertions(+), 0 deletions(-)
diff --git a/server/lib/sinatra/sinatra_verbose.rb b/server/lib/sinatra/sinatra_verbose.rb index c016ec0..043463a 100644 --- a/server/lib/sinatra/sinatra_verbose.rb +++ b/server/lib/sinatra/sinatra_verbose.rb @@ -39,6 +39,13 @@ module Sinatra end + def check_host_header + disable :logging + before { + halt 400, "Unable to find HTTP Host header" if @env['HTTP_HOST'] == nil + } + end + def enable_verbose_logging! disable :logging before { @@ -62,6 +69,7 @@ module Sinatra def self.registered(app) app.helpers VerboseLogger::Helpers app.enable_verbose_logging! if ENV['API_VERBOSE'] + app.check_host_header end end end -- 1.7.1