剝開比原看程式碼03：比原是如何監聽p2p埠的

比原鏈Bytom發表於2018-07-30

作者：freewind

比原專案倉庫：

Github地址：https://github.com/Bytom/bytom

Gitee地址：https://gitee.com/BytomBlockchain/bytom

我們知道，在使用bytomd init --chain_id mainnet/testnet/solonet初始化比原的時候，它會根據給定的chain_id的不同，使用不同的埠(參看config/toml.go#L29)：

  1. mainnet（連線到主網）: 46657
  2. testnet（連線到測試網）: 46656
  3. solonet（本地單獨節點）: 46658

對於我來說，由於只需要對本地執行的一個比原節點進行分析，所以可以採用第3個chain_id，即solonet。這樣它啟動之後，不會與其它的節點主動連線，可以減少其它節點對於我們的干擾。

所以在啟動的時候，我的命令是這樣的：

cd cmd/bytomd
./bytomd init --chain_id solonet
./bytomd node

它就會監聽46658埠，等待其它節點的連線。

連上看看

如果這時我們使用telnet來連線其46658埠，成功連線上之後，可以看到它會發給我們一些亂碼，大概如下：

$ telnet localhost 46658
Trying 127.0.0.1...
Connected to localhost.
Escape character is '^]'.
ט�S��%�z?��_�端��݂���U[e

我們也許會好奇，它發給我們的到底是什麼？

但是這個問題留待下次回答，因為首先，比原節點必須能夠監聽這個埠，我們才能連上。所以這次我們的問題是：

比原在程式碼中是如何監聽這個埠的?

埠已經寫在config.toml中

在前面，當我們使用./bytomd init --chain_id solonet初始化比原以後，比原會在本地的資料目錄中生成一個config.toml的配置檔案，內容大約如下：

# This is a TOML config file.
# For more information, see https://github.com/toml-lang/toml
fast_sync = true
db_backend = "leveldb"
api_addr = "0.0.0.0:9888"
chain_id = "solonet"
[p2p]
laddr = "tcp://0.0.0.0:46658"
seeds = ""

其中[p2p]下面的laddr，就是該節點監聽的地址和埠。

對於laddr = "tcp://0.0.0.0:46658"，它是意思是：

  1. 使用的是tcp協議
  2. 監聽的ip是0.0.0.0，是指監聽本機所有ip地址。這樣該節點既允許本地訪問，也允許外部主機訪問。如果你只想讓它監聽某一個ip，手動修改該配置檔案即可
  3. 46658，就是我們在這個問題中關注的埠了，它與該節點與其它節點互動資料使用的埠

比原在監聽這個埠的時候，並不是如我最開始預期的直接呼叫net.Listen監聽它。實際的過程要比這個複雜，因為比原設計了一個叫Switch的物件，用來統一管理與外界相關的事件，包括監聽、連線、傳送訊息等。而Switch這個物件，又是在SyncManager中建立的。

啟動直到進入Switch

所以我們首先需要知道，比原在原始碼中是如何啟動，並且一步步走進了Switch的世界。

首先還是當我們bytomd node啟動比原時，對應的入口函式如下：

cmd/bytomd/main.go#L54

func main() {
    cmd := cli.PrepareBaseCmd(commands.RootCmd, "TM", os.ExpandEnv(config.DefaultDataDir()))
    cmd.Execute()
}

它又會根據傳入的node引數，執行下面的函式：

cmd/bytomd/commands/run_node.go#L41

func runNode(cmd *cobra.Command, args []string) error {
    // Create & start node
    n := node.NewNode(config)
    // ...
}

我們需要關注的是node.NewNode(config)函式，因為是在它裡面建立了SyncManager：

node/node.go#L59

func NewNode(config *cfg.Config) *Node {
    // ...
    syncManager, _ := netsync.NewSyncManager(config, chain, txPool, newBlockCh)
    // ...
}

在建立SyncManager的時候，又建立了Switch:

netsync/handle.go#L42

func NewSyncManager(config *cfg.Config, chain *core.Chain, txPool *core.TxPool, newBlockCh chan *bc.Hash) (*SyncManager, error) {
    // ...
    manager.sw = p2p.NewSwitch(config.P2P, trustHistoryDB)

    // ...
    protocolReactor := NewProtocolReactor(chain, txPool, manager.sw, manager.blockKeeper, manager.fetcher, manager.peers, manager.newPeerCh, manager.txSyncCh, manager.dropPeerCh)
    manager.sw.AddReactor("PROTOCOL", protocolReactor)

    // Create & add listener
    p, address := protocolAndAddress(manager.config.P2P.ListenAddress)
    l := p2p.NewDefaultListener(p, address, manager.config.P2P.SkipUPNP, nil)
    manager.sw.AddListener(l)

    // ...
}

這裡需要注意一下，上面建立的protocolReactor物件，是用來處理當有節點連線上埠後，雙方如何互動的事情。跟這次問題“監聽埠”沒有直接關係，但是這裡也可以注意一下。

然後又建立了一個DefaultListener物件，而監聽埠的動作，就是在它裡面發生的。Listener建立之後，將會新增到manager.sw（即Switch）中，用於在那邊進行外界資料與事件的互動。

監聽埠

NewDefaultListener中做的事情比較多，所以我們把它分成幾塊說：

p2p/listener.go#L52

func NewDefaultListener(protocol string, lAddr string, skipUPNP bool, logger tlog.Logger) Listener {
    // Local listen IP & port
    lAddrIP, lAddrPort := splitHostPort(lAddr)

    // Create listener
    var listener net.Listener
    var err error
    for i := 0; i < tryListenSeconds; i++ {
        listener, err = net.Listen(protocol, lAddr)
        if err == nil {
            break
        } else if i < tryListenSeconds-1 {
            time.Sleep(time.Second * 1)
        }
    }
    if err != nil {
        cmn.PanicCrisis(err)
    }

    // ...

上面這部分就是真正監聽的程式碼了。通過Go語言提供的net.Listen函式，監聽了指定的地址。另外，在監聽的時候，進行了多次嘗試，因為當一個剛剛被使用的埠被放開後，還需要一小段時間才能真正釋放，所以這裡需要多嘗試幾次。

其中tryListenSeconds是一個常量，值為5，也就是說，大約會嘗試5秒鐘，要是都繫結不上，才會真正失敗，丟擲錯誤。

後面省略了一些程式碼，主要是用來獲取當前監聽的實際ip以及外網ip，並記錄在日誌中。本想在這裡簡單講講，但是發現還有點麻煩，所以打算放在後面專開一個問題。

其實本次問題到這裡就已經結束了，因為已經完成了“監聽”。但是後面還有一些初始化操作，是為了讓比原可以跟連線上該埠的節點進行互動，也值得在這裡講講。

接著剛才的方法，最後的部分是：

    dl := &DefaultListener{
        listener:    listener,
        intAddr:     intAddr,
        extAddr:     extAddr,
        connections: make(chan net.Conn, numBufferedConnections),
    }
    dl.BaseService = *cmn.NewBaseService(logger, "DefaultListener", dl)
    dl.Start() // Started upon construction
    return dl
}

需要注意的是connections，它是一個帶有緩衝的channel（numBufferedConnections值為10），用來存放連線上該埠的連線物件。這些操作將在後面的dl.Start()中執行。

dl.Start()將呼叫DefaultListener對應的OnStart方法，如下：

p2p/listener.go#L114

func (l *DefaultListener) OnStart() error {
    l.BaseService.OnStart()
    go l.listenRoutine()
    return nil
}

其中的l.listenRoutine，就是執行前面所說的向connections channel裡放入連線的函式:

p2p/listener.go#L126

func (l *DefaultListener) listenRoutine() {
    for {
        conn, err := l.listener.Accept()
        // ...
        l.connections <- conn
    }

    // Cleanup
    close(l.connections)

    // ...
}

而Switch在SyncManager啟動的時候會被啟動，在它的OnStart方法中，會拿到所有Listener（即監聽埠的物件）中connectionschannel中的連線，與它們互動。

https://github.com/freewind/bytom-v1.0.1/blob/master/p2p/switch.go#L498

func (sw *Switch) listenerRoutine(l Listener) {
    for {
        inConn, ok := <-l.Connections()
        if !ok {
            break
        }
        // ...

        err := sw.addPeerWithConnectionAndConfig(inConn, sw.peerConfig)
        // ...
    }

其中sw.addPeerWithConnectionAndConfig就是與對應節點進行互動的邏輯所在，但是這已經超出了本次問題的範疇，下次再講。

到此為止，本次的問題，應該已經講清楚了。

相關文章