/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* vim: set ts=8 sts=2 et sw=2 tw=80: */

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "gtest/gtest.h"

#include "Helpers.h"

#include "mozilla/Unused.h"

#include "nsICloneableInputStream.h"

#include "nsIMultiplexInputStream.h"

#include "nsNetUtil.h"

#include "nsStreamUtils.h"

#include "nsStringStream.h"

#include "nsComponentManagerUtils.h"

TEST(CloneInputStream, InvalidInput)

{

  nsCOMPtr<nsIInputStream> clone;

  nsresult rv = NS_CloneInputStream(nullptr, getter_AddRefs(clone));

  ASSERT_TRUE(NS_FAILED(rv));

  ASSERT_FALSE(clone);

}

TEST(CloneInputStream, CloneableInput)

{

  nsTArray<char> inputData;

  testing::CreateData(4 * 1024, inputData);

  nsDependentCSubstring inputString(inputData.Elements(), inputData.Length());

  nsCOMPtr<nsIInputStream> stream;

  nsresult rv = NS_NewCStringInputStream(getter_AddRefs(stream), inputString);

  ASSERT_TRUE(NS_SUCCEEDED(rv));

  nsCOMPtr<nsIInputStream> clone;

  rv = NS_CloneInputStream(stream, getter_AddRefs(clone));

  ASSERT_TRUE(NS_SUCCEEDED(rv));

  testing::ConsumeAndValidateStream(stream, inputString);

  testing::ConsumeAndValidateStream(clone, inputString);

}

class NonCloneableInputStream final : public nsIInputStream

{

public:

  NS_DECL_THREADSAFE_ISUPPORTS

  explicit NonCloneableInputStream(already_AddRefed<nsIInputStream> aInputStream)

    : mStream(aInputStream)

  {}

  NS_IMETHOD

  Available(uint64_t* aLength) override

  {

    return mStream->Available(aLength);

  }

  NS_IMETHOD

  Read(char* aBuffer, uint32_t aCount, uint32_t* aReadCount) override

  {

    return mStream->Read(aBuffer, aCount, aReadCount);

  }

  NS_IMETHOD

  ReadSegments(nsWriteSegmentFun aWriter, void* aClosure,

               uint32_t aCount, uint32_t *aResult) override

  {

    return mStream->ReadSegments(aWriter, aClosure, aCount, aResult);

  }

  NS_IMETHOD

  Close() override

  {

    return mStream->Close();

  }

  NS_IMETHOD

  IsNonBlocking(bool* aNonBlocking) override

  {

    return mStream->IsNonBlocking(aNonBlocking);

  }

private:

  ~NonCloneableInputStream() = default;

  nsCOMPtr<nsIInputStream> mStream;

};

NS_IMPL_ISUPPORTS(NonCloneableInputStream, nsIInputStream)

TEST(CloneInputStream, NonCloneableInput_NoFallback)

{

  nsTArray<char> inputData;

  testing::CreateData(4 * 1024, inputData);

  nsDependentCSubstring inputString(inputData.Elements(), inputData.Length());

  nsCOMPtr<nsIInputStream> base;

  nsresult rv = NS_NewCStringInputStream(getter_AddRefs(base), inputString);

  ASSERT_TRUE(NS_SUCCEEDED(rv));

  nsCOMPtr<nsIInputStream> stream = new NonCloneableInputStream(base.forget());

  nsCOMPtr<nsICloneableInputStream> cloneable = do_QueryInterface(stream);

  ASSERT_TRUE(cloneable == nullptr);

  nsCOMPtr<nsIInputStream> clone;

  rv = NS_CloneInputStream(stream, getter_AddRefs(clone));

  ASSERT_TRUE(NS_FAILED(rv));

  ASSERT_TRUE(clone == nullptr);

  testing::ConsumeAndValidateStream(stream, inputString);

}

TEST(CloneInputStream, NonCloneableInput_Fallback)

