ADReports/ADGetPhoto.cpp at master · brechtsanders/ADReports · GitHub

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
#include <stdio.h>
#include <ctype.h>
#include <string>
#include <vector>
#include "adreports_version.h"
#include "ldapconnection.h"
#include "dataoutput.h"
#include "adformats.h"

void show_help()
{
  printf(
    "ADGetPhoto v" ADREPORTS_VERSION_STRING " - extract Active Directory photos\n" \
    "Credits: " ADREPORTS_CREDITS "\n" \
    "Usage:  ADGetPhoto " LDAP_COMMAND_LINE_PARAMETERS " [-c] [-l login] [-q ldapfilter]\n" \
    "Parameters:\n" \
    LDAP_COMMAND_LINE_HELP \
    "  -c             \tExtract photo of current user (default = all users)\n" \
    "  -l login       \tLogin name of user to extract photo of\n" \
    "  -q ldapfilter  \tLDAP filter\n" \
    "\n"
  );
}

int main (int argc, char *argv[])
{
  std::string login;
  std::string ldapfilter;
  LDAPConnection* ldap = new LDAPConnection;
  LDAPResponse* result;

  //parse command line parameters
  {
    int i = 0;
    const char* param;
    bool paramerror = false;
    while (!paramerror && ++i < argc) {
      if (!argv[i][0] || (argv[i][0] != '/' && argv[i][0] != '-')) {
        paramerror = true;
        break;
      }
      if (!ldap->ProcessCommandLineParameter(argc, argv, i)) {
        param = NULL;
        switch (tolower(argv[i][1])) {
          case '?' :
            show_help();
            return 0;
          case 'c' :
            login = get_current_login ();
            break;
          case 'l' :
            if (argv[i][2])
              param = argv[i] + 2;
            else if (i + 1 < argc && argv[i + 1])
              param = argv[++i];
            if (!param)
              return false;
            login = strdup(param);
            break;
          case 'q' :
            if (argv[i][2])
              param = argv[i] + 2;
            else if (i + 1 < argc && argv[i + 1])
              param = argv[++i];
            if (!param)
              return false;
            ldapfilter = param;
            break;
          default :
            paramerror = true;
            break;
        }
      }
    }
    if (paramerror) {
      fprintf(stderr, "Invalid command line parameters\n");
      return 1;
    }
  }

  //do LDAP stuff
  const char* ldaperrmsg;
  if ((ldaperrmsg  = ldap->Open()) != NULL) {
    fprintf(stderr, "Error opening LDAP connection: %s\n", ldaperrmsg);
    return 2;
  }
  std::string searchfilter = "(&(objectCategory=person)(objectClass=user)";
  if (login.size() > 0) {
    searchfilter += "(sAMAccountName=";
    searchfilter += login;
    searchfilter += ")";
  }
  if (!ldapfilter.empty()) {
    searchfilter += "(";
    searchfilter += ldapfilter;
    searchfilter += ")";
  }
  searchfilter += ")";
/////printf("%s\n", searchfilter.c_str());
  if ((result = ldap->Search(searchfilter.c_str())) == NULL) {
    fprintf(stderr, "LDAP search failed\n");
  } else {
    if (result->Rewind()) {
      std::string filename;
      char* s;
      size_t len;
      FILE* f;
      do {
        s = result->GetAttribute("sAMAccountName");
        printf("%s\n", s);
        filename = std::string(s) + ".jpg";
        free(s);
        if ((s = (char*)result->GetAttributeBin("thumbnailPhoto", &len)) != NULL) {
          if ((f = fopen(filename.c_str(), "wb")) != NULL) {
            fwrite(s, len, 1, f);
            fclose(f);
          }
          free(s);
        }
      } while (result->Next());
    }
    delete result;
  }

  //clean up
  ldap->Close();
  delete ldap;
  return 0;
}