# SECUREAUTH LABS. Copyright 2018 SecureAuth Corporation. All rights reserved.
#
# This software is provided under under a slightly modified version
# of the Apache Software License. See the accompanying LICENSE file
# for more information.
#
# Author: Alberto Solino (@agsolino)
#
# Description:
#   [MS-RPRN] Interface implementation
#
#   Best way to learn how to use these calls is to grab the protocol standard
#   so you understand what the call does, and then read the test case located
#   at https://github.com/SecureAuthCorp/impacket/tree/master/tests/SMB_RPC
#
#   Some calls have helper functions, which makes it even easier to use.
#   They are located at the end of this file. 
#   Helper functions start with "h"<name of the call>.
#   There are test cases for them too. 
#

import io
from os import stat

from aiosmb.dcerpc.v5.dtypes import ULONGLONG, UINT, USHORT, LPWSTR, DWORD, ULONG, NULL
from aiosmb.dcerpc.v5.ndr import NDRCALL, NDRSTRUCT, NDRUNION, NDRPOINTER, NDRUniConformantArray
from aiosmb.dcerpc.v5.ndr import NDRVaryingString

from aiosmb.dcerpc.v5.rpcrt import DCERPCException
from aiosmb.dcerpc.v5 import system_errors
from aiosmb.dcerpc.v5.uuid import uuidtup_to_bin
from aiosmb.dcerpc.v5.structure import Structure

MSRPC_UUID_RPRN = uuidtup_to_bin(('12345678-1234-ABCD-EF00-0123456789AB', '1.0'))

class DCERPCSessionError(DCERPCException):
	def __init__(self, error_string=None, error_code=None, packet=None):
		DCERPCException.__init__(self, error_string, error_code, packet)

	def __str__( self ):
		key = self.error_code
		if key in system_errors.ERROR_MESSAGES:
			error_msg_short = system_errors.ERROR_MESSAGES[key][0]
			error_msg_verbose = system_errors.ERROR_MESSAGES[key][1] 
			return 'RPRN SessionError: code: 0x%x - %s - %s' % (self.error_code, error_msg_short, error_msg_verbose)
		else:
			return 'RPRN SessionError: unknown error code: 0x%x' % self.error_code

################################################################################
# CONSTANTS
################################################################################
# 2.2.1.1.7 STRING_HANDLE
STRING_HANDLE = LPWSTR
class PSTRING_HANDLE(NDRPOINTER):
	referent = (
		('Data', STRING_HANDLE),
	)

# 2.2.3.1 Access Values
JOB_ACCESS_ADMINISTER         = 0x00000010
JOB_ACCESS_READ               = 0x00000020
JOB_EXECUTE                   = 0x00020010
JOB_READ                      = 0x00020020
JOB_WRITE                     = 0x00020010
JOB_ALL_ACCESS                = 0x000F0030
PRINTER_ACCESS_ADMINISTER     = 0x00000004
PRINTER_ACCESS_USE            = 0x00000008
PRINTER_ACCESS_MANAGE_LIMITED = 0x00000040
PRINTER_ALL_ACCESS            = 0x000F000C
PRINTER_EXECUTE               = 0x00020008
PRINTER_READ                  = 0x00020008
PRINTER_WRITE                 = 0x00020008
SERVER_ACCESS_ADMINISTER      = 0x00000001
SERVER_ACCESS_ENUMERATE       = 0x00000002
SERVER_ALL_ACCESS             = 0x000F0003
SERVER_EXECUTE                = 0x00020002
SERVER_READ                   = 0x00020002
SERVER_WRITE                  = 0x00020003
SPECIFIC_RIGHTS_ALL           = 0x0000FFFF
STANDARD_RIGHTS_ALL           = 0x001F0000
STANDARD_RIGHTS_EXECUTE       = 0x00020000
STANDARD_RIGHTS_READ          = 0x00020000
STANDARD_RIGHTS_REQUIRED      = 0x000F0000
STANDARD_RIGHTS_WRITE         = 0x00020000
SYNCHRONIZE                   = 0x00100000
DELETE                        = 0x00010000
READ_CONTROL                  = 0x00020000
WRITE_DAC                     = 0x00040000
WRITE_OWNER                   = 0x00080000
GENERIC_READ                  = 0x80000000
GENERIC_WRITE                 = 0x40000000
GENERIC_EXECUTE               = 0x20000000
GENERIC_ALL                   = 0x10000000

