netwerk/base/ShutdownLayer.cpp
author Edwin Gao <egao@mozilla.com>
Wed, 13 Mar 2019 19:47:25 +0000
changeset 463893 15c16889e3b7f6e871a518b5f3254459ed5e0213
parent 448947 6f3709b3878117466168c40affa7bca0b60cf75b
child 472056 e1993a1f09ac53cd1a04fdf6a87f8cad8e44f73e
permissions -rw-r--r--
Bug 1531590, 1531598, 1534811, 1336075, 1531571, 1531572, 1531572, 1531574, 1534855, 1534857, 1535082 - skip tests in mochitest-dev-tools and mochitest-browser-chrome suites for windows10-aarch64 r=jmaher Bug 1531598 - disable browser_markup_copy_image_data.js Bug 1531598 - disable browser_markup_links_04.js Bug 1531598 - disable browser_inspector_menu-01-sensitivity.js Bug 1534811 - disable accessible/tests/browser/general Bug 1336075 - disable browser_largeAllocation_non_win32.js Bug 1531571 - disable browser_jsterm_context_menu_labels.js Bug 1531572 - disable browser_jsterm_helper_dollar_x.js Bug 1531573 - disable browser_jsterm_no_input_and_tab_key_pressed.js Bug 1531574 - disable browser_jsterm_syntax_highlight_output.js Bug 1534855 - disable accessible/tests/browser/ Bug 1534857 - disable browser_ext_slow_script.js Bug 1535082 - disable browser_jsterm_helper_dollar_dollar.js Differential Revision: https://phabricator.services.mozilla.com/D23369

/* -*- 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/Assertions.h"
#include "ShutdownLayer.h"
#include "prerror.h"
#include "private/pprio.h"
#include "prmem.h"
#include <winsock2.h>

static PRDescIdentity sWinSockShutdownLayerIdentity;
static PRIOMethods sWinSockShutdownLayerMethods;
static PRIOMethods *sWinSockShutdownLayerMethodsPtr = nullptr;

namespace mozilla {
namespace net {

extern PRDescIdentity nsNamedPipeLayerIdentity;

}  // namespace net
}  // namespace mozilla

PRStatus WinSockClose(PRFileDesc *aFd) {
  MOZ_RELEASE_ASSERT(aFd->identity == sWinSockShutdownLayerIdentity,
                     "Windows shutdown layer not on the top of the stack");

  PROsfd osfd = PR_FileDesc2NativeHandle(aFd);
  if (osfd != -1) {
    shutdown(osfd, SD_BOTH);
  }

  aFd->identity = PR_INVALID_IO_LAYER;

  if (aFd->lower) {
    return aFd->lower->methods->close(aFd->lower);
  } else {
    return PR_SUCCESS;
  }
}

nsresult mozilla::net::AttachShutdownLayer(PRFileDesc *aFd) {
  if (!sWinSockShutdownLayerMethodsPtr) {
    sWinSockShutdownLayerIdentity =
        PR_GetUniqueIdentity("windows shutdown call layer");
    sWinSockShutdownLayerMethods = *PR_GetDefaultIOMethods();
    sWinSockShutdownLayerMethods.close = WinSockClose;
    sWinSockShutdownLayerMethodsPtr = &sWinSockShutdownLayerMethods;
  }

  if (mozilla::net::nsNamedPipeLayerIdentity &&
      PR_GetIdentitiesLayer(aFd, mozilla::net::nsNamedPipeLayerIdentity)) {
    // Do not attach shutdown layer on named pipe layer,
    // it is for PR_NSPR_IO_LAYER only.
    return NS_OK;
  }

  PRFileDesc *layer;
  PRStatus status;

  layer = PR_CreateIOLayerStub(sWinSockShutdownLayerIdentity,
                               sWinSockShutdownLayerMethodsPtr);
  if (!layer) {
    return NS_OK;
  }

  status = PR_PushIOLayer(aFd, PR_NSPR_IO_LAYER, layer);

  if (status == PR_FAILURE) {
    PR_Free(layer);  // PR_CreateIOLayerStub() uses PR_Malloc().
    return NS_ERROR_FAILURE;
  }
  return NS_OK;
}