o jJ@sddlmZddlmZmZmZmZmZmZm Z ddl m Z m Z m Z mZmZddlmZddlmZmZddlmZedZedZGd d d eZeZGd d d eZd ZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4d Z5d!Z6dZ7d"Z8d#Z9d$Z:d%Z;d&ZdZ?d(Z@d)ZAd*ZBd+ZCd,ZDd-ZEd&ZFd$ZGd.ZHd/ZId'ZJd%ZKd0ZLd!ZMd"ZNdZOd1ZPd ZQd2ZRd3ZSd4ZTd5ZUd6ZVd7ZWdZXdZYd8ZZdZ[dZ\dZ]d Z^dZ_dZ`d9Zad:Zbd!ZcdZdd"Zed;Zfd5ZgdZhd!ZidZjdZkdZldZmdZnd Zodd?d?e ZtGd@dAdAeZuGdBdCdCeZvGdDdEdEe ZwGdFdGdGe ZxGdHdIdIeZyGdJdKdKe ZzGdLdMdMeZ{GdNdOdOe Z|GdPdQdQeZ}GdRdSdSe Z~GdTdUdUeZGdVdWdWe ZGdXdYdYeZGdZd[d[e ZGd\d]d]e ZGd^d_d_e ZGd`dadae ZGdbdcdceZGdddedeeZGdfdgdge ZGdhdidieZGdjdkdke ZGdldmdmeZGdndodoe ZGdpdqdqe ZGdrdsdse ZGdtdudue ZGdvdwdwe ZGdxdydye ZGdzd{d{e ZGd|d}d}e ZGd~dde ZGddde ZeefeefeefeefeefdZddZddZe e e-e fddZe dfddZddZddZdS)) system_errors) ULONGLONGUINTUSHORTLPWSTRDWORDULONGNULL)NDRCALL NDRSTRUCTNDRUNION NDRPOINTERNDRUniConformantArray)DCERPCException)uuidtup_to_bin string_to_bin)DRIVER_INFO_2_ARRAY)z$76F03F96-CDFD-44FC-A22C-64950A001209z1.0z$9940CA8E-512F-4C58-88A9-61098D6896BDc@seZdZdddZddZdS)DCERPCSessionErrorNcCst||||dS)N)r__init__)self error_string error_codepacketr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/aiosmb/dcerpc/v5/par.pyr"szDCERPCSessionError.__init__cCsF|j}|tjvrtj|d}tj|d}d|j||fSd|jS)Nrz'RPRN SessionError: code: 0x%x - %s - %sz+RPRN SessionError: unknown error code: 0x%x)rrERROR_MESSAGES)rkeyerror_msg_shorterror_msg_verboserrr__str__%s   zDCERPCSessionError.__str__)NNN)__name__ __module__ __qualname__rr rrrrr!s  rc@eZdZdeffZdS)PSTRING_HANDLEDataN)r!r"r# STRING_HANDLEreferentrrrrr%3r% ii i0@i iriiiiiiiiiiili@i iiiiiiwwiwipii i@ipiiiii@iiii c@seZdZdZddZdS)PRINTER_HANDLE))r&z20s=b""cCs|jdurdSdS)NTr-r,)_isNDR64)rrrr getAlignments zPRINTER_HANDLE.getAlignmentN)r!r"r# structurer4rrrrr2s r2c@eZdZdZdS) BYTE_ARRAYcNr!r"r#itemrrrrr7r7c@r$) PBYTE_ARRAYr&N)r!r"r#r7r(rrrrr<r)r<c@eZdZdefdeffZdS)DEVMODE_CONTAINERcbBufpDevModeN)r!r"r#rr<r5rrrrr>r>c@s:eZdZdefdefdefdefdefdefdeffZdS) SPLCLIENT_INFO_1dwSize pMachineName pUserName dwBuildNumdwMajorVersiondwMinorVersionwProcessorArchitectureN)r!r"r#rrrr5rrrrrBsrBc@r$)PSPLCLIENT_INFO_1r&N)r!r"r#rBr(rrrrrJr)rJc@r$)SPLCLIENT_INFO_2notUsedN)r!r"r#rr5rrrrrKr)rKc@r$)PSPLCLIENT_INFO_2r&N)r!r"r#rKr(rrrrrMr)rMc @sLeZdZdefdefdefdefdefdefdefdefdefd eff Zd S) SPLCLIENT_INFO_3cbSizedwFlagsrDrErFrGrHrI hSplPrinterN) r!r"r#rrrrrr5rrrrrNsrNc@r$)PSPLCLIENT_INFO_3r&N)r!r"r#rNr(rrrrrRr)rRc@r$) DRIVER_INFO_1pNameN)r!r"r#r'r5rrrrrSr)rSc@r$)PDRIVER_INFO_1r&N)r!r"r#rSr(rrrrrUr)rUc@s4eZdZdefdefdefdefdefdeffZdS) DRIVER_INFO_2cVersionrT pEnvironment pDriverPath pDataFile pConfigFileN)r!r"r#rrr5rrrrrVrVc@r$)PDRIVER_INFO_2r&N)r!r"r#rVr(rrrrr]r)r]c@s(eZdZdeffZdefdefdZdS)DRIVER_INFO_UNIONtagpNotUsedLevel2)rr/N)r!r"r#r commonHdrrUr]unionrrrrr^s  r^c@r=)DRIVER_CONTAINERLevel DriverInfoN)r!r"r#rr^r5rrrrrd rArdc@s.eZdZdeffZdefdefdefdZdS)CLIENT_INFO_UNIONr_ pClientInfo1 pNotUsed1 pNotUsed2)rr/N) r!r"r#rrbrJrMrRrcrrrrrgs rgc@r=)SPLCLIENT_CONTAINERre ClientInfoN)r!r"r#rrgr5rrrrrlrArlc@r6) USHORT_ARRAYzrrlr5rrrrrbrrc@r=)RpcAsyncOpenPrinterResponsepHandlerNr!r"r#r2rr5rrrrrlrArc@seZdZdZdeffZdS)RpcAsyncClosePrinter phPrinterN)r!r"r#rr2r5rrrrrssrc@r=)RpcAsyncClosePrinterResponserrNrrrrrryrArc@s2eZdZdZdefdefdefdefdeffZdS)RpcAsyncEnumPrinterDrivers(rTrXrepDriversr?N) r!r"r#rr'rrr<r5rrrrrrrc@r)"RpcAsyncEnumPrinterDriversResponserrrrNrrrrrrr|rc@s&eZdZdZdefdefdeffZdS)RpcAsyncAddPrinterDriver'rTpDriverContainerdwFileCopyFlagsN)r!r"r#rr'rdrr5rrrrrs rc@r$) RpcAsyncAddPrinterDriverResponserN)r!r"r#rr5rrrrrr)r)rrrrrcCs(|tkr|S|dddkr|dS|S)N)r )stringrrrcheckNullStrings rcs"t}||d<||tIdHS)a 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. rN)rrequestMSRPC_UUID_WINSPOOL)dcerrrrrhRpcAsyncClosePrinters rcslt}t||d<||d<|turt|dd<n||d<||d<|tur)td||d<||tIdHS) a 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 (RpcAsync) 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. rrrr@rzpClientInfo cannot be NULLrN)rrr Exceptionrr)r printerNamerraccessRequiredrrrrrhRpcAsyncOpenPrinters rc st}||d<||d<t|d<||d<d}z||tIdH\}}|dur(|Wn"tyK}zt|ddkr;|d}WYd}~nd}~wwt}||d<||d<||d<||d <d ||d<||tIdHS) a 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. rrrrerNERROR_INSUFFICIENT_BUFFERrr?a)r~r rrrstrfind get_packet) rflagsnamelevelr bytesNeeded_errerrrhRpcAsyncEnumPrinterss2  rcs6t}t||d<||d<||d<||tIdHS)af 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. rTrrN)rrrr)rrTrrrrrrhRpcAsyncAddPrinterDrivers  rc s t}t||d<||d<||d<t|d<d|d<z||tIdH\}}|dur,|Wn"tyO}zt|ddkr?|d }WYd}~nd}~wwt}t||d<||d<||d<d ||d<||d<||tIdH\} }|dur~d|fSt d | d| d } | j |fS) a 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. rTrXrerrr?Nrrrr) rrr rrrrrrr from_bytesjoindrivers) rrTrXrerrrrrdata_rawdatarrrhRpcAsyncEnumPrinterDrivers&s:    rN)aiosmb.dcerpc.v5raiosmb.dcerpc.v5.dtypesrrrrrrr aiosmb.dcerpc.v5.ndrr r r r raiosmb.dcerpc.v5.rpcrtraiosmb.dcerpc.v5.uuidrraiosmb.dcerpc.v5.rprnrMSRPC_UUID_PARrrr'r%JOB_ACCESS_ADMINISTERJOB_ACCESS_READ JOB_EXECUTEJOB_READ JOB_WRITEJOB_ALL_ACCESSPRINTER_ACCESS_ADMINISTERPRINTER_ACCESS_USEPRINTER_ACCESS_MANAGE_LIMITEDPRINTER_ALL_ACCESSPRINTER_EXECUTE PRINTER_READ PRINTER_WRITESERVER_ACCESS_ADMINISTERSERVER_ACCESS_ENUMERATESERVER_ALL_ACCESSSERVER_EXECUTE SERVER_READ SERVER_WRITESPECIFIC_RIGHTS_ALLSTANDARD_RIGHTS_ALLSTANDARD_RIGHTS_EXECUTESTANDARD_RIGHTS_READSTANDARD_RIGHTS_REQUIREDSTANDARD_RIGHTS_WRITE SYNCHRONIZEDELETE READ_CONTROL WRITE_DAC WRITE_OWNER GENERIC_READ GENERIC_WRITEGENERIC_EXECUTE GENERIC_ALLPRINTER_CHANGE_SET_PRINTERPRINTER_CHANGE_DELETE_PRINTERPRINTER_CHANGE_PRINTERPRINTER_CHANGE_ADD_JOBPRINTER_CHANGE_SET_JOBPRINTER_CHANGE_DELETE_JOBPRINTER_CHANGE_WRITE_JOBPRINTER_CHANGE_JOB!PRINTER_CHANGE_SET_PRINTER_DRIVERPRINTER_CHANGE_TIMEOUTPRINTER_CHANGE_ALLPRINTER_CHANGE_ALL_2!PRINTER_CHANGE_ADD_PRINTER_DRIVER$PRINTER_CHANGE_DELETE_PRINTER_DRIVERPRINTER_CHANGE_PRINTER_DRIVERPRINTER_CHANGE_ADD_FORMPRINTER_CHANGE_DELETE_FORMPRINTER_CHANGE_SET_FORMPRINTER_CHANGE_FORMPRINTER_CHANGE_ADD_PORTPRINTER_CHANGE_CONFIGURE_PORTPRINTER_CHANGE_DELETE_PORTPRINTER_CHANGE_PORT"PRINTER_CHANGE_ADD_PRINT_PROCESSOR%PRINTER_CHANGE_DELETE_PRINT_PROCESSORPRINTER_CHANGE_PRINT_PROCESSORPRINTER_CHANGE_ADD_PRINTER(PRINTER_CHANGE_FAILED_CONNECTION_PRINTERPRINTER_CHANGE_SERVERPRINTER_ENUM_LOCALPRINTER_ENUM_CONNECTIONSPRINTER_ENUM_NAMEPRINTER_ENUM_REMOTEPRINTER_ENUM_SHAREDPRINTER_ENUM_NETWORKPRINTER_ENUM_EXPANDPRINTER_ENUM_CONTAINERPRINTER_ENUM_ICON1PRINTER_ENUM_ICON2PRINTER_ENUM_ICON3PRINTER_ENUM_ICON8PRINTER_ENUM_HIDEPRINTER_NOTIFY_CATEGORY_2DPRINTER_NOTIFY_CATEGORY_ALLPRINTER_NOTIFY_CATEGORY_3DAPD_STRICT_UPGRADEAPD_STRICT_DOWNGRADEAPD_COPY_ALL_FILESAPD_COPY_NEW_FILESAPD_COPY_FROM_DIRECTORYAPD_DONT_COPY_FILES_TO_CLUSTERAPD_COPY_TO_ALL_SPOOLERSAPD_INSTALL_WARNED_DRIVERAPD_RETURN_BLOCKING_STATUS_CODEr2r7r<r>rBrJrKrMrNrRrSrUrVr]r^rdrgrlrnrorprwrxr}r~rrrrrrrrrOPNUMSrrrrrrrrrrs& $                  $&