{

  nsTArray<char> inputData;

  testing::CreateData(4 * 1024, inputData);

  nsDependentCSubstring inputString(inputData.Elements(), inputData.Length());

  nsCOMPtr<nsIInputStream> base;

  nsresult rv = NS_NewCStringInputStream(getter_AddRefs(base), inputString);

  ASSERT_TRUE(NS_SUCCEEDED(rv));

  nsCOMPtr<nsIInputStream> stream = new NonCloneableInputStream(base.forget());

  nsCOMPtr<nsICloneableInputStream> cloneable = do_QueryInterface(stream);

  ASSERT_TRUE(cloneable == nullptr);

  nsCOMPtr<nsIInputStream> clone;

  nsCOMPtr<nsIInputStream> replacement;

  rv = NS_CloneInputStream(stream, getter_AddRefs(clone),

                           getter_AddRefs(replacement));

  ASSERT_TRUE(NS_SUCCEEDED(rv));

  ASSERT_TRUE(clone != nullptr);

  ASSERT_TRUE(replacement != nullptr);

  ASSERT_TRUE(stream.get() != replacement.get());

  ASSERT_TRUE(clone.get() != replacement.get());

  stream = replacement.forget();

  // The stream is being copied asynchronously on the STS event target.  Spin

  // a yield loop here until the data is available.  Yes, this is a bit hacky,

  // but AFAICT, gtest does not support async test completion.

  uint64_t available;

  do {

    mozilla::Unused << PR_Sleep(PR_INTERVAL_NO_WAIT);

    rv = stream->Available(&available);

    ASSERT_TRUE(NS_SUCCEEDED(rv));

  } while(available < inputString.Length());

  testing::ConsumeAndValidateStream(stream, inputString);

  testing::ConsumeAndValidateStream(clone, inputString);

}

TEST(CloneInputStream, CloneMultiplexStream)

{

  nsCOMPtr<nsIMultiplexInputStream> multiplexStream =

    do_CreateInstance("@mozilla.org/io/multiplex-input-stream;1");

  ASSERT_TRUE(multiplexStream);

  nsCOMPtr<nsIInputStream> stream(do_QueryInterface(multiplexStream));

  ASSERT_TRUE(stream);

  nsTArray<char> inputData;

  testing::CreateData(1024, inputData);

  for (uint32_t i = 0; i < 2; ++i) {

    nsCString inputString(inputData.Elements(), inputData.Length());

    nsCOMPtr<nsIInputStream> base;

    nsresult rv = NS_NewCStringInputStream(getter_AddRefs(base), inputString);

    ASSERT_TRUE(NS_SUCCEEDED(rv));

    rv = multiplexStream->AppendStream(base);

    ASSERT_TRUE(NS_SUCCEEDED(rv));

  }

  // Unread stream should clone successfully.

  nsTArray<char> doubled;

  doubled.AppendElements(inputData);

  doubled.AppendElements(inputData);

  nsCOMPtr<nsIInputStream> clone;

  nsresult rv = NS_CloneInputStream(stream, getter_AddRefs(clone));

  ASSERT_TRUE(NS_SUCCEEDED(rv));

  testing::ConsumeAndValidateStream(clone, doubled);

  // Stream that has been read should fail.

  char buffer[512];

  uint32_t read;

  rv = stream->Read(buffer, 512, &read);

  ASSERT_TRUE(NS_SUCCEEDED(rv));

  nsCOMPtr<nsIInputStream> clone2;

  rv = NS_CloneInputStream(stream, getter_AddRefs(clone2));

  ASSERT_TRUE(NS_FAILED(rv));

}

TEST(CloneInputStream, CloneMultiplexStreamPartial)

{

  nsCOMPtr<nsIMultiplexInputStream> multiplexStream =

    do_CreateInstance("@mozilla.org/io/multiplex-input-stream;1");

  ASSERT_TRUE(multiplexStream);

  nsCOMPtr<nsIInputStream> stream(do_QueryInterface(multiplexStream));

  ASSERT_TRUE(stream);

  nsTArray<char> inputData;

  testing::CreateData(1024, inputData);

  for (uint32_t i = 0; i < 2; ++i) {

    nsCString inputString(inputData.Elements(), inputData.Length());

    nsCOMPtr<nsIInputStream> base;

    nsresult rv = NS_NewCStringInputStream(getter_AddRefs(base), inputString);

    ASSERT_TRUE(NS_SUCCEEDED(rv));

    rv = multiplexStream->AppendStream(base);

    ASSERT_TRUE(NS_SUCCEEDED(rv));

  }

  // Fail when first stream read, but second hasn't been started.

  char buffer[1024];

  uint32_t read;

  nsresult rv = stream->Read(buffer, 1024, &read);

  ASSERT_TRUE(NS_SUCCEEDED(rv));

  nsCOMPtr<nsIInputStream> clone;

  rv = NS_CloneInputStream(stream, getter_AddRefs(clone));

  ASSERT_TRUE(NS_FAILED(rv));

  // Fail after beginning read of second stream.

  rv = stream->Read(buffer, 512, &read);

  ASSERT_TRUE(NS_SUCCEEDED(rv) && read == 512);

  rv = NS_CloneInputStream(stream, getter_AddRefs(clone));

  ASSERT_TRUE(NS_FAILED(rv));

  // Fail at the end.

  nsAutoCString consumed;

  rv = NS_ConsumeStream(stream, UINT32_MAX, consumed);

  ASSERT_TRUE(NS_SUCCEEDED(rv));

  rv = NS_CloneInputStream(stream, getter_AddRefs(clone));

  ASSERT_TRUE(NS_FAILED(rv));

}