Apple-2-SW/emailler
1
0
Fork 0
mirror of https://github.com/bobbimanners/emailler.git synced 2024-07-12 11:28:54 +00:00
Code Issues Projects Releases Wiki Activity

tftp server extended to support DIR opcode

Browse Source 
git-svn-id: http://svn.code.sf.net/p/netboot65/code@24 93682198-c243-4bdb-bd91-e943c89aac3b
This commit is contained in:
jonnosan 2009-01-25 01:53:51 +00:00
parent 870fa89600
commit 67eb982b59
2 changed files with 21 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/bin/netboot65_server.rb
View File

@ -20,7 +20,7 @@ bootfile_dir=File.expand_path(File.dirname(__FILE__)+'/../boot')
tftp_server=Netboot65TFTPServer.new(bootfile_dir) tftp_server=Netboot65TFTPServer.new(bootfile_dir)
tftp_server.start tftp_server.start
tndp_server=TNDPServer.new(File.dirname(__FILE__)+"/../test_images") tndp_server=TNDPServer.new(File.dirname(__FILE__)+"/../file_system_images")
tndp_server.start tndp_server.start
begin begin
loop do loop do

23
server/lib/tftp_server.rb
View File

@ -1,6 +1,8 @@
# #
# minimal TFTP server implementation for use with netboot65 # minimal TFTP server implementation for use with netboot65
# #
# supports RRQ and DIR requests per http://www.watersprings.org/pub/id/draft-johnston-tftp-directory-01.txt
# however, DIR "information string" contains file size only (as an ASCII string), NOT the full UTC timestamp
# Jonno Downes (jonno@jamtronix.com) - January, 2009 # Jonno Downes (jonno@jamtronix.com) - January, 2009
# #
# #
@ -14,6 +16,7 @@ class Netboot65TFTPServer
3=>'DATA', 3=>'DATA',
4=>'ACK', 4=>'ACK',
5=>'ERROR', 5=>'ERROR',
7=>'DIR',
} }
TFTP_ERRORCODES={ TFTP_ERRORCODES={
@ -41,11 +44,11 @@ class Netboot65TFTPServer
log_msg("sent error #{error_code}:'#{error_msg}' to #{client_ip}:#{client_port}") log_msg("sent error #{error_code}:'#{error_msg}' to #{client_ip}:#{client_port}")
end end
def send_file(client_ip,client_port,filename) def send_data(client_ip,client_port,filename,data_to_send)
client_sock=UDPSocket.open client_sock=UDPSocket.open
client_sock.connect(client_ip,client_port) client_sock.connect(client_ip,client_port)
data_to_send=File.open(filename,"rb").read
blocks_to_send=(data_to_send.length.to_f/512.0).ceil blocks_to_send=(data_to_send.length.to_f/512.0).ceil
log_msg("sending #{filename} to #{client_ip}:#{client_port} (#{blocks_to_send} blocks)") log_msg("sending #{filename} to #{client_ip}:#{client_port} (#{blocks_to_send} blocks)")
blocks_to_send.times do |block_number| blocks_to_send.times do |block_number|
@ -106,11 +109,25 @@ class Netboot65TFTPServer
else else
full_filename="#{bootfile_dir}/#{filename}" full_filename="#{bootfile_dir}/#{filename}"
if File.file?(full_filename) then if File.file?(full_filename) then
Thread.new {send_file(client_ip,client_port,full_filename)} data_to_send=File.open(full_filename,"rb").read
Thread.new {send_data(client_ip,client_port,full_filename,data_to_send)}
else else
send_error(client_ip,client_port,1,"'#{filename}' not found") send_error(client_ip,client_port,1,"'#{filename}' not found")
end end
end end
when 7 : #DIR REQUEST
opcode,filemask_and_mode=data.unpack("nA*")
filemask,mode=filemask_and_mode.split(0.chr)
log_msg "DIR for #{filemask} (#{mode})"
if filename=~/^\./ || filename=~/\.\./ then #looks like something dodgy - either a dotfile or a directory traversal attempt
send_error(client_ip,client_port,1,"'#{filemask}' invalid")
else
data_to_send=""
Dir.chdir(bootfile_dir) do
Dir.glob(filemask).each {|filename| data_to_send<<"#{filename}\000#{File.size(filename)}\000"}
end
Thread.new {send_data(client_ip,client_port,full_filename,data_to_send)}
end
else else
send_error(client_ip,client_port,4,"opcode #{opcode} not supported") send_error(client_ip,client_port,4,"opcode #{opcode} not supported")
end end