# 2.2.3.6.1 Printer Change Flags for Use with a Printer Handle
PRINTER_CHANGE_SET_PRINTER        = 0x00000002
PRINTER_CHANGE_DELETE_PRINTER     = 0x00000004
PRINTER_CHANGE_PRINTER            = 0x000000FF
PRINTER_CHANGE_ADD_JOB            = 0x00000100
PRINTER_CHANGE_SET_JOB            = 0x00000200
PRINTER_CHANGE_DELETE_JOB         = 0x00000400
PRINTER_CHANGE_WRITE_JOB          = 0x00000800
PRINTER_CHANGE_JOB                = 0x0000FF00
PRINTER_CHANGE_SET_PRINTER_DRIVER = 0x20000000
PRINTER_CHANGE_TIMEOUT            = 0x80000000
PRINTER_CHANGE_ALL                = 0x7777FFFF
PRINTER_CHANGE_ALL_2              = 0x7F77FFFF

# 2.2.3.6.2 Printer Change Flags for Use with a Server Handle
PRINTER_CHANGE_ADD_PRINTER_DRIVER        = 0x10000000
PRINTER_CHANGE_DELETE_PRINTER_DRIVER     = 0x40000000
PRINTER_CHANGE_PRINTER_DRIVER            = 0x70000000
PRINTER_CHANGE_ADD_FORM                  = 0x00010000
PRINTER_CHANGE_DELETE_FORM               = 0x00040000
PRINTER_CHANGE_SET_FORM                  = 0x00020000
PRINTER_CHANGE_FORM                      = 0x00070000
PRINTER_CHANGE_ADD_PORT                  = 0x00100000
PRINTER_CHANGE_CONFIGURE_PORT            = 0x00200000
PRINTER_CHANGE_DELETE_PORT               = 0x00400000
PRINTER_CHANGE_PORT                      = 0x00700000
PRINTER_CHANGE_ADD_PRINT_PROCESSOR       = 0x01000000
PRINTER_CHANGE_DELETE_PRINT_PROCESSOR    = 0x04000000
PRINTER_CHANGE_PRINT_PROCESSOR           = 0x07000000
PRINTER_CHANGE_ADD_PRINTER               = 0x00000001
PRINTER_CHANGE_FAILED_CONNECTION_PRINTER = 0x00000008
PRINTER_CHANGE_SERVER                    = 0x08000000

# 2.2.3.7 Printer Enumeration Flags
PRINTER_ENUM_LOCAL       = 0x00000002
PRINTER_ENUM_CONNECTIONS = 0x00000004
PRINTER_ENUM_NAME        = 0x00000008
PRINTER_ENUM_REMOTE      = 0x00000010
PRINTER_ENUM_SHARED      = 0x00000020
PRINTER_ENUM_NETWORK     = 0x00000040
PRINTER_ENUM_EXPAND      = 0x00004000
PRINTER_ENUM_CONTAINER   = 0x00008000
PRINTER_ENUM_ICON1       = 0x00010000
PRINTER_ENUM_ICON2       = 0x00020000
PRINTER_ENUM_ICON3       = 0x00040000
PRINTER_ENUM_ICON8       = 0x00800000
PRINTER_ENUM_HIDE        = 0x01000000


# 2.2.3.8 Printer Notification Values
PRINTER_NOTIFY_CATEGORY_2D  = 0x00000000
PRINTER_NOTIFY_CATEGORY_ALL = 0x00010000
PRINTER_NOTIFY_CATEGORY_3D  = 0x00020000

# 3.1.4.4.8 RpcAddPrinterDriverEx Values
APD_STRICT_UPGRADE              = 0x00000001
APD_STRICT_DOWNGRADE            = 0x00000002
APD_COPY_ALL_FILES              = 0x00000004
APD_COPY_NEW_FILES              = 0x00000008
APD_COPY_FROM_DIRECTORY         = 0x00000010
APD_DONT_COPY_FILES_TO_CLUSTER  = 0x00001000
APD_COPY_TO_ALL_SPOOLERS        = 0x00002000
APD_INSTALL_WARNED_DRIVER       = 0x00008000
APD_RETURN_BLOCKING_STATUS_CODE = 0x00010000


################################################################################
# STRUCTURES
################################################################################
# 2.2.1.1.4 PRINTER_HANDLE
class PRINTER_HANDLE(NDRSTRUCT):
	structure =  (
		('Data','20s=b""'),
	)
	def getAlignment(self):
		if self._isNDR64 is True:
			return 8
		else:
			return 4

