You need to sign in or sign up before continuing.
util.h 9.9 KB
Newer Older
1 2 3
/*
 * utils.h: common, generic utility functions
 *
4
 * Copyright (C) 2010-2012 Red Hat, Inc.
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
 * Copyright (C) 2006, 2007 Binary Karma
 * Copyright (C) 2006 Shuveb Hussain
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
 *
 * File created Jul 18, 2007 - Shuveb Hussain <shuveb@binarykarma.com>
 */

25
#ifndef __VIR_UTIL_H__
26
# define __VIR_UTIL_H__
27

28 29 30 31 32
# include "verify.h"
# include "internal.h"
# include <unistd.h>
# include <sys/select.h>
# include <sys/types.h>
E
Eric Blake 已提交
33
# include <stdarg.h>
34

35 36 37
# ifndef MIN
#  define MIN(a, b) ((a) < (b) ? (a) : (b))
# endif
L
Laine Stump 已提交
38 39 40
# ifndef MAX
#  define MAX(a, b) ((a) > (b) ? (a) : (b))
# endif
41

E
Eric Blake 已提交
42
ssize_t saferead(int fd, void *buf, size_t count) ATTRIBUTE_RETURN_CHECK;
43 44
ssize_t safewrite(int fd, const void *buf, size_t count)
    ATTRIBUTE_RETURN_CHECK;
E
Eric Blake 已提交
45
int safezero(int fd, off_t offset, off_t len)
46
    ATTRIBUTE_RETURN_CHECK;
D
Daniel P. Berrange 已提交
47

48
int virSetBlocking(int fd, bool blocking) ATTRIBUTE_RETURN_CHECK;
49
int virSetNonBlock(int fd) ATTRIBUTE_RETURN_CHECK;
E
Eric Blake 已提交
50
int virSetInherit(int fd, bool inherit) ATTRIBUTE_RETURN_CHECK;
51
int virSetCloseExec(int fd) ATTRIBUTE_RETURN_CHECK;
52

53 54
int virPipeReadUntilEOF(int outfd, int errfd,
                        char **outbuf, char **errbuf);
55

L
Laine Stump 已提交
56 57
int virSetUIDGID(uid_t uid, gid_t gid);

58
int virFileReadLimFD(int fd, int maxlen, char **buf) ATTRIBUTE_RETURN_CHECK;
59

60
int virFileReadAll(const char *path, int maxlen, char **buf) ATTRIBUTE_RETURN_CHECK;
61

62 63
int virFileWriteStr(const char *path, const char *str, mode_t mode)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(2) ATTRIBUTE_RETURN_CHECK;
M
Mark McLoughlin 已提交
64

65 66 67 68 69 70 71
int virFileMatchesNameSuffix(const char *file,
                             const char *name,
                             const char *suffix);

int virFileHasSuffix(const char *str,
                     const char *suffix);

72
int virFileStripSuffix(char *str,
73
                       const char *suffix) ATTRIBUTE_RETURN_CHECK;
74

75 76
int virFileLinkPointsTo(const char *checkLink,
                        const char *checkDest);
77

D
Daniel P. Berrange 已提交
78
int virFileResolveLink(const char *linkpath,
79
                       char **resultpath) ATTRIBUTE_RETURN_CHECK;
80 81
int virFileResolveAllLinks(const char *linkpath,
                           char **resultpath) ATTRIBUTE_RETURN_CHECK;
D
Daniel P. Berrange 已提交
82

83 84 85
int virFileIsLink(const char *linkpath)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_RETURN_CHECK;

86 87
char *virFindFileInPath(const char *file);

88
bool virFileIsDir (const char *file) ATTRIBUTE_NONNULL(1);
89 90
bool virFileExists(const char *file) ATTRIBUTE_NONNULL(1);
bool virFileIsExecutable(const char *file) ATTRIBUTE_NONNULL(1);
91

92 93
char *virFileSanitizePath(const char *path);

94
enum {
95
    VIR_FILE_OPEN_NONE        = 0,
L
Laine Stump 已提交
96 97 98 99
    VIR_FILE_OPEN_NOFORK      = (1 << 0),
    VIR_FILE_OPEN_FORK        = (1 << 1),
    VIR_FILE_OPEN_FORCE_MODE  = (1 << 2),
    VIR_FILE_OPEN_FORCE_OWNER = (1 << 3),
100
};
101 102 103
int virFileAccessibleAs(const char *path, int mode,
                        uid_t uid, gid_t gid)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_RETURN_CHECK;
