深入出不来nodejs源码-内置模块引入初探

　　重新审视了一下上一篇的内容，配合源码发现有些地方说的不太对，或者不太严谨。

　　主要是关于内置模块引入的问题，当时我是这样描述的：

需要关注的只要那个RegisterBuiltinModules方法，从名字也可以看出来，就是加载内置模块。

　　然而并不是啊……从名字可以看出来，这只是一个注册方法。

　　Register：登记、注册。

　　因此，这里并不会真正加载内置模块，而只是做一个登记，表示有哪些模块一会要加载，统计一下。

　　上一节简单看了下该方法的宏，是一个_register_XX方法的批量调用，而该方法的定义地点还是在一个宏里面，源码如下所示：

#define NODE_MODULE_CONTEXT_AWARE_CPP(modname, regfunc, priv, flags)          \    
// 模块结构体
  static node::node_module _module = {                                        \
    NODE_MODULE_VERSION,   /*模块版本*/                                        \
    flags,                 /*模块类型：builtin、internal、linked*/              \
    nullptr,               /*不懂*/                                            \
    __FILE__,              /*不懂*/                                            \
    nullptr,               /*注册方法*/                                         \
    (node::addon_context_register_func) (regfunc),   /*注册方法上下文*/         \
    NODE_STRINGIFY(modname),      /*模块名*/                                  \
    priv,                         /*私有*/                                    \
    nullptr                       /*指针*/                                    \
  };                                                                          \
// _register_函数定义 跳到真正的注册方法
  void _register_ ## modname() {                                              \
    node_module_register(&_module);                                           \
  }

// 这个宏的调用地点在另一个C++文件里
#define NODE_BUILTIN_MODULE_CONTEXT_AWARE(modname, regfunc)                   \
  NODE_MODULE_CONTEXT_AWARE_CPP(modname, regfunc, nullptr, NM_F_BUILTIN)

　　看下面那个宏，其中第一个参数就是模块名，比如fs、os等等。第二个参数是指定模块的特殊方法，先暂不做深入研究。

　　宏在调用注册某一个方法前，会根据模块定义一个静态结构体，然后将对应指针传入真正的注册方法中去，结构体包含了模块的具体信息。

　　注册的方式非常简单，源码如下：

static node_module* modlist_builtin;

extern "C" void node_module_register(void* m) {
    // 定义一个新结构体指针
    struct node_module* mp = reinterpret_cast<struct node_module*>(m);
    // 判断类型并转换成链表
    if (mp->nm_flags & NM_F_BUILTIN) {
        mp->nm_link = modlist_builtin;
        modlist_builtin = mp;
    }
    // 其余类型模块的处理
}

　　在头部有一个默认的静态指针，然后每次注册定义了一个新的模块指针，用nm_link做链接，最后生成一个链表，图示如下：

　　这样，通过一个静态指针，即可访问到所有注册的内置模块。

 

　　注册完后，还是需要加载的，而这个加载地点仍然是上一节提到的一个方法：LoadEnviornment。

　　这个方法中包装了一个get_binging_fn方法，也就是上一节提到的C++注入参数的第二个，如下：

  // Create binding loaders
  v8::Local<v8::Function> get_binding_fn =
      env->NewFunctionTemplate(GetBinding)->GetFunction(env->context())
          .ToLocalChecked();

　　关键点就是那个GetBinding方法。这里需要通过JS代码来辅助讲解，首先假设调用了require('fs')，先走JS文件。

　　从上一节可以得知，由于加载的是内部模块，会走另一套逻辑，相关代码如下：

NativeModule.require = function(id) {
    if (id === loaderId) {
        return loaderExports;
    }
    // 取缓存
    const cached = NativeModule.getCached(id);
    if (cached && (cached.loaded || cached.loading)) {
        return cached.exports;
    }
    // 不合法的模块名
    if (!NativeModule.exists(id)) {
        // ...
    }

    moduleLoadList.push(`NativeModule ${id}`);
    // 这里进行模块加载
    const nativeModule = new NativeModule(id);
    // 编译并缓存
    nativeModule.cache();
    nativeModule.compile();

    return nativeModule.exports;
};