# 2.2.1.2.1 DEVMODE_CONTAINER
class BYTE_ARRAY(NDRUniConformantArray):
	item = 'c'

class PBYTE_ARRAY(NDRPOINTER):
	referent = (
		('Data', BYTE_ARRAY),
	)

class DEVMODE_CONTAINER(NDRSTRUCT):
	structure =  (
		('cbBuf',DWORD),
		('pDevMode',PBYTE_ARRAY),
	)

# 2.2.1.11.1 SPLCLIENT_INFO_1
class SPLCLIENT_INFO_1(NDRSTRUCT):
	structure =  (
		('dwSize',DWORD),
		('pMachineName',LPWSTR),
		('pUserName',LPWSTR),
		('dwBuildNum',DWORD),
		('dwMajorVersion',DWORD),
		('dwMinorVersion',DWORD),
		('wProcessorArchitecture',USHORT),
	)

class PSPLCLIENT_INFO_1(NDRPOINTER):
	referent = (
		('Data', SPLCLIENT_INFO_1),
	)

# 2.2.1.11.2 SPLCLIENT_INFO_2
class SPLCLIENT_INFO_2(NDRSTRUCT):
	structure =  (
		('notUsed',ULONGLONG),
	)

class PSPLCLIENT_INFO_2(NDRPOINTER):
	referent = (
		('Data', SPLCLIENT_INFO_2),
	)
# 2.2.1.11.3 SPLCLIENT_INFO_3
class SPLCLIENT_INFO_3(NDRSTRUCT):
	structure =  (
		('cbSize',UINT),
		('dwFlags',DWORD),
		('dwFlags',DWORD),
		('pMachineName',LPWSTR),
		('pUserName',LPWSTR),
		('dwBuildNum',DWORD),
		('dwMajorVersion',DWORD),
		('dwMinorVersion',DWORD),
		('wProcessorArchitecture',USHORT),
		('hSplPrinter',ULONGLONG),
	)

class PSPLCLIENT_INFO_3(NDRPOINTER):
	referent = (
		('Data', SPLCLIENT_INFO_3),
	)
# 2.2.1.2.14 SPLCLIENT_CONTAINER
class CLIENT_INFO_UNION(NDRUNION):
	commonHdr = (
		('tag', ULONG),
	)
	union = {
		1 : ('pClientInfo1', PSPLCLIENT_INFO_1),
		2 : ('pNotUsed1', PSPLCLIENT_INFO_2),
		3 : ('pNotUsed2', PSPLCLIENT_INFO_3),
	}

class SPLCLIENT_CONTAINER(NDRSTRUCT):
	structure =  (
		('Level',DWORD),
		('ClientInfo',CLIENT_INFO_UNION),
	)


# 2.2.1.13.2 RPC_V2_NOTIFY_OPTIONS_TYPE
class USHORT_ARRAY(NDRUniConformantArray):
	item = '<H'

class PUSHORT_ARRAY(NDRPOINTER):
	referent = (
		('Data', USHORT_ARRAY),
	)

class RPC_V2_NOTIFY_OPTIONS_TYPE(NDRSTRUCT):
	structure =  (
		('Type',USHORT),
		('Reserved0',USHORT),
		('Reserved1',DWORD),
		('Reserved2',DWORD),
		('Count',DWORD),
		('pFields',PUSHORT_ARRAY),
	)

class PRPC_V2_NOTIFY_OPTIONS_TYPE_ARRAY(NDRPOINTER):
	referent = (
		('Data', RPC_V2_NOTIFY_OPTIONS_TYPE),
	)

# 2.2.1.13.1 RPC_V2_NOTIFY_OPTIONS
class RPC_V2_NOTIFY_OPTIONS(NDRSTRUCT):
	structure =  (
		('Version',DWORD),
		('Reserved',DWORD),
		('Count',DWORD),
		('pTypes',PRPC_V2_NOTIFY_OPTIONS_TYPE_ARRAY),
	)

class PRPC_V2_NOTIFY_OPTIONS(NDRPOINTER):
	referent = (
		('Data', RPC_V2_NOTIFY_OPTIONS),
	)

# 2.2.1.5.1 DRIVER_INFO_1
class DRIVER_INFO_1(NDRSTRUCT):
	structure = (
		('pName', STRING_HANDLE ),
	)
class PDRIVER_INFO_1(NDRPOINTER):
	referent = (
		('Data', DRIVER_INFO_1),
	)

