Какой командой узнать количество открытых tcp соединений?

Question

[markinaras]

Какой командой или как то иначе узнать количество открытых tcp соединений в данный момент в системе? Хочу в своей программе (на Golang) получать количество открытых tcp соединений, чтобы контролировать и не открывать сверх нормы (иначе провайдер блокирует). Можно парсить "netstat -s" это слишком медленно , но подумал может есть проще способ или переменная где-то лежит и можно обратится к ней напрямую?

Comments

[mayton2019]

А можешь объяснить что именно и кто блокирует. Потому что для интернет-провайдера количество открытых
сокетов - безразлично. А вот для того сервера (хоста) куда ты пошел - может быть лимиты есть. Но может быть парсинг netstat это не то что тебе надо.

[markinaras]

mayton2019, я писал тест, в котором открывались новые соеденения, и если их больше определенного числа (например 1000), то их просто обрывают и прилетают ошибки. При чем это число(например 1000) одинакого, как для одного хоста, так и если обращатся к 10 хостам.

[mayton2019]

markinaras, а какой интервал между открытиями следующего сокета? Просто со стороны это может выглядить как syn-flood.

[markinaras]

mayton2019, Практически нет. Такая задача у софта, выполнить много сетевых запросов за короткий срок. Я понимаю, что могут специально ограничивать мое соеденение. И конечно буду оптимизировать и использовать повторно уже установленных tcp соеденений. Хотел бы сразу защиту написать от перегрузки, чтобы уж быть уверенным.

[mayton2019]

markinaras, ты пишешь на Go? Вот и напиши себе (или найди готовый) пул сокетов в котором количество
заранее ограничено.

Подозреваю что речь идет о парсинге сайтов. Так в этой предметной области - огромное количество уже
готовых решений. Тогда сразу не сокетов а HttpConnections. Вот. И запускай хоть мильон своих Go-потоков
а они пускай уже дерутся за эти 1000 коннектов по очереди.

[markinaras]

mayton2019, Это хорошая мысль, наверное так и сделаю в дополнение к оптимизации. Но контролировать все равно придется, вдруг у пользователя работает браузер или торрент, которые нагружают доролнительно. Спасибо за советы

Answer 1

[markinaras]

В гитхабе нашел код, с помощью которого можно получить сетевые данные. Число открытых tcp соединений получает за 1-2 миллисекунды. https://gist.github.com/adriansr/d2c1201be6bf2c203...

Немного изменил под себя.

package temp

import (
	"fmt"
	"syscall"
	"unsafe"
	"golang.org/x/sys/windows"
)

const (
	iphlpapiDll         = "iphlpapi.dll"
	tcpFn               = "GetExtendedTcpTable"
	tcpTableOwnerPidAll = 1
)

func GetTcpCounter() int {
	var tcp_counter int

	moduleHandle, err := windows.LoadLibrary(iphlpapiDll)
	if err != nil {
		panic(err)
	}

	ptr, err := windows.GetProcAddress(moduleHandle, tcpFn)
	if err != nil {
		panic(err)
	}

	res, err := getNetTable(ptr, false, windows.AF_INET, tcpTableOwnerPidAll)
	if err == nil {
		if res != nil && len(res) >= 4 {
			tcp_counter := int(*(*uint32)(unsafe.Pointer(&res[0])))
			fmt.Println("tcp counter:", tcp_counter)

		} else {
			fmt.Printf("nil result!\n")
		}
	} else {
		fmt.Printf("failed err = %v\n", err)
	}
	return tcp_counter
}

func getNetTable(fn uintptr, order bool, family int, class int) ([]byte, error) {
	var sorted uintptr
	if order {
		sorted = 1
	}
	for size, ptr, addr := uint32(8), []byte(nil), uintptr(0); ; {
		err, _, _ := syscall.Syscall6(fn, 5, addr, uintptr(unsafe.Pointer(&size)), sorted, uintptr(family), uintptr(class), 0)
		if err == 0 {
			return ptr, nil
		} else if err == uintptr(syscall.ERROR_INSUFFICIENT_BUFFER) {
			ptr = make([]byte, size)
			addr = uintptr(unsafe.Pointer(&ptr[0]))
		} else {
			return nil, fmt.Errorf("getNetTable failed: %v", err)
		}
	}
}