Elliott Hughes | 3810445 | 2015-04-17 13:57:15 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2015 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 | #ifndef _ADB_UTILS_H_ |
| 18 | #define _ADB_UTILS_H_ |
| 19 | |
Josh Gao | 4b64084 | 2017-05-31 11:54:56 -0700 | [diff] [blame] | 20 | #include <condition_variable> |
| 21 | #include <mutex> |
Elliott Hughes | 3810445 | 2015-04-17 13:57:15 -0700 | [diff] [blame] | 22 | #include <string> |
Josh Gao | 4b64084 | 2017-05-31 11:54:56 -0700 | [diff] [blame] | 23 | #include <vector> |
Elliott Hughes | 3810445 | 2015-04-17 13:57:15 -0700 | [diff] [blame] | 24 | |
Josh Gao | 1b533c8 | 2016-03-04 15:15:56 -0800 | [diff] [blame] | 25 | #include <android-base/macros.h> |
| 26 | |
Elliott Hughes | 65bc227 | 2017-04-18 14:34:16 -0700 | [diff] [blame] | 27 | int syntax_error(const char*, ...); |
Elliott Hughes | 97e74bb | 2017-02-06 16:20:30 -0800 | [diff] [blame] | 28 | |
Josh Gao | 1eef478 | 2015-11-20 15:37:31 -0800 | [diff] [blame] | 29 | void close_stdin(); |
| 30 | |
Elliott Hughes | 7cf3575 | 2015-04-17 17:03:59 -0700 | [diff] [blame] | 31 | bool getcwd(std::string* cwd); |
Elliott Hughes | 3810445 | 2015-04-17 13:57:15 -0700 | [diff] [blame] | 32 | bool directory_exists(const std::string& path); |
Spencer Low | 939d000 | 2015-08-01 17:29:23 -0700 | [diff] [blame] | 33 | |
Yurii Zubrytskyi | 70ae70a | 2016-05-25 15:17:10 -0700 | [diff] [blame] | 34 | // Return the user's home directory. |
Josh Gao | 62ff9d4 | 2016-08-30 15:23:35 -0700 | [diff] [blame] | 35 | std::string adb_get_homedir_path(); |
| 36 | |
| 37 | // Return the adb user directory. |
| 38 | std::string adb_get_android_dir_path(); |
Yurii Zubrytskyi | 70ae70a | 2016-05-25 15:17:10 -0700 | [diff] [blame] | 39 | |
Alex Vallée | 28d1f8d | 2015-05-06 16:26:00 -0400 | [diff] [blame] | 40 | bool mkdirs(const std::string& path); |
Alex Vallée | e916315 | 2015-05-06 17:22:25 -0400 | [diff] [blame] | 41 | |
Elliott Hughes | 3810445 | 2015-04-17 13:57:15 -0700 | [diff] [blame] | 42 | std::string escape_arg(const std::string& s); |
| 43 | |
Yabin Cui | 19bec5b | 2015-09-22 15:52:57 -0700 | [diff] [blame] | 44 | std::string dump_hex(const void* ptr, size_t byte_count); |
Elliott Hughes | 88b4c85 | 2015-04-30 17:32:03 -0700 | [diff] [blame] | 45 | |
Elliott Hughes | b628cb1 | 2015-08-03 10:38:08 -0700 | [diff] [blame] | 46 | std::string perror_str(const char* msg); |
| 47 | |
Yabin Cui | 5fc2231 | 2015-10-06 15:10:05 -0700 | [diff] [blame] | 48 | bool set_file_block_mode(int fd, bool block); |
| 49 | |
Josh Gao | 1b533c8 | 2016-03-04 15:15:56 -0800 | [diff] [blame] | 50 | extern int adb_close(int fd); |
| 51 | |
David Pursell | 19d0c23 | 2016-04-07 11:25:48 -0700 | [diff] [blame] | 52 | // Given forward/reverse targets, returns true if they look sane. If an error is found, fills |
| 53 | // |error| and returns false. |
| 54 | // Currently this only checks "tcp:" targets. Additional checking could be added for other targets |
| 55 | // if needed. |
| 56 | bool forward_targets_are_valid(const std::string& source, const std::string& dest, |
| 57 | std::string* error); |
| 58 | |
Josh Gao | 4b64084 | 2017-05-31 11:54:56 -0700 | [diff] [blame] | 59 | // A thread-safe blocking queue. |
| 60 | template <typename T> |
| 61 | class BlockingQueue { |
| 62 | std::mutex mutex; |
| 63 | std::condition_variable cv; |
| 64 | std::vector<T> queue; |
| 65 | |
| 66 | public: |
| 67 | void Push(const T& t) { |
| 68 | { |
| 69 | std::unique_lock<std::mutex> lock(mutex); |
| 70 | queue.push_back(t); |
| 71 | } |
| 72 | cv.notify_one(); |
| 73 | } |
| 74 | |
| 75 | template <typename Fn> |
| 76 | void PopAll(Fn fn) { |
Josh Gao | 772d835 | 2017-06-05 14:54:45 -0700 | [diff] [blame] | 77 | std::vector<T> popped; |
Josh Gao | 4b64084 | 2017-05-31 11:54:56 -0700 | [diff] [blame] | 78 | |
Josh Gao | 772d835 | 2017-06-05 14:54:45 -0700 | [diff] [blame] | 79 | { |
| 80 | std::unique_lock<std::mutex> lock(mutex); |
| 81 | cv.wait(lock, [this]() { return !queue.empty(); }); |
| 82 | popped = std::move(queue); |
| 83 | queue.clear(); |
| 84 | } |
| 85 | |
| 86 | for (const T& t : popped) { |
Josh Gao | 4b64084 | 2017-05-31 11:54:56 -0700 | [diff] [blame] | 87 | fn(t); |
| 88 | } |
Josh Gao | 4b64084 | 2017-05-31 11:54:56 -0700 | [diff] [blame] | 89 | } |
| 90 | }; |
| 91 | |
Elliott Hughes | cd61ae3 | 2017-06-15 08:35:24 -0700 | [diff] [blame^] | 92 | std::string GetLogFilePath(); |
| 93 | |
Elliott Hughes | 3810445 | 2015-04-17 13:57:15 -0700 | [diff] [blame] | 94 | #endif |