/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
 *
 * 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 "mozilla/BasePrincipal.h"
#include "nsTLSSocketProvider.h"
#include "nsNSSIOLayer.h"
#include "nsError.h"

using mozilla::OriginAttributes;

nsTLSSocketProvider::nsTLSSocketProvider() = default;

nsTLSSocketProvider::~nsTLSSocketProvider() = default;

NS_IMPL_ISUPPORTS(nsTLSSocketProvider, nsISocketProvider)

NS_IMETHODIMP
nsTLSSocketProvider::NewSocket(int32_t family, const char* host, int32_t port,
                               nsIProxyInfo* proxy,
                               const OriginAttributes& originAttributes,
                               uint32_t flags, uint32_t tlsFlags,
                               PRFileDesc** _result,
                               nsITLSSocketControl** tlsSocketControl) {
  nsresult rv =
      nsSSLIOLayerNewSocket(family, host, port, proxy, originAttributes,
                            _result, tlsSocketControl, true, flags, tlsFlags);

  return (NS_FAILED(rv)) ? NS_ERROR_SOCKET_CREATE_FAILED : NS_OK;
}

// Add the SSL IO layer to an existing socket
NS_IMETHODIMP
nsTLSSocketProvider::AddToSocket(int32_t family, const char* host, int32_t port,
                                 nsIProxyInfo* proxy,
                                 const OriginAttributes& originAttributes,
                                 uint32_t flags, uint32_t tlsFlags,
                                 PRFileDesc* aSocket,
                                 nsITLSSocketControl** tlsSocketControl) {
  nsresult rv =
      nsSSLIOLayerAddToSocket(family, host, port, proxy, originAttributes,
                              aSocket, tlsSocketControl, true, flags, tlsFlags);

  return (NS_FAILED(rv)) ? NS_ERROR_SOCKET_CREATE_FAILED : NS_OK;
}
