1 /*
2  * Copyright 2018 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include <cstdint>
20 #include <string>
21 #include <vector>
22 
23 #include <errno.h>
24 #include <fcntl.h>
25 #include <netinet/in.h>
26 #include <sys/socket.h>
27 #include <unistd.h>
28 
29 namespace test_vendor_lib {
30 namespace net {
31 
32 #define WHILE_EINTR(fn) \
33   do {                  \
34   } while ((fn) == -1 && errno == EINTR)
35 
36 class PolledSocket {
37  public:
38   PolledSocket(int file_descriptor);
39   PolledSocket(PolledSocket&& p);
40   virtual ~PolledSocket();
41 
42   size_t TrySend(const std::vector<uint8_t>& packet);
43   // size_t TrySendVector(const std::vector<const std::vector<uint8_t>&>& data);
44   size_t TryReceive(size_t num_bytes, uint8_t* data);
45 
46  private:
47   void CleanUp();
48   int file_descriptor_{-1};
49 };
50 
51 }  // namespace net
52 }  // namespace test_vendor_lib
53