strarc/linktrack.hpp at master · LTRData/strarc · 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
#include <malloc.h>

class LinkTrackerItem
{

private:

  DWORD dwVolumeSerialNumber;
  LONGLONG NodeNumber;
  UNICODE_STRING Name;
  LinkTrackerItem *Next;

  LinkTrackerItem(LinkTrackerItem * _Next,
		  PUNICODE_STRING _Name,
		  DWORD _dwVolumeSerialNumber,
		  LONGLONG _NodeNumber)
    : Next(_Next),
      dwVolumeSerialNumber(_dwVolumeSerialNumber),
      NodeNumber(_NodeNumber & 0x0000FFFFFFFFFFFF)
  {
    Name.Buffer = (PWSTR) malloc(_Name->Length);

    if (Name.Buffer == NULL)
      return;

    Name.Length = _Name->Length;
    Name.MaximumLength = _Name->Length;

    RtlCopyUnicodeString(&Name, _Name);
  }

  ~LinkTrackerItem()
  {
    if (Name.Buffer != NULL)
      free(Name.Buffer);
  }

public:

  LinkTrackerItem *DeleteAndGetNext()
  {
    LinkTrackerItem *next_item = Next;
    delete this;
    return next_item;
  }

  static
  LinkTrackerItem *NewItem(LinkTrackerItem * Next,
			   PUNICODE_STRING Name,
			   DWORD dwVolumeSerialNumber,
			   LONGLONG NodeNumber)
  {
    LinkTrackerItem *item =
      new LinkTrackerItem(Next, Name, dwVolumeSerialNumber, NodeNumber);

    if (item == NULL)
      return NULL;

    if (item->Name.Buffer == NULL)
      return NULL;

    return item;
  }

  bool
  Match(DWORD _dwVolumeSerialNumber,
	LONGLONG _NodeNumber,
	PUNICODE_STRING * _Name)
  {
    if (this == NULL)
      return false;

    if ((dwVolumeSerialNumber == _dwVolumeSerialNumber) &&
	(NodeNumber == (_NodeNumber & 0x0000FFFFFFFFFFFF)))
      {
	*_Name = &Name;
	return true;
      }

    return false;
  }

  LinkTrackerItem *GetNext()
  {
    if (this == NULL)
      return NULL;

    return Next;
  }
};