summaryrefslogtreecommitdiffstats
path: root/plugin/handler_socket/libhsclient/auto_file.hpp
blob: 03c357f4d4ed289775f2d9f3e1cc0596175c2558 (plain)
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
// vim:sw=2:ai

/*
 * Copyright (C) 2010 DeNA Co.,Ltd.. All rights reserved.
 * See COPYRIGHT.txt for details.
 */

#ifndef DENA_AUTO_FILE_HPP
#define DENA_AUTO_FILE_HPP

/* Workaround for _LARGE_FILES and _LARGE_FILE_API incompatibility on AIX */
#if defined(_AIX) && defined(_LARGE_FILE_API)
#undef _LARGE_FILE_API
#endif

#include <unistd.h>
#include <sys/types.h>
#include <dirent.h>
#include <stdio.h>

#include "util.hpp"

namespace dena {

struct auto_file : private noncopyable {
  auto_file() : fd(-1) { }
  ~auto_file() {
    reset();
  }
  int get() const { return fd; }
  int close() {
    if (fd < 0) {
      return 0;
    }
    const int r = ::close(fd);
    fd = -1;
    return r;
  }
  void reset(int x = -1) {
    if (fd >= 0) {
      this->close();
    }
    fd = x;
  }
 private:
  int fd;
};

struct auto_dir : private noncopyable {
  auto_dir() : dp(0) { }
  ~auto_dir() {
    reset();
  }
  DIR *get() const { return dp; }
  void reset(DIR *d = 0) {
    if (dp != 0) {
      closedir(dp);
    }
    dp = d;
  }
 private:
  DIR *dp;
};

};

#endif