ModelFileUtils.cpp

Enable keyboard shortcuts

/* -*- Mode: C++; tab-width: 8; 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 "ModelFileUtils.h"

#include "mozilla/ErrorNames.h"

#include "mozilla/ErrorResult.h"

#include "mozilla/Logging.h"

#include "mozilla/dom/BindingUtils.h"

#include "mozilla/dom/Blob.h"

#include "mozilla/dom/BlobBinding.h"

#include "mozilla/ipc/FileDescriptor.h"

#include "nsIFileStreams.h"

#include "nsIInputStream.h"

#include "prio.h"

#include "private/pprio.h"

namespace mozilla::hwinference {

extern LazyLogModule gHWInferenceLog;

#define LOGE(...) MOZ_LOG_FMT(gHWInferenceLog, LogLevel::Error, __VA_ARGS__)

nsresult BlobJSObjectToFileDescriptor(JSContext* aCx,

                                      JS::Handle<JS::Value> aValue,

                                      ipc::FileDescriptor* aDesc) {

  if (!aValue.isObject()) {

    return NS_ERROR_UNEXPECTED;

  RefPtr<dom::Blob> blob;

  nsresult rv = UNWRAP_OBJECT(Blob, &aValue.toObject(), blob);

  if (NS_FAILED(rv)) {

    LOGE("BlobJSObjectToFileDescriptor - ERROR: Failed to unwrap Blob: {}", rv);

    return rv;

  ErrorResult errorResult;

  nsCOMPtr<nsIInputStream> stream;

  blob->CreateInputStream(getter_AddRefs(stream), errorResult);

  if (errorResult.Failed()) {

    LOGE(

        "BlobJSObjectToFileDescriptor - ERROR: Failed to create input stream "

        "from blob");

    return NS_ERROR_UNEXPECTED;

  nsCOMPtr<nsIFileMetadata> fileMetadata = do_QueryInterface(stream);

  if (!fileMetadata) {

    LOGE(

        "BlobJSObjectToFileDescriptor - ERROR: Stream doesn't support "

        "nsIFileMetadata");

    return NS_ERROR_UNEXPECTED;

  PRFileDesc* fileDesc;

  nsresult getRv = fileMetadata->GetFileDescriptor(&fileDesc);

  if (NS_FAILED(getRv)) {

    LOGE("BlobJSObjectToFileDescriptor - ERROR: GetFileDescriptor failed: {}",

         getRv);

    return getRv;

  ipc::FileDescriptor fd(ipc::FileDescriptor::PlatformHandleType(

      PR_FileDesc2NativeHandle(fileDesc)));

  if (!fd.IsValid()) {

    LOGE("BlobJSObjectToFileDescriptor - ERROR: Failed to get native handle");

    return NS_ERROR_UNEXPECTED;

  *aDesc = std::move(fd);

  return NS_OK;

#undef LOGE

}  // namespace mozilla::hwinference