# 2.2.1.5.2 DRIVER_INFO_2
class DRIVER_INFO_2(NDRSTRUCT):
	structure = (
		('cVersion',DWORD),
		('pName', STRING_HANDLE),
		('pEnvironment', STRING_HANDLE),
		('pDriverPath', STRING_HANDLE),
		('pDataFile', STRING_HANDLE),
		('pConfigFile', STRING_HANDLE),
	)
class PDRIVER_INFO_2(NDRPOINTER):
	referent = (
		('Data', DRIVER_INFO_2),
	)

class DRIVER_INFO_UNION(NDRUNION):
	commonHdr = (
		('tag', ULONG),
	)
	union = {
		1 : ('pNotUsed', PDRIVER_INFO_1),
		2 : ('Level2', PDRIVER_INFO_2),
	}

class DRIVER_CONTAINER(NDRSTRUCT):
	structure =  (
		('Level', DWORD),
		('DriverInfo', DRIVER_INFO_UNION),
	)

################################################################################
# Structure defs
################################################################################
class DRIVER_INFO_2_ENTRY:
	def __init__(self):
		self.Name = None
		self.Environment = None
		self.DriverPath = None
		self.DataFile = None
		self.ConfigFile = None
	
	def __str__(self):
		t = "DRIVER_INFO_2_ENTRY\r\n"
		t += "Name       : %s\r\n" % self.Name
		t += "Environment: %s\r\n" % self.Environment
		t += "DriverPath : %s\r\n" % self.DriverPath
		t += "DataFile   : %s\r\n" % self.DataFile
		t += "ConfigFile : %s\r\n" % self.ConfigFile
		return t

class DRIVER_INFO_2_ARRAY:
	def __init__(self):
		self.fixed = []
		self.fixed_size = 24

		self.drivers = []

	@staticmethod
	def from_bytes(data, elementNo):
		return DRIVER_INFO_2_ARRAY.from_buffer(io.BytesIO(data), elementNo)
	
	@staticmethod
	def from_buffer(buff, elementNo):
		res = DRIVER_INFO_2_ARRAY()
		for _ in range(elementNo):
			res.fixed.append(DRIVER_INFO_2_FIXED.from_buffer(buff))
		
		
		for i, fixed in enumerate(res.fixed):
			res.drivers.append(fixed.read_data(buff, res.fixed_size*i))
		
		return res
	
	def __str__(self):
		t = "DRIVER_INFO_2_ARRAY\r\n"
		for x in self.fixed:
			t += "FIXED: %s\r\n" % x
			t += "DRIVER: %s\r\n" % x.driver
		return t

	
class DRIVER_INFO_2_FIXED:
	def __init__(self):
		self.cVersion = None
		self.NameOffset = None
		self.EnvironmentOffset = None
		self.DriverPathOffset = None
		self.DataFileOffset = None
		self.ConfigFileOffset = None

		self.driver = None

	@staticmethod
	def read_utf16le(buff):
		res = b''
		while True:
			res += buff.read(1)
			if res[-3:] == b'\x00\x00\x00':
				break
		return res[:-2].decode('utf-16-le')

	def read_data(self, buff, offset_delta):
		self.driver = DRIVER_INFO_2_ENTRY()
		buff.seek(self.ConfigFileOffset + offset_delta,0)
		self.driver.ConfigFile = DRIVER_INFO_2_FIXED.read_utf16le(buff)

		buff.seek(self.DataFileOffset + offset_delta,0)
		self.driver.DataFile = DRIVER_INFO_2_FIXED.read_utf16le(buff)

		buff.seek(self.DriverPathOffset + offset_delta,0)
		self.driver.DriverPath = DRIVER_INFO_2_FIXED.read_utf16le(buff)

		buff.seek(self.EnvironmentOffset + offset_delta,0)
		self.driver.Environment = DRIVER_INFO_2_FIXED.read_utf16le(buff)

		buff.seek(self.NameOffset + offset_delta,0)
		self.driver.Name = DRIVER_INFO_2_FIXED.read_utf16le(buff)

		return self.driver

	@staticmethod
	def from_bytes(data):
		return DRIVER_INFO_2_FIXED.from_buffer(io.BytesIO(data))
	
	@staticmethod
	def from_buffer(buff):
		res = DRIVER_INFO_2_FIXED()
		res.cVersion = int.from_bytes(buff.read(4), byteorder='little', signed=False)
		res.NameOffset = int.from_bytes(buff.read(4), byteorder='little', signed=False)
		res.EnvironmentOffset = int.from_bytes(buff.read(4), byteorder='little', signed=False)
		res.DriverPathOffset = int.from_bytes(buff.read(4), byteorder='little', signed=False)
		res.DataFileOffset = int.from_bytes(buff.read(4), byteorder='little', signed=False)
		res.ConfigFileOffset = int.from_bytes(buff.read(4), byteorder='little', signed=False)
		return res
			

	def __str__(self):
		t = "DRIVER_INFO_2_FIXED\r\n"
		t += "Version          : %s\r\n" % self.cVersion
		t += "NameOffset       : %s\r\n" % self.NameOffset
		t += "EnvironmentOffset: %s\r\n" % self.EnvironmentOffset
		t += "DriverPathOffset : %s\r\n" % self.DriverPathOffset
		t += "DataFileOffset   : %s\r\n" % self.DataFileOffset
		t += "ConfigFileOffset : %s\r\n" % self.ConfigFileOffset
		return t