　　代码非常简单，可以看到在加载的时候会生成一个新的NativeModule对象，这个对象跟webpack的十分相似：

// Set up NativeModule
function NativeModule(id) {
    this.filename = `${id}.js`;
    this.id = id;
    this.exports = {};
    this.loaded = false;
    this.loading = false;
}

　　属性比较简单，这里就不做解释。主要问题放在那个编译方法上，相关代码如下：

const ContextifyScript = process.binding('contextify').ContextifyScript;

NativeModule._source = getBinding('natives');
NativeModule._cache = {};

NativeModule.wrap = function(script) {
    return NativeModule.wrapper[0] + script + NativeModule.wrapper[1];
};

NativeModule.wrapper = [
    '(function (exports, require, module, process) {',
    '\n});'
];
NativeModule.prototype.compile = function() {
    // return NativeModule._source[id]
    let source = NativeModule.getSource(this.id);
    // 代码包装
    source = NativeModule.wrap(source);

    this.loading = true;
    try {
        // 执行JS代码
        const script = new ContextifyScript(source, this.filename);
        // Arguments: timeout, displayErrors, breakOnSigint
        const fn = script.runInThisContext(-1, true, false);
        const requireFn = this.id.startsWith('internal/deps/') ?
            NativeModule.requireForDeps :
            NativeModule.require;
        fn(this.exports, requireFn, this, process);

        this.loaded = true;
    } finally {
        this.loading = false;
    }
};

　　可以看出，比较关键的几步都调用了process.binding或者getBinding方法，这两个方法正是来源于C++的代码注入。

　　同时这里还解释了为什么代码中的exports、require、module、process四个变量都是默认可用的，因为代码会被node自动进行包装，然后同样通过C++代码注入对应的函数参数。

　　因此，JS层面的代码都只是普通的方法分发逻辑，真正的调用都来源于底层的C++。

 

　　现在回到C++，直接看关键方法getBinding，只取关键代码：

static void GetBinding(const FunctionCallbackInfo<Value>& args) {
    // ...
    // 从链表获取对应模块信息
    node_module* mod = get_builtin_module(*module_v);
    // 新建输出对象
    Local<Object> exports;
    if (mod != nullptr) {
        // 生成指定模块
        exports = InitModule(env, mod, module);
    }
    // ...其他情况

    args.GetReturnValue().Set(exports);
}

　　在这里，获取对应模块信息就需要用到刚刚生成的注册信息链表，代码很简单，如下：

// name即模块名
node_module* get_builtin_module(const char* name) {
    return FindModule(modlist_builtin, name, NM_F_BUILTIN);
}

inline struct node_module* FindModule(struct node_module* list,
    const char* name,
    int flag) {
    struct node_module* mp;
    // 遍历链表
    for (mp = list; mp != nullptr; mp = mp->nm_link) {
        // strcmp比较两个字符串
        if (strcmp(mp->nm_modname, name) == 0)
            break;
    }
    // 检测一下 没找到mp就是空指针
    CHECK(mp == nullptr || (mp->nm_flags & flag) != 0);
    return mp;
}

　　这样，就得到了内置模块的信息，下一步就是模块加载。

　　之前在讲解模块结构体时提到过，除了模块名，还有一个指定模块的注册函数被一并添加进去了，这个地方就会用到对应的方法，如下：

static Local<Object> InitModule(Environment* env,
    node_module* mod,
    Local<String> module) {
    // 模块输出对象
    Local<Object> exports = Object::New(env->isolate());
    // 检测是否有对应的注册函数
    CHECK_EQ(mod->nm_register_func, nullptr);
    CHECK_NE(mod->nm_context_register_func, nullptr);
    Local<Value> unused = Undefined(env->isolate());
    // 编译生成对应的内置模块
    mod->nm_context_register_func(exports,
        unused,
        env->context(),
        mod->nm_priv);
    return exports;
}

　　就这样，在C++内部成功的加载了内置模块并返回，最后传到了JS代码层。

　　虽然对于模块注册函数来源、模块生成过程、JS2C的过程、C2JS的过程等等具体细节没有进行说明，但是对于内置模块的引入总体已经有了一个大概的印象，剩下的可以一步一步慢慢剖析。