1*61c4878aSAndroid Build Coastguard Worker // Copyright 2021 The Pigweed Authors
2*61c4878aSAndroid Build Coastguard Worker //
3*61c4878aSAndroid Build Coastguard Worker // Licensed under the Apache License, Version 2.0 (the "License"); you may not
4*61c4878aSAndroid Build Coastguard Worker // use this file except in compliance with the License. You may obtain a copy of
5*61c4878aSAndroid Build Coastguard Worker // the License at
6*61c4878aSAndroid Build Coastguard Worker //
7*61c4878aSAndroid Build Coastguard Worker // https://www.apache.org/licenses/LICENSE-2.0
8*61c4878aSAndroid Build Coastguard Worker //
9*61c4878aSAndroid Build Coastguard Worker // Unless required by applicable law or agreed to in writing, software
10*61c4878aSAndroid Build Coastguard Worker // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
11*61c4878aSAndroid Build Coastguard Worker // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12*61c4878aSAndroid Build Coastguard Worker // License for the specific language governing permissions and limitations under
13*61c4878aSAndroid Build Coastguard Worker // the License.
14*61c4878aSAndroid Build Coastguard Worker
15*61c4878aSAndroid Build Coastguard Worker #include "pw_rpc/integration_testing.h"
16*61c4878aSAndroid Build Coastguard Worker
17*61c4878aSAndroid Build Coastguard Worker #include <limits>
18*61c4878aSAndroid Build Coastguard Worker
19*61c4878aSAndroid Build Coastguard Worker #include "pw_log/log.h"
20*61c4878aSAndroid Build Coastguard Worker #include "pw_rpc/integration_test_socket_client.h"
21*61c4878aSAndroid Build Coastguard Worker #include "pw_unit_test/framework.h"
22*61c4878aSAndroid Build Coastguard Worker #include "pw_unit_test/logging_event_handler.h"
23*61c4878aSAndroid Build Coastguard Worker
24*61c4878aSAndroid Build Coastguard Worker namespace pw::rpc::integration_test {
25*61c4878aSAndroid Build Coastguard Worker namespace {
26*61c4878aSAndroid Build Coastguard Worker
27*61c4878aSAndroid Build Coastguard Worker // Hard-coded to 1055 bytes, which is enough to fit 512-byte payloads when using
28*61c4878aSAndroid Build Coastguard Worker // HDLC framing.
29*61c4878aSAndroid Build Coastguard Worker SocketClientContext<1055> context;
30*61c4878aSAndroid Build Coastguard Worker unit_test::LoggingEventHandler log_test_events;
31*61c4878aSAndroid Build Coastguard Worker
32*61c4878aSAndroid Build Coastguard Worker } // namespace
33*61c4878aSAndroid Build Coastguard Worker
client()34*61c4878aSAndroid Build Coastguard Worker Client& client() { return context.client(); }
35*61c4878aSAndroid Build Coastguard Worker
SetClientSockOpt(int level,int optname,const void * optval,unsigned int optlen)36*61c4878aSAndroid Build Coastguard Worker int SetClientSockOpt(int level,
37*61c4878aSAndroid Build Coastguard Worker int optname,
38*61c4878aSAndroid Build Coastguard Worker const void* optval,
39*61c4878aSAndroid Build Coastguard Worker unsigned int optlen) {
40*61c4878aSAndroid Build Coastguard Worker return context.SetSockOpt(level, optname, optval, optlen);
41*61c4878aSAndroid Build Coastguard Worker }
42*61c4878aSAndroid Build Coastguard Worker
SetEgressChannelManipulator(ChannelManipulator * new_channel_manipulator)43*61c4878aSAndroid Build Coastguard Worker void SetEgressChannelManipulator(ChannelManipulator* new_channel_manipulator) {
44*61c4878aSAndroid Build Coastguard Worker context.SetEgressChannelManipulator(new_channel_manipulator);
45*61c4878aSAndroid Build Coastguard Worker }
46*61c4878aSAndroid Build Coastguard Worker
SetIngressChannelManipulator(ChannelManipulator * new_channel_manipulator)47*61c4878aSAndroid Build Coastguard Worker void SetIngressChannelManipulator(ChannelManipulator* new_channel_manipulator) {
48*61c4878aSAndroid Build Coastguard Worker context.SetIngressChannelManipulator(new_channel_manipulator);
49*61c4878aSAndroid Build Coastguard Worker }
50*61c4878aSAndroid Build Coastguard Worker
InitializeClient(int port)51*61c4878aSAndroid Build Coastguard Worker Status InitializeClient(int port) {
52*61c4878aSAndroid Build Coastguard Worker unit_test::RegisterEventHandler(&log_test_events);
53*61c4878aSAndroid Build Coastguard Worker if (port <= 0 || port > std::numeric_limits<uint16_t>::max()) {
54*61c4878aSAndroid Build Coastguard Worker PW_LOG_CRITICAL("Port numbers must be between 1 and 65535; %d is invalid",
55*61c4878aSAndroid Build Coastguard Worker port);
56*61c4878aSAndroid Build Coastguard Worker return Status::InvalidArgument();
57*61c4878aSAndroid Build Coastguard Worker }
58*61c4878aSAndroid Build Coastguard Worker
59*61c4878aSAndroid Build Coastguard Worker PW_LOG_INFO("Connecting to pw_rpc client at localhost:%d", port);
60*61c4878aSAndroid Build Coastguard Worker return context.Start(port);
61*61c4878aSAndroid Build Coastguard Worker }
62*61c4878aSAndroid Build Coastguard Worker
TerminateClient()63*61c4878aSAndroid Build Coastguard Worker void TerminateClient() { context.Terminate(); }
64*61c4878aSAndroid Build Coastguard Worker
InitializeClient(int argc,char * argv[],const char * usage_args)65*61c4878aSAndroid Build Coastguard Worker Status InitializeClient(int argc, char* argv[], const char* usage_args) {
66*61c4878aSAndroid Build Coastguard Worker if (argc < 2) {
67*61c4878aSAndroid Build Coastguard Worker PW_LOG_INFO("Usage: %s %s", argv[0], usage_args);
68*61c4878aSAndroid Build Coastguard Worker return Status::InvalidArgument();
69*61c4878aSAndroid Build Coastguard Worker }
70*61c4878aSAndroid Build Coastguard Worker
71*61c4878aSAndroid Build Coastguard Worker const int port = std::atoi(argv[1]);
72*61c4878aSAndroid Build Coastguard Worker return InitializeClient(port);
73*61c4878aSAndroid Build Coastguard Worker }
74*61c4878aSAndroid Build Coastguard Worker
75*61c4878aSAndroid Build Coastguard Worker } // namespace pw::rpc::integration_test
76