################################################################################
# RPC CALLS
################################################################################
# 3.1.4.2.1 RpcEnumPrinters (Opnum 0)
class RpcEnumPrinters(NDRCALL):
	opnum = 0
	structure = (
	   ('Flags', DWORD),
	   ('Name', STRING_HANDLE),
	   ('Level', DWORD),
	   ('pPrinterEnum', PBYTE_ARRAY),
	   ('cbBuf', DWORD),
	)

class RpcEnumPrintersResponse(NDRCALL):
	structure = (
	   ('pPrinterEnum', PBYTE_ARRAY),
	   ('pcbNeeded', DWORD),
	   ('pcReturned', DWORD),
	   ('ErrorCode', ULONG),
	)
# 3.1.4.2.2 RpcOpenPrinter (Opnum 1)
class RpcOpenPrinter(NDRCALL):
	opnum = 1
	structure = (
	   ('pPrinterName', STRING_HANDLE),
	   ('pDatatype', LPWSTR),
	   ('pDevModeContainer', DEVMODE_CONTAINER),
	   ('AccessRequired', DWORD),
	)

class RpcOpenPrinterResponse(NDRCALL):
	structure = (
	   ('pHandle', PRINTER_HANDLE),
	   ('ErrorCode', ULONG),
	)

# 3.1.4.2.9 RpcClosePrinter (Opnum 29)
class RpcClosePrinter(NDRCALL):
	opnum = 29
	structure = (
	   ('phPrinter', PRINTER_HANDLE),
	)

class RpcClosePrinterResponse(NDRCALL):
	structure = (
	   ('phPrinter', PRINTER_HANDLE),
	   ('ErrorCode', ULONG),
	)

# 3.1.4.10.4 RpcRemoteFindFirstPrinterChangeNotificationEx (Opnum 62)
class RpcRemoteFindFirstPrinterChangeNotification(NDRCALL):
	opnum = 65
	structure = (
	   ('hPrinter', PRINTER_HANDLE),
	   ('fdwFlags', DWORD),
	   ('fdwOptions', DWORD),
	   ('pszLocalMachine', LPWSTR),
	   ('dwPrinterLocal', DWORD),
	   ('pOptions', PRPC_V2_NOTIFY_OPTIONS),
	)

class RpcRemoteFindFirstPrinterChangeNotificationResponse(NDRCALL):
	structure = (
	   ('ErrorCode', ULONG),
	)

# 3.1.4.10.4 RpcRemoteFindFirstPrinterChangeNotificationEx (Opnum 65)
class RpcRemoteFindFirstPrinterChangeNotificationEx(NDRCALL):
	opnum = 65
	structure = (
	   ('hPrinter', PRINTER_HANDLE),
	   ('fdwFlags', DWORD),
	   ('fdwOptions', DWORD),
	   ('pszLocalMachine', LPWSTR),
	   ('dwPrinterLocal', DWORD),
	   ('pOptions', PRPC_V2_NOTIFY_OPTIONS),
	)

class RpcRemoteFindFirstPrinterChangeNotificationExResponse(NDRCALL):
	structure = (
	   ('ErrorCode', ULONG),
	)

# 3.1.4.2.14 RpcOpenPrinterEx (Opnum 69)
class RpcOpenPrinterEx(NDRCALL):
	opnum = 69
	structure = (
	   ('pPrinterName', STRING_HANDLE),
	   ('pDatatype', LPWSTR),
	   ('pDevModeContainer', DEVMODE_CONTAINER),
	   ('AccessRequired', DWORD),
	   ('pClientInfo', SPLCLIENT_CONTAINER),
	)

