さいしょに
ナンデモアリなAdventCalendar2018の23日目の記事です。
最初はこれ用になんか実装でもするかと思ってましたが修論のせいで時間がとれなかったのと直前で風邪を引いたので、適当に研究室で動かしてる在室管理botの話でお茶を濁します。
やりたいこと
研究室内に誰が居るかというのを自動で取得したい。
SlackなりLINEなりで誰が今居るか確認したいのでできればWeb APIでJSONとかで結果が返ってくるとうれしい。
そしていちいち入退室のたびに自分の在室状況を紙なりアプリなりで変更するのは面倒なので__更新作業とか絶対したくない__。
前提条件
- __研究室内でLAN__を用意していること
- 研究室ほとんどのメンバーが__そのLANを利用していること__
- 研究室内に一台以上__常時稼働できるコンピュータ__が転がっていること
どうやって在室状況取るか
arpとMACアドレスで在室状況とってngrokでグローバルに出す。
構成はこんな感じ
- 適当なPCの適当なHTTPのポートで入力待つ
- 入力受け取ったPC側がARPテーブル更新
- ARPテーブルのMACアドレスと手元に持ってる研究室メンバーのMACアドレスを突き合わせる
- 誰が居るか返す
という流れ。
研究室内ネットワークにPCなりスマホなりが繋がっていれば検出されることになる。あまり精確ではないため、厳密に把握したいという場合には向いてない。
技術的に面倒なことは特に無いのでグッとやると動く。
実装
まずarpリクエスト送って在室者一覧を返すarp.go
package recorder
import (
"fmt"
"os/exec"
"regexp"
"runtime"
"strconv"
"strings"
"sync"
"github.com/taigak/room_watch/util"
)
var os = strings.ToLower(runtime.GOOS)
type Member struct {
Name string
Devices []string
}
func NewMembers() []Member {
members := []Member{
{Name: "人1", Devices: []string{"00:00:00:00:00:00", "00:00:00:00:00:00"}},
{Name: "人2", Devices: []string{"00:00:00:00:00:00"}},
}
return members
}
// ARPテーブル内のMACアドレスリスト取得
func getARPTable() []string {
// send ping
var wg sync.WaitGroup
for i := 2; i < 254; i++ {
go func(i int) {
wg.Add(1)
switch os {
case "windows":
exec.Command("ping", "192.168.11."+strconv.Itoa(i)).Run()
case "linux":
exec.Command("ping", "-c", "2", "192.168.11."+strconv.Itoa(i)).Run()
}
wg.Done()
}(i)
}
wg.Wait()
// get ARP Table
out, err := exec.Command("arp", "-a").Output()
if err != nil {
fmt.Println(err)
panic("err")
}
r := regexp.MustCompile(`([0-9A-Fa-f]{2}[:-]){5}([0-9A-Fa-f]{2})`)
res := r.FindAllStringSubmatch(string(out), -1)
// formatting
table := []string{}
for _, mac := range res {
m := mac[0][0:17]
if os == "windows" {
m = strings.Replace(m, "-", ":", -1)
}
table = append(table, m)
}
return table
}
// 事前登録MACアドレスとARPテーブル内のMACアドレス突き合わせ
func GetAttendance() []string {
members := NewMembers()
table := getARPTable()
attendances := []string{}
for _, member := range members {
original := table
table = util.Difference(table, member.Devices)
if len(original) > len(table) {
attendances = append(attendances, member.Name)
}
}
return attendances
}
やってることは
- 取りうる範囲のIPアドレスにping送信
- ARPテーブル更新
- MACアドレス切り出し
- 既に登録してあるMACアドレスと突き合わせて在席者の配列を返す
というだけ。
OS側のarpコマンドを叩く形にしたけど環境によって微妙に返ってくる形式が違うのでformattingしてる。
次にarp.goを呼び出す側。
package recorder
import (
"fmt"
"time"
)
var (
record = Record{}
)
type Record struct {
Time string `json:"time"`
Members []string `json:"members"`
}
func Watch() {
go func() {
set() //最初に一回だけmembers記録
t := time.NewTicker(5 * time.Minute)
defer t.Stop()
for {
select {
case <-t.C:
set()
}
}
}()
}
func set() {
record = Record{
Time: time.Now().Format("2006/1/2-15:04:05"),
Members: GetAttendance(),
}
fmt.Println(time.Now().Format("2006/1/2-15:04:05"), "New Member log Recorded")
}
func Get() Record {
return record
}
実際のところHTTP Requestが来る度にARPテーブル更新してたらResponseが遅いし負荷もかかるので5分に一回更新して雑にcacheする形にしてる。
やってることは大体arp.go呼んで結果を整形してるだけ。
次にServer起動部分
package main
import (
"encoding/json"
"fmt"
"net/http"
"github.com/taigak/room_watch/recorder"
)
const (
Auth = "SET_YOUR_PASSWORD"
)
func main() {
// 5分毎の更新
recorder.Watch()
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
status := recorder.Get()
jsonBytes, err := json.Marshal(status)
if err != nil {
w.WriteHeader(500)
fmt.Fprint(w, err)
return
}
// 認証エラー
if pw := r.Header.Get("Authorization"); pw != Auth {
w.WriteHeader(401)
fmt.Fprint(w, "Unauthorized")
return
}
// その他エラー
if err != nil {
w.WriteHeader(500)
fmt.Fprint(w, err)
return
}
fmt.Fprint(w, string(jsonBytes))
})
http.ListenAndServe(":8080", nil)
}
やってることは5分たびに在席者を更新しつつ8080番でHTTP待ち受けて結果を返すだけ。
あとグローバルに公開したときにあんまり誰にでも見て欲しい情報ではないのでHTTPのAuthorization Headerに適当なパスワードを設定させて認証してる。
あとはgo run *.goなりで起動すれば8080番で待ち受けて在席者リストを返してくれる。
golangに関してはまだ初心者なのでコードにつっこみなり何なりもらえると嬉しいです。
公開
公開はngrokで雑にやる。
ngrok.ioと自前のマシンの間でトンネル掘っていい感じにポートフォワーディングして外部に公開してくれるいい感じのやつ。
要は自身のマシンが外部からのアクセス受け取るようになるということなので、この行為のリスクはちゃんと認識したうえで実行しましょう。
あとネットワークの管理者とも合意を取るべきとか色々ありますが、組織の偉い人に何を言われるか分からないので注意しましょうとだけ言っておきます。
実行は公式サイトに行ってバイナリ落としてきて
$ ./ngrok http 8080
とすればURLが払い出される。
これでAPIがどこからでも呼び出せるようになるので
curl -H "Authorization:YOUR_PASSWORD" https://xxxxxxxx.ngrok.io
とかやれば動いてると思います。
後はSlackと連携なりLINEと連携なり好きにやってください。
LINEとかだとこんな感じで在席者一覧が見れる（人名はぼかしてる）
その他注意
在席者管理とは言っても本質的には監視と変わらないので他のメンバーの同意無く動かすとかはやめましょう。
勝手に他人のMACアドレスを取って監視対象にするとかもやめときましょう。
以下参考にした資料
The Go Programming Language Package sync
Golangで外部コマンドを実行する方法まとめ
1984
Go言語 - Tickerを使った定期処理