OMS-Auditd-Plugin/AuditStatus.h

121 строка
4.7 KiB
C++

/*
microsoft-oms-auditd-plugin
Copyright (c) Microsoft Corporation
All rights reserved.
MIT License
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the ""Software""), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
#ifndef AUOMS_AUDITSTATUS_H
#define AUOMS_AUDITSTATUS_H
#include "Netlink.h"
class AuditStatus {
public:
enum class FieldMask: uint32_t {
Enabled = 0x0001, // AUDIT_STATUS_ENABLED
Failure = 0x0002, // AUDIT_STATUS_FAILURE
Pid = 0x0004, // AUDIT_STATUS_PID
RateLimit = 0x0008, // AUDIT_STATUS_RATE_LIMIT
BacklogLimit = 0x0010, // AUDIT_STATUS_BACKLOG_LIMIT
BacklogWaitTime = 0x0020, // AUDIT_STATUS_BACKLOG_WAIT_TIME
Lost = 0x0040, // AUDIT_STATUS_LOST
V1Status = 0x001F, // Only fields available in old kernels (e.g. < 3.12.0)
};
enum class Feature: uint32_t {
BacklogLimit = 0x00000001, // AUDIT_FEATURE_BITMAP_BACKLOG_LIMIT
BacklogWaitTime = 0x00000002, // AUDIT_FEATURE_BITMAP_BACKLOG_WAIT_TIME
ExecutablePath = 0x00000004, // AUDIT_FEATURE_BITMAP_EXECUTABLE_PATH
ExcludeExtend = 0x00000008, // AUDIT_FEATURE_BITMAP_EXCLUDE_EXTEND
SessionidFilter = 0x00000010, // AUDIT_FEATURE_BITMAP_SESSIONID_FILTER
LostReset = 0x00000020, // AUDIT_FEATURE_BITMAP_LOST_RESET
FilterFs = 0x00000040, // AUDIT_FEATURE_BITMAP_FILTER_FS
};
AuditStatus() {
::memset(this, 0, sizeof(AuditStatus));
}
inline bool HasFeature(Feature feature) {
if (feature == Feature::BacklogLimit) {
return _feature_bitmap == 0 || (_feature_bitmap & static_cast<uint32_t>(Feature::BacklogLimit)) != 0;
} else {
return (_feature_bitmap & static_cast<uint32_t>(feature)) != 0;
}
}
uint32_t GetEnabled() { return _enabled; }
uint32_t GetFailure() { return _failure; }
uint32_t GetPid() { return _pid; }
uint32_t GetRateLimit() { return _rate_limit; }
uint32_t GetBacklogLimit() { return _backlog_limit; }
uint32_t GetLost() { return _lost; }
uint32_t GetBacklog() { return _backlog; }
uint32_t GetBacklogWaitTime() { return _backlog_wait_time; }
void SetEnabled(uint32_t value) {
_mask |= static_cast<uint32_t>(FieldMask::Enabled);
_enabled = value;
}
void SetFailure(uint32_t value) {
_mask |= static_cast<uint32_t>(FieldMask::Failure);
_failure = value;
}
void SetPid(uint32_t value) {
_mask |= static_cast<uint32_t>(FieldMask::Pid);
_pid = value;
}
void SetRateLimit(uint32_t value) {
_mask |= static_cast<uint32_t>(FieldMask::RateLimit);
_rate_limit = value;
}
void SetBacklogLimit(uint32_t value) {
_mask |= static_cast<uint32_t>(FieldMask::BacklogLimit);
_backlog_limit = value;
}
void SetBacklogWaitTime(uint32_t value) {
_mask |= static_cast<uint32_t>(FieldMask::BacklogWaitTime);
_backlog_wait_time = value;
}
void SetLost(uint32_t value) {
_mask |= static_cast<uint32_t>(FieldMask::Lost);
_lost = value;
}
int GetStatus(Netlink& netlink);
int UpdateStatus(Netlink& netlink);
private:
// These fields mirror the struct audit_status fields defined in /usr/include/linux/audit.h
// DO NOT ADD OR REMOVE ANY FIELDS IN THE CLASS (except to mirror struct audit_status fields defined in /usr/include/linux/audit.h)
uint32_t _mask;
uint32_t _enabled;
uint32_t _failure;
uint32_t _pid;
uint32_t _rate_limit;
uint32_t _backlog_limit;
uint32_t _lost;
uint32_t _backlog;
uint32_t _feature_bitmap;
uint32_t _backlog_wait_time;
};
#endif //AUOMS_AUDITSTATUS_H