class RpcOpenPrinterExResponse(NDRCALL):
	structure = (
	   ('pHandle', PRINTER_HANDLE),
	   ('ErrorCode', ULONG),
	)

# 3.1.4.4.2 RpcEnumPrinterDrivers (Opnum 10)
class RpcEnumPrinterDrivers(NDRCALL):
	opnum = 10
	structure = (
	   ('pName', STRING_HANDLE),
	   ('pEnvironment', LPWSTR),
	   ('Level', DWORD),
	   ('pDrivers', PBYTE_ARRAY),
	   ('cbBuf', DWORD),
	)

class RpcEnumPrinterDriversResponse(NDRCALL):
	structure = (
	   ('pDrivers', PBYTE_ARRAY), # 
	   ('pcbNeeded', DWORD),
	   ('pcReturned', DWORD),
	   ('ErrorCode', ULONG),
	)

# 3.1.4.4.8 RpcAddPrinterDriverEx (Opnum 89)
class RpcAddPrinterDriverEx(NDRCALL):
	opnum = 89
	structure = (
	   ('pName', STRING_HANDLE),
	   ('pDriverContainer', DRIVER_CONTAINER),
	   ('dwFileCopyFlags', DWORD),
	)

class RpcAddPrinterDriverExResponse(NDRCALL):
	structure = (
	   ('ErrorCode', ULONG),
	)


################################################################################
# OPNUMs and their corresponding structures
################################################################################
OPNUMS = {
	0  : (RpcEnumPrinters, RpcEnumPrintersResponse),
	1  : (RpcOpenPrinter, RpcOpenPrinterResponse),
	10 : (RpcEnumPrinterDrivers, RpcEnumPrinterDriversResponse),
	29 : (RpcClosePrinter, RpcClosePrinterResponse),
	62 : (RpcRemoteFindFirstPrinterChangeNotification, RpcRemoteFindFirstPrinterChangeNotificationResponse),
	65 : (RpcRemoteFindFirstPrinterChangeNotificationEx, RpcRemoteFindFirstPrinterChangeNotificationExResponse),
	69 : (RpcOpenPrinterEx, RpcOpenPrinterExResponse),
}

################################################################################
# HELPER FUNCTIONS
################################################################################
def checkNullString(string):
	if string == NULL:
		return string

	if string[-1:] != '\x00':
		return string + '\x00'
	else:
		return string

async def hRpcOpenPrinter(dce, printerName, pDatatype = NULL, pDevModeContainer = NULL, accessRequired = SERVER_READ):
	"""
	RpcOpenPrinter retrieves a handle for a printer, port, port monitor, print job, or print server.
	Full Documentation: https://msdn.microsoft.com/en-us/library/cc244808.aspx

	:param DCERPC_v5 dce: a connected DCE instance.
	:param string printerName: A string for a printer connection, printer object, server object, job object, port
	object, or port monitor object. This MUST be a Domain Name System (DNS), NetBIOS, Internet Protocol version 4
	(IPv4), Internet Protocol version 6 (IPv6), or Universal Naming Convention (UNC) name that remote procedure
	call (RPC) binds to, and it MUST uniquely identify a print server on the network.
	:param string pDatatype: A string that specifies the data type to be associated with the printer handle.
	:param DEVMODE_CONTAINER pDevModeContainer: A DEVMODE_CONTAINER structure. This parameter MUST adhere to the specification in
	DEVMODE_CONTAINER Parameters (section 3.1.4.1.8.1).
	:param int accessRequired: The access level that the client requires for interacting with the object to which a
	handle is being opened.

	:return: a RpcOpenPrinterResponse instance, raises DCERPCSessionError on error.
	"""
	request = RpcOpenPrinter()
	request['pPrinterName'] = checkNullString(printerName)
	request['pDatatype'] = pDatatype
	if pDevModeContainer is NULL:
		request['pDevModeContainer']['pDevMode'] = NULL
	else:
		request['pDevModeContainer'] = pDevModeContainer

	request['AccessRequired'] = accessRequired
	return await dce.request(request)

async def hRpcClosePrinter(dce, phPrinter):
	"""
	RpcClosePrinter closes a handle to a printer object, server object, job object, or port object.
	Full Documentation: https://msdn.microsoft.com/en-us/library/cc244768.aspx

	:param DCERPC_v5 dce: a connected DCE instance.
	:param PRINTER_HANDLE phPrinter: A handle to a printer object, server object, job object, or port object.

	:return: a RpcClosePrinterResponse instance, raises DCERPCSessionError on error.
	"""
	request = RpcClosePrinter()
	request['phPrinter'] = phPrinter
	return await dce.request(request)


