another go at file upload: slightly less naive and _working_

This commit is contained in:
Alexey Verkhovsky 2005-01-22 19:30:49 +00:00
parent 7dc399650f
commit 295e41c245
2 changed files with 19 additions and 9 deletions

View file

@ -71,7 +71,7 @@ class ApplicationController < ActionController::Base
not @web_name.nil? not @web_name.nil?
end end
@@REMEMBER_NOT = ['locked', 'save', 'back'] @@REMEMBER_NOT = ['locked', 'save', 'back', 'file']
def remember_location def remember_location
if @response.headers['Status'] == '200 OK' if @response.headers['Status'] == '200 OK'
unless @@REMEMBER_NOT.include? action_name or @request.method != :get unless @@REMEMBER_NOT.include? action_name or @request.method != :get

View file

@ -7,10 +7,12 @@ class FileController < ApplicationController
layout 'default' layout 'default'
def file def file
sanitize_file_name
if @params['file'] if @params['file']
# received a file # form supplied
File.open(file_name) { |f| f.write(@params['file'].read } upload_file
flash[:info] = "File '#{@file_name}' successfully uploaded"
return_to_last_remembered
elsif have_file? elsif have_file?
send_file(file_path) send_file(file_path)
else else
@ -23,22 +25,30 @@ class FileController < ApplicationController
return_to_last_remembered return_to_last_remembered
end end
def
private private
def have_file? def have_file?
sanitize_file_name
File.file?(file_path) File.file?(file_path)
end end
SANE_FILE_NAME = /[-_A-Za-z0-9]{1,255}/ def upload_file
if @params['file'].kind_of?(Tempfile)
@params['file'].close
FileUtils.mv(@params['file'].path, file_path)
elsif @params['file'].kind_of?(IO)
File.open(file_path, 'wb') { |f| f.write(@params['file'].read) }
else
raise 'File to be uploaded is not an IO object'
end
end
SANE_FILE_NAME = /[-_\.A-Za-z0-9]{1,255}/
def sanitize_file_name def sanitize_file_name
raise Instiki::ValidationError.new("Invalid path: no file name") unless @file_name raise Instiki::ValidationError.new("Invalid path: no file name") unless @file_name
unless @file_name =~ SANE_FILE_NAME unless @file_name =~ SANE_FILE_NAME
raise ValidationError.new("Invalid file name: '#{@file_name}'.\n" + raise ValidationError.new("Invalid file name: '#{@file_name}'.\n" +
"Only latin characters, digits, underscores and dashes are accepted.") "Only latin characters, digits, dots, underscores and dashes are accepted.")
end end
end end