Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 /* | 1 /* |
| 2 * Copyright 2009, Google Inc. | 2 * Copyright 2009, Google Inc. |
| 3 * All rights reserved. | 3 * All rights reserved. |
| 4 * | 4 * |
| 5 * Redistribution and use in source and binary forms, with or without | 5 * Redistribution and use in source and binary forms, with or without |
| 6 * modification, are permitted provided that the following conditions are | 6 * modification, are permitted provided that the following conditions are |
| 7 * met: | 7 * met: |
| 8 * | 8 * |
| 9 * * Redistributions of source code must retain the above copyright | 9 * * Redistributions of source code must retain the above copyright |
| 10 * notice, this list of conditions and the following disclaimer. | 10 * notice, this list of conditions and the following disclaimer. |
| (...skipping 19 matching lines...) Expand all Loading... | |
| 30 */ | 30 */ |
| 31 | 31 |
| 32 // | 32 // |
| 33 // TarProcessor processes a tar byte stream (uncompressed). | 33 // TarProcessor processes a tar byte stream (uncompressed). |
| 34 | 34 |
| 35 #include "base/logging.h" | 35 #include "base/logging.h" |
| 36 #include "import/cross/tar_processor.h" | 36 #include "import/cross/tar_processor.h" |
| 37 | 37 |
| 38 namespace o3d { | 38 namespace o3d { |
| 39 | 39 |
| 40 static const int kFileSizeOffset = 124; | |
| 41 static const int kLinkFlagOffset = 156; | |
| 42 | |
| 40 // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ | 43 // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
| 41 int TarProcessor::ProcessBytes(MemoryReadStream *stream, size_t n) { | 44 int TarProcessor::ProcessBytes(MemoryReadStream *stream, size_t n) { |
| 42 // Keep processing the byte-stream until we've consumed all we're given | 45 // Keep processing the byte-stream until we've consumed all we're given |
| 43 // | 46 // |
| 44 size_t bytes_to_consume = n; | 47 size_t bytes_to_consume = n; |
| 45 | 48 |
| 46 while (bytes_to_consume > 0) { | 49 while (bytes_to_consume > 0) { |
| 47 // First see if we have any more header bytes to read | 50 // First see if we have any more header bytes to read |
| 48 if (header_bytes_read_ < TAR_HEADER_SIZE) { | 51 if (header_bytes_read_ < TAR_HEADER_SIZE) { |
| 49 // Read header bytes | 52 // Read header bytes |
| 50 size_t header_bytes_remaining = TAR_HEADER_SIZE - header_bytes_read_; | 53 size_t header_bytes_remaining = TAR_HEADER_SIZE - header_bytes_read_; |
| 51 size_t bytes_to_read = std::min(bytes_to_consume, header_bytes_remaining); | 54 size_t bytes_to_read = std::min(bytes_to_consume, header_bytes_remaining); |
| 52 size_t bytes_read = | 55 size_t bytes_read = |
| 53 stream->Read(reinterpret_cast<uint8*>(header_ + header_bytes_read_), | 56 stream->Read(reinterpret_cast<uint8*>(header_ + header_bytes_read_), |
| 54 bytes_to_read); | 57 bytes_to_read); |
| 55 if (bytes_read != bytes_to_read) { | 58 if (bytes_read != bytes_to_read) { |
| 56 return -1; | 59 return -1; |
| 57 } | 60 } |
| 58 | 61 |
| 59 header_bytes_read_ += bytes_to_read; | 62 header_bytes_read_ += bytes_to_read; |
| 60 bytes_to_consume -= bytes_to_read; | 63 bytes_to_consume -= bytes_to_read; |
| 61 | 64 |
| 62 if (header_bytes_read_ == TAR_HEADER_SIZE) { | 65 if (header_bytes_read_ == TAR_HEADER_SIZE) { |
| 63 const char *filename = (const char *)header_; | |
| 64 | |
| 65 // The tar format stupidly represents size_teger values as | 66 // The tar format stupidly represents size_teger values as |
|
Chris Rogers
2009/07/21 19:14:28
might as well fix this comment now "size_teger"
| |
| 66 // octal strings!! | 67 // octal strings!! |
| 67 size_t file_size = 0; | 68 size_t file_size = 0; |
| 68 sscanf(header_ + 124, "%o", &file_size); | 69 sscanf(header_ + kFileSizeOffset, "%o", &file_size); |
| 69 | 70 |
| 70 // Only callback client if this is a "real" header | 71 // Check if it's a long filename |
| 71 // (filename is not NULL) | 72 char type = header_[kLinkFlagOffset]; |
| 72 // Extra zero-padding can be added by the gzip compression | 73 if (type == 'L') { |
| 73 // (at end of archive), so ignore these ones. | 74 getting_filename_ = true; |
| 74 // | 75 // We should pick some size that's too large. |
| 75 // Also, ignore entries for directories (which have zero size) | 76 if (file_size > 1024) { |
| 76 if (header_[0] != 0 && file_size > 0) { | 77 return -1; |
| 77 ArchiveFileInfo info(filename, file_size); | 78 } |
| 78 callback_client_->ReceiveFileHeader(info); | 79 } else { |
| 79 } else if (header_[0] == 0) { | 80 getting_filename_ = false; |
| 80 // If filename is NULL due to zero-padding then file size | 81 const char *filename = (const char *)header_; |
| 81 // should also be NULL | 82 if (!file_name_.empty()) { |
| 82 assert(file_size == 0); | 83 filename = file_name_.c_str(); |
| 84 } | |
| 85 | |
| 86 // Only callback client if this is a "real" header | |
| 87 // (filename is not NULL) | |
| 88 // Extra zero-padding can be added by the gzip compression | |
| 89 // (at end of archive), so ignore these ones. | |
| 90 // | |
| 91 // Also, ignore entries for directories (which have zero size) | |
| 92 if (header_[0] != 0 && file_size > 0) { | |
| 93 ArchiveFileInfo info(filename, file_size); | |
| 94 callback_client_->ReceiveFileHeader(info); | |
| 95 } else if (header_[0] == 0) { | |
| 96 // If filename is NULL due to zero-padding then file size | |
| 97 // should also be NULL | |
| 98 // TODO(gman): Won't this crash the plugin if I make a bad tar? | |
| 99 assert(file_size == 0); | |
| 100 } | |
| 83 } | 101 } |
| 84 | 102 |
| 85 // Round filesize up to nearest block size | 103 // Round filesize up to nearest block size |
| 86 file_bytes_to_read_ = | 104 file_bytes_to_read_ = |
| 87 (file_size + TAR_BLOCK_SIZE - 1) & ~(TAR_BLOCK_SIZE - 1); | 105 (file_size + TAR_BLOCK_SIZE - 1) & ~(TAR_BLOCK_SIZE - 1); |
| 88 | 106 |
| 89 // Our client doesn't want to be bothered with the block padding, | 107 // Our client doesn't want to be bothered with the block padding, |
| 90 // so only send him the actual file bytes | 108 // so only send him the actual file bytes |
| 91 client_file_bytes_to_read_ = file_size; | 109 client_file_bytes_to_read_ = file_size; |
| 110 | |
| 111 // Clear the file_name_ so we don't use it next time. | |
| 112 file_name_.clear(); | |
| 92 } | 113 } |
| 93 } | 114 } |
| 94 | 115 |
| 95 if (bytes_to_consume > 0) { | 116 if (bytes_to_consume > 0) { |
| 96 // Looks like we have some left-over bytes past the header | 117 // Looks like we have some left-over bytes past the header |
| 97 // -- size_terpret as file bytes | 118 // -- size_terpret as file bytes |
| 98 if (client_file_bytes_to_read_ > 0) { | 119 if (client_file_bytes_to_read_ > 0) { |
| 99 // Callback to client with some file data | 120 // Callback to client with some file data |
| 100 | 121 |
| 101 // Use a copy of the stream in case the client doesn't read | 122 // Use a copy of the stream in case the client doesn't read |
| 102 // the amount they're supposed to | 123 // the amount they're supposed to |
| 103 MemoryReadStream client_read_stream(*stream); | 124 MemoryReadStream client_read_stream(*stream); |
| 104 | 125 |
| 105 size_t client_bytes_this_time = | 126 size_t client_bytes_this_time = |
| 106 std::min(bytes_to_consume, client_file_bytes_to_read_); | 127 std::min(bytes_to_consume, client_file_bytes_to_read_); |
| 107 | 128 |
| 108 if (!callback_client_->ReceiveFileData(&client_read_stream, | 129 if (getting_filename_) { |
| 109 client_bytes_this_time)) { | 130 String name_piece( |
| 110 return -1; | 131 client_read_stream.GetDirectMemoryPointerAs<const char>(), |
| 132 client_bytes_this_time); | |
| 133 client_read_stream.Skip(client_bytes_this_time); | |
| 134 file_name_ += name_piece; | |
| 135 } else { | |
| 136 if (!callback_client_->ReceiveFileData(&client_read_stream, | |
| 137 client_bytes_this_time)) { | |
| 138 return -1; | |
| 139 } | |
| 111 } | 140 } |
| 112 | 141 |
| 113 client_file_bytes_to_read_ -= client_bytes_this_time; | 142 client_file_bytes_to_read_ -= client_bytes_this_time; |
| 114 file_bytes_to_read_ -= client_bytes_this_time; | 143 file_bytes_to_read_ -= client_bytes_this_time; |
| 115 | 144 |
| 116 // Advance stream the amount the client should have consumed | 145 // Advance stream the amount the client should have consumed |
| 117 stream->Skip(client_bytes_this_time); | 146 stream->Skip(client_bytes_this_time); |
| 118 | 147 |
| 119 bytes_to_consume -= client_bytes_this_time; | 148 bytes_to_consume -= client_bytes_this_time; |
| 120 } | 149 } |
| (...skipping 14 matching lines...) Expand all Loading... | |
| 135 // setting to 0 makes us want more header bytes | 164 // setting to 0 makes us want more header bytes |
| 136 header_bytes_read_ = 0; | 165 header_bytes_read_ = 0; |
| 137 } | 166 } |
| 138 } | 167 } |
| 139 } | 168 } |
| 140 | 169 |
| 141 return 0; | 170 return 0; |
| 142 } | 171 } |
| 143 | 172 |
| 144 } // namespace o3d | 173 } // namespace o3d |
| OLD | NEW |