@@ -2,11 +2,17 @@ package tui
22
33import (
44 "fmt"
5+ "os"
6+ "path/filepath"
57 "runtime/debug"
68 "strings"
9+ "time"
710
811 "github.com/ddddddO/packemon"
912 "github.com/gdamore/tcell/v2"
13+ "github.com/gopacket/gopacket"
14+ "github.com/gopacket/gopacket/layers"
15+ "github.com/gopacket/gopacket/pcapgo"
1016 "github.com/rivo/tview"
1117)
1218
@@ -30,22 +36,35 @@ func (t *tui) updateView(passive *packemon.Passive) {
3036 t .grid .Clear ()
3137 })
3238
33- rows := make ([]int , len (viewers ))
34- columns := make ([]int , len (viewers ))
39+ t .grid .RemoveItem (t .grid ) // ほんと？
40+
41+ // +1 分は、PCAP保存領域用(savingPCAPView)
42+ rows := make ([]int , len (viewers )+ 1 )
43+ columns := make ([]int , len (viewers )+ 1 )
44+ rows [0 ] = 5
45+ columns [0 ] = 30
3546 for i := range viewers {
3647 rows [i ] = viewers [i ].rows ()
3748 columns [i ] = viewers [i ].columns ()
3849 }
39- t .grid .RemoveItem (t .grid ) // ほんと？
40- t .grid .SetRows (rows ... ).SetColumns (columns ... ).SetBorders (false )
50+
51+ // SetRows しなくなったので、各テーブルの rows メソッドいらないかも
52+ // t.grid.SetRows(rows...).SetColumns(columns...).SetBorders(false)
53+ t .grid .SetColumns (columns ... ).SetBorders (false )
54+
4155 for i := range viewers {
4256 t .grid .AddItem (viewers [i ].viewTable (), i , 0 , 1 , 3 , 0 , 0 , false ) // focus=true にするとスクロールしない
4357 }
58+ savingPCAPView := t .savingPCAPView (passive )
59+ row := len (viewers )
60+ t .grid .AddItem (savingPCAPView , row , 0 , 1 , 3 , 0 , 0 , false )
61+
4462 t .grid .SetInputCapture (
4563 func (event * tcell.EventKey ) * tcell.EventKey {
4664 if event .Key () == tcell .KeyEscape {
4765 t .grid .Clear ()
4866 t .pages .SwitchToPage ("history" )
67+ t .app .SetFocus (t .pages )
4968 }
5069 return event
5170 })
@@ -130,6 +149,61 @@ func passiveToViewers(passive *packemon.Passive) []Viewer {
130149 return viewers
131150}
132151
152+ func (t * tui ) savingPCAPView (p * packemon.Passive ) * tview.Form {
153+ now := time .Now ()
154+ fpath := fmt .Sprintf ("./packemon_pcap/%s.pcap" , now .Format ("20060102150405" ))
155+ form := tview .NewForm ().
156+ AddInputField ("File Name" , fpath , 40 , func (textToCheck string , lastChar rune ) bool {
157+ if len (textToCheck ) < 40 {
158+ fpath = textToCheck
159+ return true
160+ } else if len (textToCheck ) > 40 {
161+ return false
162+ }
163+ fpath = textToCheck
164+ return true
165+ }, nil ).
166+ AddButton ("Save" , func () {
167+ if p .EthernetFrame == nil {
168+ t .addErrPageForMonitor (fmt .Errorf ("Empty ethernet frame..." ))
169+ return
170+ }
171+
172+ dir := filepath .Dir (fpath )
173+ if err := os .MkdirAll (dir , 0755 ); err != nil {
174+ t .addErrPageForMonitor (err )
175+ return
176+ }
177+
178+ f , err := os .Create (fpath )
179+ if err != nil {
180+ t .addErrPageForMonitor (err )
181+ return
182+ }
183+ defer f .Close ()
184+
185+ pcapw := pcapgo .NewWriter (f )
186+ if err := pcapw .WriteFileHeader (1500 , layers .LinkTypeEthernet ); err != nil {
187+ t .addErrPageForMonitor (err )
188+ return
189+ }
190+ ci := gopacket.CaptureInfo {
191+ Timestamp : now ,
192+ CaptureLength : 1500 ,
193+ Length : 1500 ,
194+ // InterfaceIndex: intf.Index, // 必須ではなさそう
195+ }
196+ if err := pcapw .WritePacket (ci , p .EthernetFrame .Bytes ()); err != nil {
197+ t .addErrPageForMonitor (err )
198+ return
199+ }
200+ })
201+ form .SetBorder (true )
202+ form .Box = tview .NewBox ().SetBorder (true ).SetTitle (" Save PACP " ).SetTitleAlign (tview .AlignLeft ).SetBorderPadding (1 , 1 , 1 , 1 )
203+
204+ return form
205+ }
206+
133207func tableCellTitle (title string ) * tview.TableCell {
134208 return tview .NewTableCell (padding (title ))
135209}
0 commit comments