mirror of
https://github.com/nestriness/cdc-file-transfer.git
synced 2026-01-30 16:15:36 +02:00
5fd86e4625
Fixes an issue with port forwarding when localhost on the remote system maps to the IPV6 localhost. In that case the server would time out on accept() since it creates an IPV4 socket, so the connection is never established. Also allows passing in port 0, so that it will auto-detect an available port. This will be used in a future CL to remove the necessity of running netstat/ss.
69 lines
2.2 KiB
C++
69 lines
2.2 KiB
C++
/*
|
|
* Copyright 2022 Google LLC
|
|
*
|
|
* 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 CDC_RSYNC_SERVER_SERVER_SOCKET_H_
|
|
#define CDC_RSYNC_SERVER_SERVER_SOCKET_H_
|
|
|
|
#include "absl/status/status.h"
|
|
#include "absl/status/statusor.h"
|
|
#include "cdc_rsync/base/socket.h"
|
|
|
|
struct addrinfo;
|
|
|
|
namespace cdc_ft {
|
|
|
|
class ServerSocket : public Socket {
|
|
public:
|
|
ServerSocket();
|
|
~ServerSocket();
|
|
|
|
// Starts listening for connections on |port|.
|
|
// Passing 0 as port will bind to any available port.
|
|
// Returns the port that was bound to.
|
|
absl::StatusOr<int> StartListening(int port);
|
|
|
|
// Stops listening for connections. No-op if already stopped/never started.
|
|
void StopListening();
|
|
|
|
// Waits for a client to connect. Only supports one connection. Repeating
|
|
// the call with an existing connection results in an error.
|
|
absl::Status WaitForConnection();
|
|
|
|
// Disconnects the client. No-op if not connected.
|
|
void Disconnect();
|
|
|
|
// Shuts down the sending end of the socket. This will interrupt any receive
|
|
// calls on the client and shut it down.
|
|
absl::Status ShutdownSendingEnd();
|
|
|
|
// Socket:
|
|
absl::Status Send(const void* buffer, size_t size) override;
|
|
absl::Status Receive(void* buffer, size_t size, bool allow_partial_read,
|
|
size_t* bytes_received) override;
|
|
|
|
private:
|
|
// Called by StartListening() for a specific IPV4 or IPV6 |addr_info|.
|
|
// Passing 0 as port will bind to any available port.
|
|
// Returns the port that was bound to.
|
|
absl::StatusOr<int> StartListeningInternal(int port, addrinfo* addr);
|
|
|
|
std::unique_ptr<struct ServerSocketInfo> socket_info_;
|
|
};
|
|
|
|
} // namespace cdc_ft
|
|
|
|
#endif // CDC_RSYNC_SERVER_SERVER_SOCKET_H_
|