File: util.cpp

package info (click to toggle)
git-crypt 0.7.0-0.1
  • links: PTS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 416 kB
  • sloc: cpp: 3,404; xml: 376; makefile: 63
file content (157 lines) | stat: -rw-r--r-- 4,077 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
/*
 * Copyright 2012, 2014 Andrew Ayer
 *
 * This file is part of git-crypt.
 *
 * git-crypt is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * git-crypt is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with git-crypt.  If not, see <http://www.gnu.org/licenses/>.
 *
 * Additional permission under GNU GPL version 3 section 7:
 *
 * If you modify the Program, or any covered work, by linking or
 * combining it with the OpenSSL project's OpenSSL library (or a
 * modified version of that library), containing parts covered by the
 * terms of the OpenSSL or SSLeay licenses, the licensors of the Program
 * grant you additional permission to convey the resulting work.
 * Corresponding Source for a non-source form of such a combination
 * shall include the source code for the parts of OpenSSL used as well
 * as that of the covered work.
 */

#include "git-crypt.hpp"
#include "util.hpp"
#include "coprocess.hpp"
#include <string>
#include <iostream>

int exec_command (const std::vector<std::string>& args)
{
	Coprocess	proc;
	proc.spawn(args);
	return proc.wait();
}

int exec_command (const std::vector<std::string>& args, std::ostream& output)
{
	Coprocess	proc;
	std::istream*	proc_stdout = proc.stdout_pipe();
	proc.spawn(args);
	output << proc_stdout->rdbuf();
	return proc.wait();
}

int exec_command_with_input (const std::vector<std::string>& args, const char* p, size_t len)
{
	Coprocess	proc;
	std::ostream*	proc_stdin = proc.stdin_pipe();
	proc.spawn(args);
	proc_stdin->write(p, len);
	proc.close_stdin();
	return proc.wait();
}

std::string	escape_shell_arg (const std::string& str)
{
	std::string	new_str;
	new_str.push_back('"');
	for (std::string::const_iterator it(str.begin()); it != str.end(); ++it) {
		if (*it == '"' || *it == '\\' || *it == '$' || *it == '`') {
			new_str.push_back('\\');
		}
		new_str.push_back(*it);
	}
	new_str.push_back('"');
	return new_str;
}

uint32_t	load_be32 (const unsigned char* p)
{
	return (static_cast<uint32_t>(p[3]) << 0) |
	       (static_cast<uint32_t>(p[2]) << 8) |
	       (static_cast<uint32_t>(p[1]) << 16) |
	       (static_cast<uint32_t>(p[0]) << 24);
}

void		store_be32 (unsigned char* p, uint32_t i)
{
	p[3] = i; i >>= 8;
	p[2] = i; i >>= 8;
	p[1] = i; i >>= 8;
	p[0] = i;
}

bool		read_be32 (std::istream& in, uint32_t& i)
{
	unsigned char buffer[4];
	in.read(reinterpret_cast<char*>(buffer), 4);
	if (in.gcount() != 4) {
		return false;
	}
	i = load_be32(buffer);
	return true;
}

void		write_be32 (std::ostream& out, uint32_t i)
{
	unsigned char buffer[4];
	store_be32(buffer, i);
	out.write(reinterpret_cast<const char*>(buffer), 4);
}

void*		explicit_memset (void* s, int c, std::size_t n)
{
	volatile unsigned char* p = reinterpret_cast<unsigned char*>(s);

	while (n--) {
		*p++ = c;
	}

	return s;
}

static bool	leakless_equals_char (const unsigned char* a, const unsigned char* b, std::size_t len)
{
	volatile int	diff = 0;

	while (len > 0) {
		diff |= *a++ ^ *b++;
		--len;
	}

	return diff == 0;
}

bool 		leakless_equals (const void* a, const void* b, std::size_t len)
{
	return leakless_equals_char(reinterpret_cast<const unsigned char*>(a), reinterpret_cast<const unsigned char*>(b), len);
}

static void	init_std_streams_platform (); // platform-specific initialization

void		init_std_streams ()
{
	// The following two lines are essential for achieving good performance:
	std::ios_base::sync_with_stdio(false);
	std::cin.tie(0);

	std::cin.exceptions(std::ios_base::badbit);
	std::cout.exceptions(std::ios_base::badbit);

	init_std_streams_platform();
}

#ifdef _WIN32
#include "util-win32.cpp"
#else
#include "util-unix.cpp"
#endif