content_decoder_tool.cc 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
  1. // Copyright 2016 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "net/tools/content_decoder_tool/content_decoder_tool.h"
  5. #include <memory>
  6. #include <utility>
  7. #include "base/containers/adapters.h"
  8. #include "base/logging.h"
  9. #include "base/strings/string_util.h"
  10. #include "net/base/completion_once_callback.h"
  11. #include "net/base/io_buffer.h"
  12. #include "net/base/test_completion_callback.h"
  13. #include "net/filter/brotli_source_stream.h"
  14. #include "net/filter/gzip_source_stream.h"
  15. #include "net/filter/source_stream.h"
  16. namespace net {
  17. namespace {
  18. const int kBufferLen = 4096;
  19. const char kDeflate[] = "deflate";
  20. const char kGZip[] = "gzip";
  21. const char kXGZip[] = "x-gzip";
  22. const char kBrotli[] = "br";
  23. class StdinSourceStream : public SourceStream {
  24. public:
  25. explicit StdinSourceStream(std::istream* input_stream)
  26. : SourceStream(SourceStream::TYPE_NONE), input_stream_(input_stream) {}
  27. StdinSourceStream(const StdinSourceStream&) = delete;
  28. StdinSourceStream& operator=(const StdinSourceStream&) = delete;
  29. ~StdinSourceStream() override = default;
  30. // SourceStream implementation.
  31. int Read(IOBuffer* dest_buffer,
  32. int buffer_size,
  33. CompletionOnceCallback callback) override {
  34. if (input_stream_->eof())
  35. return OK;
  36. if (input_stream_) {
  37. input_stream_->read(dest_buffer->data(), buffer_size);
  38. int bytes = input_stream_->gcount();
  39. return bytes;
  40. }
  41. return ERR_FAILED;
  42. }
  43. std::string Description() const override { return ""; }
  44. bool MayHaveMoreBytes() const override { return true; }
  45. private:
  46. std::istream* input_stream_;
  47. };
  48. } // namespace
  49. // static
  50. bool ContentDecoderToolProcessInput(std::vector<std::string> content_encodings,
  51. std::istream* input_stream,
  52. std::ostream* output_stream) {
  53. std::unique_ptr<SourceStream> upstream(
  54. std::make_unique<StdinSourceStream>(input_stream));
  55. for (const auto& content_encoding : base::Reversed(content_encodings)) {
  56. std::unique_ptr<SourceStream> downstream;
  57. if (base::EqualsCaseInsensitiveASCII(content_encoding, kBrotli)) {
  58. downstream = CreateBrotliSourceStream(std::move(upstream));
  59. } else if (base::EqualsCaseInsensitiveASCII(content_encoding, kDeflate)) {
  60. downstream = GzipSourceStream::Create(std::move(upstream),
  61. SourceStream::TYPE_DEFLATE);
  62. } else if (base::EqualsCaseInsensitiveASCII(content_encoding, kGZip) ||
  63. base::EqualsCaseInsensitiveASCII(content_encoding, kXGZip)) {
  64. downstream = GzipSourceStream::Create(std::move(upstream),
  65. SourceStream::TYPE_GZIP);
  66. } else {
  67. LOG(ERROR) << "Unsupported decoder '" << content_encoding << "'.";
  68. return false;
  69. }
  70. if (downstream == nullptr) {
  71. LOG(ERROR) << "Couldn't create the decoder.";
  72. return false;
  73. }
  74. upstream = std::move(downstream);
  75. }
  76. if (!upstream) {
  77. LOG(ERROR) << "Couldn't create the decoder.";
  78. return false;
  79. }
  80. scoped_refptr<IOBuffer> read_buffer =
  81. base::MakeRefCounted<IOBufferWithSize>(kBufferLen);
  82. while (true) {
  83. TestCompletionCallback callback;
  84. int bytes_read =
  85. upstream->Read(read_buffer.get(), kBufferLen, callback.callback());
  86. if (bytes_read == ERR_IO_PENDING)
  87. bytes_read = callback.WaitForResult();
  88. if (bytes_read < 0) {
  89. LOG(ERROR) << "Couldn't decode stdin.";
  90. return false;
  91. }
  92. output_stream->write(read_buffer->data(), bytes_read);
  93. // If EOF is read, break out the while loop.
  94. if (bytes_read == 0)
  95. break;
  96. }
  97. return true;
  98. }
  99. } // namespace net