async def hRpcOpenPrinterEx(dce, printerName, pDatatype=NULL, pDevModeContainer=NULL, accessRequired=SERVER_READ,
					  pClientInfo=NULL):
	"""
	RpcOpenPrinterEx retrieves a handle for a printer, port, port monitor, print job, or print server
	Full Documentation: https://msdn.microsoft.com/en-us/library/cc244809.aspx

	:param DCERPC_v5 dce: a connected DCE instance.
	:param string printerName: A string for a printer connection, printer object, server object, job object, port
	object, or port monitor object. This MUST be a Domain Name System (DNS), NetBIOS, Internet Protocol version 4
	(IPv4), Internet Protocol version 6 (IPv6), or Universal Naming Convention (UNC) name that remote procedure
	call (RPC) binds to, and it MUST uniquely identify a print server on the network.
	:param string pDatatype: A string that specifies the data type to be associated with the printer handle.
	:param DEVMODE_CONTAINER pDevModeContainer: A DEVMODE_CONTAINER structure. This parameter MUST adhere to the specification in
	DEVMODE_CONTAINER Parameters (section 3.1.4.1.8.1).
	:param int accessRequired: The access level that the client requires for interacting with the object to which a
	handle is being opened.
	:param SPLCLIENT_CONTAINER pClientInfo: This parameter MUST adhere to the specification in SPLCLIENT_CONTAINER Parameters.

	:return: a RpcOpenPrinterExResponse instance, raises DCERPCSessionError on error.
	"""
	request = RpcOpenPrinterEx()
	request['pPrinterName'] = checkNullString(printerName)
	request['pDatatype'] = pDatatype
	if pDevModeContainer is NULL:
		request['pDevModeContainer']['pDevMode'] = NULL
	else:
		request['pDevModeContainer'] = pDevModeContainer

	request['AccessRequired'] = accessRequired
	if pClientInfo is NULL:
		raise Exception('pClientInfo cannot be NULL')

	request['pClientInfo'] = pClientInfo
	return await dce.request(request)


async def hRpcRemoteFindFirstPrinterChangeNotificationEx(dce, hPrinter, fdwFlags, fdwOptions=0, pszLocalMachine=NULL,
												   dwPrinterLocal=0, pOptions=NULL):
	"""
	creates a remote change notification object that monitors changes to printer objects and sends change notifications
	to a print client using either RpcRouterReplyPrinter (section 3.2.4.1.2) or RpcRouterReplyPrinterEx (section 3.2.4.1.4)
	Full Documentation: https://msdn.microsoft.com/en-us/library/cc244813.aspx

	:param DCERPC_v5 dce: a connected DCE instance.
	:param PRINTER_HANDLE hPrinter: A handle to a printer or server object.
	:param int fdwFlags: Flags that specify the conditions that are required for a change notification object to enter a signaled state.
	:param int fdwOptions: The category of printers for which change notifications are returned.
	:param string pszLocalMachine: A string that represents the name of the client computer.
	:param int dwPrinterLocal: An implementation-specific unique value that MUST be sufficient for the client to determine
	whether a call to RpcReplyOpenPrinter by the server is associated with the hPrinter parameter in this call.
	:param RPC_V2_NOTIFY_OPTIONS pOptions:  An RPC_V2_NOTIFY_OPTIONS structure that specifies printer or job members that the client listens to for notifications.

	:return: a RpcRemoteFindFirstPrinterChangeNotificationExResponse instance, raises DCERPCSessionError on error.
	"""
	request = RpcRemoteFindFirstPrinterChangeNotificationEx()

	request['hPrinter'] = hPrinter
	request['fdwFlags'] = fdwFlags
	request['fdwOptions'] = fdwOptions
	request['dwPrinterLocal'] = dwPrinterLocal
	if pszLocalMachine is NULL:
		raise Exception('pszLocalMachine cannot be NULL')
	request['pszLocalMachine'] = checkNullString(pszLocalMachine)
	request['pOptions'] = pOptions
	return await dce.request(request)

