forked from chef/chef
/
ftp.rb
95 lines (85 loc) · 2.91 KB
/
ftp.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#
# Author:: Jesse Campbell (<hikeit@gmail.com>)
# Copyright:: Copyright (c) 2013 Jesse Campbell
# License:: Apache License, Version 2.0
#
# 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.
#
require 'uri'
require 'tempfile'
require 'net/ftp'
require 'chef/provider/remote_file'
class Chef
class Provider
class RemoteFile
class FTP
# Fetches the file at uri using Net::FTP, returning a Tempfile
def self.fetch(uri, ftp_active_mode)
self.new(uri, ftp_active_mode).fetch()
end
# Parse the uri into instance variables
def initialize(uri, ftp_active_mode)
@directories, @filename = parse_path(uri.path)
@typecode = uri.typecode
# Only support ascii and binary types
if @typecode && /\A[ai]\z/ !~ @typecode
raise ArgumentError, "invalid typecode: #{@typecode.inspect}"
end
@ftp_active_mode = ftp_active_mode
@hostname = uri.hostname
@port = uri.port
if uri.userinfo
@user = URI.unescape(uri.user)
@pass = URI.unescape(uri.password)
else
@user = 'anonymous'
@pass = nil
end
end
# Fetches using Net::FTP, returns a Tempfile with the content
def fetch()
tempfile = Tempfile.new(@filename)
# The access sequence is defined by RFC 1738
ftp = Net::FTP.new
ftp.connect(@hostname, @port)
ftp.passive = !@ftp_active_mode
ftp.login(@user, @pass)
@directories.each do |cwd|
ftp.voidcmd("CWD #{cwd}")
end
if @typecode
ftp.voidcmd("TYPE #{@typecode.upcase}")
end
ftp.getbinaryfile(@filename, tempfile.path)
ftp.close
tempfile
end
private
def parse_path(path)
path = path.sub(%r{\A/}, '%2F') # re-encode the beginning slash because uri library decodes it.
directories = path.split(%r{/}, -1)
directories.each {|d|
d.gsub!(/%([0-9A-Fa-f][0-9A-Fa-f])/) { [$1].pack("H2") }
}
unless filename = directories.pop
raise ArgumentError, "no filename: #{path.inspect}"
end
if filename.length == 0 || filename.end_with?( "/" )
raise ArgumentError, "no filename: #{path.inspect}"
end
return directories, filename
end
end
end
end
end