blob: 6d12225dc12567611e026cecdd69171e4db106c8 [file] [log] [blame]
Elliott Hughes58305772015-04-17 13:57:15 -07001/*
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
Elliott Hughesd2aaede2018-10-22 17:02:51 -070017#pragma once
Elliott Hughes58305772015-04-17 13:57:15 -070018
Josh Gao0b13c892017-05-31 11:54:56 -070019#include <condition_variable>
20#include <mutex>
Elliott Hughes58305772015-04-17 13:57:15 -070021#include <string>
Josh Gao0b13c892017-05-31 11:54:56 -070022#include <vector>
Elliott Hughes58305772015-04-17 13:57:15 -070023
Josh Gaof0d3b4f2016-03-04 15:15:56 -080024#include <android-base/macros.h>
25
Josh Gao99d3d702018-10-11 13:49:24 -070026#include "adb.h"
27
Josh Gao7d586072015-11-20 15:37:31 -080028void close_stdin();
29
Elliott Hughesa7090b92015-04-17 17:03:59 -070030bool getcwd(std::string* cwd);
Elliott Hughes58305772015-04-17 13:57:15 -070031bool directory_exists(const std::string& path);
Spencer Low22191c32015-08-01 17:29:23 -070032
Yurii Zubrytskyi376b2752016-05-25 15:17:10 -070033// Return the user's home directory.
Josh Gaoe0b75022016-08-30 15:23:35 -070034std::string adb_get_homedir_path();
35
36// Return the adb user directory.
37std::string adb_get_android_dir_path();
Yurii Zubrytskyi376b2752016-05-25 15:17:10 -070038
Alex Vallée47d67c92015-05-06 16:26:00 -040039bool mkdirs(const std::string& path);
Alex Vallée14216142015-05-06 17:22:25 -040040
Elliott Hughes58305772015-04-17 13:57:15 -070041std::string escape_arg(const std::string& s);
42
Yabin Cuiaed3c612015-09-22 15:52:57 -070043std::string dump_hex(const void* ptr, size_t byte_count);
Josh Gao99d3d702018-10-11 13:49:24 -070044std::string dump_header(const amessage* msg);
45std::string dump_packet(const char* name, const char* func, const apacket* p);
Elliott Hughese67f1f82015-04-30 17:32:03 -070046
Elliott Hughesaa245492015-08-03 10:38:08 -070047std::string perror_str(const char* msg);
48
Elliott Hughesd2aaede2018-10-22 17:02:51 -070049[[noreturn]] void error_exit(const char* fmt, ...) __attribute__((__format__(__printf__, 1, 2)));
50[[noreturn]] void perror_exit(const char* fmt, ...) __attribute__((__format__(__printf__, 1, 2)));
51
Yabin Cui6dfef252015-10-06 15:10:05 -070052bool set_file_block_mode(int fd, bool block);
53
Elliott Hughesd2aaede2018-10-22 17:02:51 -070054int adb_close(int fd);
Josh Gaof0d3b4f2016-03-04 15:15:56 -080055
David Purselleaae97e2016-04-07 11:25:48 -070056// Given forward/reverse targets, returns true if they look sane. If an error is found, fills
57// |error| and returns false.
58// Currently this only checks "tcp:" targets. Additional checking could be added for other targets
59// if needed.
60bool forward_targets_are_valid(const std::string& source, const std::string& dest,
61 std::string* error);
62
Josh Gao0b13c892017-05-31 11:54:56 -070063// A thread-safe blocking queue.
64template <typename T>
65class BlockingQueue {
66 std::mutex mutex;
67 std::condition_variable cv;
68 std::vector<T> queue;
69
70 public:
71 void Push(const T& t) {
72 {
73 std::unique_lock<std::mutex> lock(mutex);
74 queue.push_back(t);
75 }
76 cv.notify_one();
77 }
78
79 template <typename Fn>
80 void PopAll(Fn fn) {
Josh Gao664a6182017-06-05 14:54:45 -070081 std::vector<T> popped;
Josh Gao0b13c892017-05-31 11:54:56 -070082
Josh Gao664a6182017-06-05 14:54:45 -070083 {
84 std::unique_lock<std::mutex> lock(mutex);
85 cv.wait(lock, [this]() { return !queue.empty(); });
86 popped = std::move(queue);
87 queue.clear();
88 }
89
90 for (const T& t : popped) {
Josh Gao0b13c892017-05-31 11:54:56 -070091 fn(t);
92 }
Josh Gao0b13c892017-05-31 11:54:56 -070093 }
94};
95
Elliott Hughes6eadee82017-06-15 08:35:24 -070096std::string GetLogFilePath();