81 lines
3 KiB
C++
81 lines
3 KiB
C++
//
|
|
// Copyright (C) 2013 The Android Open Source Project
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
#ifndef SHILL_SOCKET_INFO_READER_H_
|
|
#define SHILL_SOCKET_INFO_READER_H_
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <base/files/file_path.h>
|
|
#include <base/macros.h>
|
|
#include <gtest/gtest_prod.h>
|
|
|
|
#include "shill/socket_info.h"
|
|
|
|
namespace shill {
|
|
|
|
class SocketInfoReader {
|
|
public:
|
|
SocketInfoReader();
|
|
virtual ~SocketInfoReader();
|
|
|
|
// Returns the file path (/proc/net/tcp by default) from where TCP/IPv4
|
|
// socket information are read. Overloadded by unit tests to return a
|
|
// different file path.
|
|
virtual base::FilePath GetTcpv4SocketInfoFilePath() const;
|
|
|
|
// Returns the file path (/proc/net/tcp6 by default) from where TCP/IPv6
|
|
// socket information are read. Overloadded by unit tests to return a
|
|
// different file path.
|
|
virtual base::FilePath GetTcpv6SocketInfoFilePath() const;
|
|
|
|
// Loads TCP socket information from /proc/net/tcp and /proc/net/tcp6.
|
|
// Existing entries in |info_list| are always discarded. Returns false
|
|
// if when neither /proc/net/tcp nor /proc/net/tcp6 can be read.
|
|
virtual bool LoadTcpSocketInfo(std::vector<SocketInfo>* info_list);
|
|
|
|
private:
|
|
FRIEND_TEST(SocketInfoReaderTest, AppendSocketInfo);
|
|
FRIEND_TEST(SocketInfoReaderTest, ParseConnectionState);
|
|
FRIEND_TEST(SocketInfoReaderTest, ParseIPAddress);
|
|
FRIEND_TEST(SocketInfoReaderTest, ParseIPAddressAndPort);
|
|
FRIEND_TEST(SocketInfoReaderTest, ParsePort);
|
|
FRIEND_TEST(SocketInfoReaderTest, ParseSocketInfo);
|
|
FRIEND_TEST(SocketInfoReaderTest, ParseTimerState);
|
|
FRIEND_TEST(SocketInfoReaderTest, ParseTransimitAndReceiveQueueValues);
|
|
|
|
bool AppendSocketInfo(const base::FilePath& info_file_path,
|
|
std::vector<SocketInfo>* info_list);
|
|
bool ParseSocketInfo(const std::string& input, SocketInfo* socket_info);
|
|
bool ParseIPAddressAndPort(
|
|
const std::string& input, IPAddress* ip_address, uint16_t* port);
|
|
bool ParseIPAddress(const std::string& input, IPAddress* ip_address);
|
|
bool ParsePort(const std::string& input, uint16_t* port);
|
|
bool ParseTransimitAndReceiveQueueValues(
|
|
const std::string& input,
|
|
uint64_t* transmit_queue_value, uint64_t* receive_queue_value);
|
|
bool ParseConnectionState(const std::string& input,
|
|
SocketInfo::ConnectionState* connection_state);
|
|
bool ParseTimerState(const std::string& input,
|
|
SocketInfo::TimerState* timer_state);
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(SocketInfoReader);
|
|
};
|
|
|
|
} // namespace shill
|
|
|
|
#endif // SHILL_SOCKET_INFO_READER_H_
|