104 105 106
int virFileOpenAs(const char *path, int openflags, mode_t mode,
                  uid_t uid, gid_t gid,
                  unsigned int flags)
107
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_RETURN_CHECK;
108

109 110 111 112 113 114
enum {
    VIR_DIR_CREATE_NONE        = 0,
    VIR_DIR_CREATE_AS_UID      = (1 << 0),
    VIR_DIR_CREATE_FORCE_PERMS = (1 << 1),
    VIR_DIR_CREATE_ALLOW_EXIST = (1 << 2),
};
115 116
int virDirCreate(const char *path, mode_t mode, uid_t uid, gid_t gid,
                 unsigned int flags) ATTRIBUTE_RETURN_CHECK;
117
int virFileMakePath(const char *path) ATTRIBUTE_RETURN_CHECK;
118 119
int virFileMakePathWithMode(const char *path,
                            mode_t mode) ATTRIBUTE_RETURN_CHECK;
120

121 122 123
char *virFileBuildPath(const char *dir,
                       const char *name,
                       const char *ext) ATTRIBUTE_RETURN_CHECK;
124

125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147

# ifdef WIN32
/* On Win32, the canonical directory separator is the backslash, and
 * the search path separator is the semicolon. Note that also the
 * (forward) slash works as directory separator.
 */
#  define VIR_FILE_DIR_SEPARATOR '\\'
#  define VIR_FILE_DIR_SEPARATOR_S "\\"
#  define VIR_FILE_IS_DIR_SEPARATOR(c) ((c) == VIR_FILE_DIR_SEPARATOR || (c) == '/')
#  define VIR_FILE_PATH_SEPARATOR ';'
#  define VIR_FILE_PATH_SEPARATOR_S ";"

# else  /* !WIN32 */

#  define VIR_FILE_DIR_SEPARATOR '/'
#  define VIR_FILE_DIR_SEPARATOR_S "/"
#  define VIR_FILE_IS_DIR_SEPARATOR(c) ((c) == VIR_FILE_DIR_SEPARATOR)
#  define VIR_FILE_PATH_SEPARATOR ':'
#  define VIR_FILE_PATH_SEPARATOR_S ":"

# endif /* !WIN32 */

bool virFileIsAbsPath(const char *path);
A
Amy Griffis 已提交
148
int virFileAbsPath(const char *path,
149
                   char **abspath) ATTRIBUTE_RETURN_CHECK;
150
const char *virFileSkipRoot(const char *path);
A
Amy Griffis 已提交
151

152 153 154
int virFileOpenTty(int *ttymaster,
                   char **ttyName,
                   int rawmode);
155

156 157
char *virArgvToString(const char *const *argv);

D
Daniel P. Berrange 已提交
158
int virStrToLong_i(char const *s,
159 160 161 162 163 164 165 166
                     char **end_ptr,
                     int base,
                     int *result);

int virStrToLong_ui(char const *s,
                    char **end_ptr,
                    int base,
                    unsigned int *result);
167 168 169 170 171 172 173 174
int virStrToLong_l(char const *s,
                   char **end_ptr,
                   int base,
                   long *result);
int virStrToLong_ul(char const *s,
                    char **end_ptr,
                    int base,
                    unsigned long *result);
175 176 177 178
int virStrToLong_ll(char const *s,
                    char **end_ptr,
                    int base,
                    long long *result);
D
Daniel P. Berrange 已提交
179 180 181 182
int virStrToLong_ull(char const *s,
                     char **end_ptr,
                     int base,
                     unsigned long long *result);
183 184 185
int virStrToDouble(char const *s,
                   char **end_ptr,
                   double *result);
D
Daniel P. Berrange 已提交
186

187 188 189 190
int virScaleInteger(unsigned long long *value, const char *suffix,
                    unsigned long long scale, unsigned long long limit)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_RETURN_CHECK;

191 192
int virHexToBin(unsigned char c);