async def hRpcRemoteFindFirstPrinterChangeNotification(dce, hPrinter, fdwFlags, fdwOptions=0, pszLocalMachine=NULL,
												   dwPrinterLocal=0, pOptions=NULL):
	"""
	"""
	request = RpcRemoteFindFirstPrinterChangeNotification()

	request['hPrinter'] = hPrinter
	request['fdwFlags'] = fdwFlags
	request['fdwOptions'] = fdwOptions
	request['dwPrinterLocal'] = dwPrinterLocal
	if pszLocalMachine is NULL:
		raise Exception('pszLocalMachine cannot be NULL')
	request['pszLocalMachine'] = checkNullString(pszLocalMachine)
	request['pOptions'] = pOptions
	return await dce.request(request)

async def hRpcEnumPrinters(dce, flags, name = NULL, level = 1):
	"""
	RpcEnumPrinters enumerates available printers, print servers, domains, or print providers.
	Full Documentation: https://msdn.microsoft.com/en-us/library/cc244794.aspx

	:param DCERPC_v5 dce: a connected DCE instance.
	:param int flags: The types of print objects that this method enumerates. The value of this parameter is the
	result of a bitwise OR of one or more of the Printer Enumeration Flags (section 2.2.3.7).
	:param string name: NULL or a server name parameter as specified in Printer Server Name Parameters (section 3.1.4.1.4).
	:param level: The level of printer information structure.

	:return: a RpcEnumPrintersResponse instance, raises DCERPCSessionError on error.
	"""
	request = RpcEnumPrinters()
	request['Flags'] = flags
	request['Name'] = name
	request['pPrinterEnum'] = NULL
	request['Level'] = level
	bytesNeeded = 0
	try:
		dce.request(request)
	except DCERPCSessionError as e:
		if str(e).find('ERROR_INSUFFICIENT_BUFFER') < 0:
			raise
		bytesNeeded = e.get_packet()['pcbNeeded']

	request = RpcEnumPrinters()
	request['Flags'] = flags
	request['Name'] = name
	request['Level'] = level

	request['cbBuf'] = bytesNeeded
	request['pPrinterEnum'] = b'a' * bytesNeeded
	return await dce.request(request)

async def hRpcAddPrinterDriverEx(dce, pName, pDriverContainer, dwFileCopyFlags):
	"""
	RpcAddPrinterDriverEx installs a printer driver on the print server
	Full Documentation: https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rprn/b96cc497-59e5-4510-ab04-5484993b259b

	:param DCERPC_v5 dce: a connected DCE instance.
	:param pName
	:param pDriverContainer
	:param dwFileCopyFlags

	:return: raises DCERPCSessionError on error.
	"""
	request = RpcAddPrinterDriverEx()
	request['pName'] = checkNullString(pName)
	request['pDriverContainer'] = pDriverContainer
	request['dwFileCopyFlags'] = dwFileCopyFlags

	#return request
	return await dce.request(request)


async def hRpcEnumPrinterDrivers(dce, pName, pEnvironment, Level):
	"""
	RpcEnumPrinterDrivers enumerates the printer drivers installed on a specified print server.
	Full Documentation: https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rprn/857d00ac-3682-4a0d-86ca-3d3c372e5e4a

	:param DCERPC_v5 dce: a connected DCE instance.
	:param pName
	:param pEnvironment
	:param Level
	:param pDrivers
	:param cbBuf
	:param pcbNeeded
	:param pcReturned

	:return: raises DCERPCSessionError on error.
	"""
	# get value for cbBuf
	request = RpcEnumPrinterDrivers()
	request['pName']        = checkNullString(pName)
	request['pEnvironment'] = pEnvironment
	request['Level']        = Level
	request['pDrivers']     = NULL
	request['cbBuf']        = 0
	try:
		_, err = await dce.request(request)
		if err is not None:
			raise err
	except DCERPCSessionError as e:
		if str(e).find('ERROR_INSUFFICIENT_BUFFER') < 0:
			raise
		bytesNeeded = e.get_packet()['pcbNeeded']
	except Exception as e:
		return None, e

	# now do RpcEnumPrinterDrivers again
	request = RpcEnumPrinterDrivers()
	request['pName']        = checkNullString(pName)
	request['pEnvironment'] = pEnvironment
	request['Level']        = Level
	request['pDrivers']     = b'a' * bytesNeeded
	request['cbBuf']        = bytesNeeded

	#return request
	data_raw, err = await dce.request(request)
	if err is not None:
		return None, err
	
	data = DRIVER_INFO_2_ARRAY.from_bytes(b''.join(data_raw['pDrivers']), data_raw['pcReturned'])
	return data.drivers, err
