Module: Rmega::Nodes::Uploadable
Instance Method Summary collapse
- #encrypt_chunck(start, clean_buffer, aes_key, nonce) ⇒ Object
- #read_chunk(file, start, size) ⇒ Object
- #upload(path) ⇒ Object
- #upload_chunk(base_url, start, buffer) ⇒ Object
- #upload_dir(dir) ⇒ Object
Methods included from Rmega::Net
#http_get_content, #http_post, #survive
Methods included from Options
Methods included from Loggable
Instance Method Details
#encrypt_chunck(start, clean_buffer, aes_key, nonce) ⇒ Object
23 24 25 26 27 28 29 30 31 32 |
# File 'lib/rmega/nodes/uploadable.rb', line 23 def encrypt_chunck(start, clean_buffer, aes_key, nonce) iv = nonce + [start/0x1000000000, start/0x10].pack('l>*') enc_data = aes_ctr_encrypt(aes_key, clean_buffer, iv) # calculate mac mac_iv = nonce * 2 mac = aes_cbc_mac(aes_key, clean_buffer, mac_iv) return [enc_data, mac] end |
#read_chunk(file, start, size) ⇒ Object
18 19 20 21 |
# File 'lib/rmega/nodes/uploadable.rb', line 18 def read_chunk(file, start, size) file.seek(start) file.read(size) end |
#upload(path) ⇒ Object
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 |
# File 'lib/rmega/nodes/uploadable.rb', line 34 def upload(path) path = ::File.(path) filesize = ::File.size(path) raise "Empty file - #{path}" if filesize == 0 file = ::File.open(path, 'rb') rnd_node_key = NodeKey.random file_handle = nil base_url = upload_url(filesize) pool = Pool.new read_mutex = Mutex.new progress = Progress.new(filesize, caption: 'Upload', filename: ::File.basename(path)) chunk_macs = {} self.class.each_chunk(filesize) do |start, size| pool.process do clean_buffer = nil read_mutex.synchronize do clean_buffer = read_chunk(file, start, size) end encrypted_buffer, chunk_mac = *encrypt_chunck(start, clean_buffer, rnd_node_key.aes_key, rnd_node_key.ctr_nonce) file_handle = upload_chunk(base_url, start, encrypted_buffer) chunk_macs[start] = chunk_mac progress.increment(size) end end pool.wait_done # encrypt attributes _attr = serialize_attributes(:n => Utils.utf8(::File.basename(path))) _attr = aes_cbc_encrypt(rnd_node_key.aes_key, _attr) # Calculate meta_mac file_mac = aes_cbc_mac(rnd_node_key.aes_key, chunk_macs.sort.map(&:last).join, "\x0"*16) rnd_node_key. = Utils.compact_to_8_bytes(file_mac) encrypted_key = aes_ecb_encrypt(session.master_key, rnd_node_key.generate) resp = request(a: 'p', t: handle, n: [ {h: file_handle, t: 0, a: Utils.base64urlencode(_attr), k: Utils.base64urlencode(encrypted_key)} ]) return Nodes::Factory.build(session, resp['f'][0]) ensure file.close if file end |
#upload_chunk(base_url, start, buffer) ⇒ Object
6 7 8 9 10 11 12 13 14 15 16 |
# File 'lib/rmega/nodes/uploadable.rb', line 6 def upload_chunk(base_url, start, buffer) size = buffer.length stop = start + size - 1 url = "#{base_url}/#{start}-#{stop}" survive do response = http_post(url, buffer) raise("Upload failed") if response.code.to_i != 200 return response.body end end |
#upload_dir(dir) ⇒ Object
89 90 91 92 93 94 95 96 97 98 99 100 101 |
# File 'lib/rmega/nodes/uploadable.rb', line 89 def upload_dir(dir) return create_folder(::File.basename(dir)).upload_dir(dir) unless name == ::File.basename(dir) ::Dir["#{dir}/*"].each do |path| name = ::File.basename(path) if ::File.directory?(path) create_folder(name).upload_dir(path) elsif ::File.size(path) > 0 upload(path) end end end |