E
Eric Blake 已提交
193 194 195 196 197 198
void virSkipSpaces(const char **str) ATTRIBUTE_NONNULL(1);
void virSkipSpacesAndBackslash(const char **str) ATTRIBUTE_NONNULL(1);
void virTrimSpaces(char *str, char **endp) ATTRIBUTE_NONNULL(1);
void virSkipSpacesBackwards(const char *str, char **endp)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(2);

199
int virParseNumber(const char **str);
200 201
int virParseVersionString(const char *str, unsigned long *version,
                          bool allowMissing);
E
Eric Blake 已提交
202 203 204 205
int virAsprintf(char **strp, const char *fmt, ...)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(2) ATTRIBUTE_FMT_PRINTF(2, 3);
int virVasprintf(char **strp, const char *fmt, va_list list)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(2) ATTRIBUTE_FMT_PRINTF(2, 0);
C
Chris Lalancette 已提交
206 207 208 209
char *virStrncpy(char *dest, const char *src, size_t n, size_t destbytes)
    ATTRIBUTE_RETURN_CHECK;
char *virStrcpy(char *dest, const char *src, size_t destbytes)
    ATTRIBUTE_RETURN_CHECK;
210
# define virStrcpyStatic(dest, src) virStrcpy((dest), (src), sizeof(dest))
211

212
int virDiskNameToIndex(const char* str);
213
char *virIndexToDiskName(int idx, const char *prefix);
214 215 216 217 218 219 220 221 222

int virEnumFromString(const char *const*types,
                      unsigned int ntypes,
                      const char *type);

const char *virEnumToString(const char *const*types,
                            unsigned int ntypes,
                            int type);

223
# define VIR_ENUM_IMPL(name, lastVal, ...)                               \
224
    static const char *const name ## TypeList[] = { __VA_ARGS__ };      \
E
Eric Blake 已提交
225
    verify(ARRAY_CARDINALITY(name ## TypeList) == lastVal);             \
226 227 228 229 230 231 232 233 234 235 236
    const char *name ## TypeToString(int type) {                        \
        return virEnumToString(name ## TypeList,                        \
                               ARRAY_CARDINALITY(name ## TypeList),     \
                               type);                                   \
    }                                                                   \
    int name ## TypeFromString(const char *type) {                      \
        return virEnumFromString(name ## TypeList,                      \
                                 ARRAY_CARDINALITY(name ## TypeList),   \
                                 type);                                 \
    }

237
# define VIR_ENUM_DECL(name)                             \
238 239 240
    const char *name ## TypeToString(int type);         \
    int name ## TypeFromString(const char*type);

241
# ifndef HAVE_GETUID
D
Daniel P. Berrange 已提交
242
static inline int getuid (void) { return 0; }
243
# endif
D
Daniel P. Berrange 已提交
244

245 246 247 248
# ifndef HAVE_GETEUID
static inline int geteuid (void) { return 0; }
# endif

249
# ifndef HAVE_GETGID
D
Daniel P. Berrange 已提交
250
static inline int getgid (void) { return 0; }
251
# endif
D
Daniel P. Berrange 已提交
252

253
char *virGetHostname(virConnectPtr conn);
254

G
Guido Günther 已提交
255 256
int virKillProcess(pid_t pid, int sig);

257 258 259 260
char *virGetUserDirectory(void);
char *virGetUserConfigDirectory(void);
char *virGetUserCacheDirectory(void);
char *virGetUserRuntimeDirectory(void);
261
char *virGetUserName(uid_t uid);
262
char *virGetGroupName(gid_t gid);
263
int virGetUserID(const char *name,
264
                 uid_t *uid) ATTRIBUTE_RETURN_CHECK;
265
int virGetGroupID(const char *name,
266
                  gid_t *gid) ATTRIBUTE_RETURN_CHECK;
267

268 269
char *virFileFindMountPoint(const char *type);

270
void virFileWaitForDevices(void);
D
Daniel P. Berrange 已提交
271

272
# define virBuildPath(path, ...) virBuildPathInternal(path, __VA_ARGS__, NULL)
273 274
int virBuildPathInternal(char **path, ...) ATTRIBUTE_SENTINEL;

275
bool virIsDevMapperDevice(const char *dev_name) ATTRIBUTE_NONNULL(1);
276

277
#endif /* __VIR_UTIL_H__ */