mirror of
https://github.com/denoland/deno.git
synced 2024-12-23 07:44:48 -05:00
9bfb0df805
Co-authored-by: Bartek Iwańczuk <biwanczuk@gmail.com>
138 lines
4.4 KiB
Rust
138 lines
4.4 KiB
Rust
// Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.
|
|
use super::dispatch_json::{JsonOp, Value};
|
|
use crate::colors;
|
|
use crate::state::State;
|
|
use crate::version;
|
|
use crate::DenoSubcommand;
|
|
use deno_core::CoreIsolate;
|
|
use deno_core::ErrBox;
|
|
use deno_core::ModuleSpecifier;
|
|
use deno_core::ZeroCopyBuf;
|
|
use std::env;
|
|
use std::rc::Rc;
|
|
|
|
pub fn init(i: &mut CoreIsolate, s: &Rc<State>) {
|
|
i.register_op("op_start", s.stateful_json_op(op_start));
|
|
i.register_op("op_main_module", s.stateful_json_op(op_main_module));
|
|
i.register_op("op_metrics", s.stateful_json_op(op_metrics));
|
|
}
|
|
|
|
fn op_start(
|
|
state: &Rc<State>,
|
|
_args: Value,
|
|
_zero_copy: &mut [ZeroCopyBuf],
|
|
) -> Result<JsonOp, ErrBox> {
|
|
let gs = &state.global_state;
|
|
|
|
Ok(JsonOp::Sync(json!({
|
|
// TODO(bartlomieju): `cwd` field is not used in JS, remove?
|
|
"args": gs.flags.argv.clone(),
|
|
"cwd": &env::current_dir().unwrap(),
|
|
"debugFlag": gs.flags.log_level.map_or(false, |l| l == log::Level::Debug),
|
|
"denoVersion": version::DENO,
|
|
"noColor": !colors::use_color(),
|
|
"pid": std::process::id(),
|
|
"ppid": ppid(),
|
|
"repl": gs.flags.subcommand == DenoSubcommand::Repl,
|
|
"target": env!("TARGET"),
|
|
"tsVersion": version::TYPESCRIPT,
|
|
"unstableFlag": gs.flags.unstable,
|
|
"v8Version": version::v8(),
|
|
"versionFlag": gs.flags.version,
|
|
})))
|
|
}
|
|
|
|
fn op_main_module(
|
|
state: &Rc<State>,
|
|
_args: Value,
|
|
_zero_copy: &mut [ZeroCopyBuf],
|
|
) -> Result<JsonOp, ErrBox> {
|
|
let main = &state.main_module.to_string();
|
|
let main_url = ModuleSpecifier::resolve_url_or_path(&main)?;
|
|
if main_url.as_url().scheme() == "file" {
|
|
let main_path = std::env::current_dir().unwrap().join(main_url.to_string());
|
|
state.check_read_blind(&main_path, "main_module")?;
|
|
}
|
|
Ok(JsonOp::Sync(json!(&main)))
|
|
}
|
|
|
|
fn op_metrics(
|
|
state: &Rc<State>,
|
|
_args: Value,
|
|
_zero_copy: &mut [ZeroCopyBuf],
|
|
) -> Result<JsonOp, ErrBox> {
|
|
let m = &state.metrics.borrow();
|
|
|
|
Ok(JsonOp::Sync(json!({
|
|
"opsDispatched": m.ops_dispatched,
|
|
"opsDispatchedSync": m.ops_dispatched_sync,
|
|
"opsDispatchedAsync": m.ops_dispatched_async,
|
|
"opsDispatchedAsyncUnref": m.ops_dispatched_async_unref,
|
|
"opsCompleted": m.ops_completed,
|
|
"opsCompletedSync": m.ops_completed_sync,
|
|
"opsCompletedAsync": m.ops_completed_async,
|
|
"opsCompletedAsyncUnref": m.ops_completed_async_unref,
|
|
"bytesSentControl": m.bytes_sent_control,
|
|
"bytesSentData": m.bytes_sent_data,
|
|
"bytesReceived": m.bytes_received
|
|
})))
|
|
}
|
|
|
|
fn ppid() -> Value {
|
|
#[cfg(windows)]
|
|
{
|
|
// Adopted from rustup:
|
|
// https://github.com/rust-lang/rustup/blob/1.21.1/src/cli/self_update.rs#L1036
|
|
// Copyright Diggory Blake, the Mozilla Corporation, and rustup contributors.
|
|
// Licensed under either of
|
|
// - Apache License, Version 2.0
|
|
// - MIT license
|
|
use std::mem;
|
|
use winapi::shared::minwindef::DWORD;
|
|
use winapi::um::handleapi::{CloseHandle, INVALID_HANDLE_VALUE};
|
|
use winapi::um::processthreadsapi::GetCurrentProcessId;
|
|
use winapi::um::tlhelp32::{
|
|
CreateToolhelp32Snapshot, Process32First, Process32Next, PROCESSENTRY32,
|
|
TH32CS_SNAPPROCESS,
|
|
};
|
|
unsafe {
|
|
// Take a snapshot of system processes, one of which is ours
|
|
// and contains our parent's pid
|
|
let snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
|
|
if snapshot == INVALID_HANDLE_VALUE {
|
|
return serde_json::to_value(-1).unwrap();
|
|
}
|
|
|
|
let mut entry: PROCESSENTRY32 = mem::zeroed();
|
|
entry.dwSize = mem::size_of::<PROCESSENTRY32>() as DWORD;
|
|
|
|
// Iterate over system processes looking for ours
|
|
let success = Process32First(snapshot, &mut entry);
|
|
if success == 0 {
|
|
CloseHandle(snapshot);
|
|
return serde_json::to_value(-1).unwrap();
|
|
}
|
|
|
|
let this_pid = GetCurrentProcessId();
|
|
while entry.th32ProcessID != this_pid {
|
|
let success = Process32Next(snapshot, &mut entry);
|
|
if success == 0 {
|
|
CloseHandle(snapshot);
|
|
return serde_json::to_value(-1).unwrap();
|
|
}
|
|
}
|
|
CloseHandle(snapshot);
|
|
|
|
// FIXME: Using the process ID exposes a race condition
|
|
// wherein the parent process already exited and the OS
|
|
// reassigned its ID.
|
|
let parent_id = entry.th32ParentProcessID;
|
|
serde_json::to_value(parent_id).unwrap()
|
|
}
|
|
}
|
|
#[cfg(not(windows))]
|
|
{
|
|
use std::os::unix::process::parent_id;
|
|
serde_json::to_value(parent_id()).unwrap()
|
|
}
|
|
}
|