PureBoard

Code: Alles auswählen

;http://msdn.microsoft.com/en-us/library/aa378941%28VS.85%29.aspx
;http://msdn.microsoft.com/en-us/library/aa378847%28VS.85%29.aspx

EnableExplicit

Define text.s="Restore Point"
Define Libef
Define *Func

Libef = LoadLibrary_("SrClient.dll")

Debug Libef

If Libef
  ;*Func = GetProcAddress_(Libef, "SRSetRestorePoint")
  *Func = GetProcAddress_(Libef, "CreateRestorePoint")
 
  Debug *Func
 
  If *Func
    CallFunctionFast(*Func, @text , #APPLICATION_INSTALL, #BEGIN_SYSTEM_CHANGE)
  EndIf
EndIf
FreeLibrary_(Libef)

Code: Alles auswählen

;typedef BOOL (WINAPI *PFN_SETRESTOREPTW) (PRESTOREPOINTINFOW, PSTATEMGRSTATUS);
EnableExplicit

#RPC_C_AUTHN_LEVEL_DEFAULT 		= 0
#RPC_C_IMP_LEVEL_IMPERSONATE 	= 3
#EOAC_NONE										= 0

; typedef struct _RESTOREPTINFO {
;   DWORD dwEventType;
;   DWORD dwRestorePtType;
;   INT64 llSequenceNumber;
;   TCHAR szDescription[MAX_DESC];
; } RESTOREPOINTINFO, *PRESTOREPOINTINFO;
Structure RESTOREPOINTINFO
	dwEventType.i
	dwRestorePtType.i
	llSequenceNumber.q
	szDescription.s{#MAX_COLUMN_DESC_LEN}
EndStructure

; typedef struct _SMGRSTATUS {
;   UINT  nStatus;
;   INT64 llSequenceNumber;
; } STATEMGRSTATUS, *PSTATEMGRSTATUS;
Structure STATEMGRSTATUS
	nStatus.l
	llSequenceNumber.q
EndStructure

Procedure.b Failed(hr.l)
	ProcedureReturn hr <> #S_OK
EndProcedure

Procedure.s PBSetRestorePoint()
   Protected RestorePtInfo.RESTOREPOINTINFO
   Protected SMgrStatus.STATEMGRSTATUS
   Protected dwErr.l = #ERROR_SUCCESS
   Protected hSrClient.l = #Null
   Protected hr.i = #S_OK
   Protected fRet.b = #False
   Protected *pFunc.l = #Null
  
   hr = CoInitializeEx_(#Null, #COINIT_MULTITHREADED )
   If Failed(hr)
   	ProcedureReturn "Unexpected error: CoInitializeEx() failed with " + Str(hr) + "."
   EndIf

   ; Initialize COM security To enable NetworkService,
   ; LocalService And System To make callbacks To the process
   ; calling  System Restore. This is required For any process
   ; that calls SRSetRestorePoint.
   
   ; fRet = InitializeCOMSecurity_()
		hr =  CoInitializeSecurity_(#Null, -1, #Null, #Null, #RPC_C_AUTHN_LEVEL_DEFAULT, #RPC_C_IMP_LEVEL_IMPERSONATE, #Null, #EOAC_NONE, #Null);
   If Failed(hr)
   	ProcedureReturn "Unexpected error: failed to initialize COM security. (" + Str(hr) + ")"
   EndIf
   
   ; Initialize the RESTOREPOINTINFO Structure
   RestorePtInfo\dwEventType = #BEGIN_SYSTEM_CHANGE;

   ; Notify the system that changes are about To be made.
   ; An application is To be installed.
   RestorePtInfo\dwRestorePtType = #APPLICATION_INSTALL;

   ; RestPtInfo.llSequenceNumber must be 0 when creating a Restore point.
   RestorePtInfo\llSequenceNumber = 0

   ; String To be displayed by System Restore For this Restore point.
   ;StringCbCopyW(RestorePtInfo.szDescription,
   ;      SizeOf(RestorePtInfo.szDescription),
   ;      L"First Restore Point");
   RestorePtInfo\szDescription = "PB Test Restore Point"
   
   ; Load the DLL, which may Not exist on Windows server
   ;hSrClient = LoadLibraryW(L"srclient.dll");
   hSrClient = LoadLibrary_("SrClient.dll")
   If hSrClient = #Null
   	ProcedureReturn "System Restore is not present."
   EndIf

   ; If the library is loaded, find the entry point
;    fnSRSetRestorePointW = (PFN_SETRESTOREPTW) GetProcAddress(
;       hSrClient, "SRSetRestorePointW");
;    If (NULL == fnSRSetRestorePointW)
;    {
;       wprintf(L"Failed to find SRSetRestorePointW.\n");
;       Goto exit;
;    }
	*pFunc = GetProcAddress_(hSrClient, "SRSetRestorePointA")
	If *pFunc = #Null
		ProcedureReturn "Failed to find SRSetRestorePointW. " + Str(GetLastError_())
	EndIf


;    fRet = fnSRSetRestorePointW(&RestorePtInfo, &SMgrStatus);
;    If(!fRet)
;    {
;       dwErr = SMgrStatus.nStatus;
;       If(dwErr == ERROR_SERVICE_DISABLED)
;       {
;          wprintf(L"System Restore is turned off.\n");
;          Goto exit;
;       }
;       wprintf(L"Failure to create the restore point; error=%u.\n", dwErr);
;       Goto exit;
;    }
; 
;    wprintf(L"Restore point created; number=%I64d.\n", SMgrStatus.llSequenceNumber);
	fRet = CallFunctionFast(*pFunc, @RestorePtInfo, @SMgrStatus)
	If Not fRet
		dwErr = SMgrStatus\nStatus
		If dwErr = #ERROR_SERVICE_DISABLED
			ProcedureReturn "System Restore is turned off. (Error code: " + Str(dwErr) + ")"
		EndIf
	EndIf

	; The application performs some installation operations here.
	MessageBox_(#Null, "ich mache was ;o)", "", #MB_OK)

   ; It is Not necessary To call SrSetRestorePoint To indicate that the
   ; installation is complete except in the Case of ending a nested
   ; Restore point. Every BEGIN_NESTED_SYSTEM_CHANGE must have a
   ; corresponding END_NESTED_SYSTEM_CHANGE Or the application cannot
   ; create new Restore points.

   ; Update the RESTOREPOINTINFO Structure To notify the
   ; system that the operation is finished.
   RestorePtInfo\dwEventType = #END_SYSTEM_CHANGE;

   ; End the system change by using the sequence number
   ; received from the first call To SRSetRestorePoint.
   RestorePtInfo\llSequenceNumber = SMgrStatus\llSequenceNumber;

   ; Notify the system that the operation is done And that this
   ; is the End of the Restore point.
;    fRet = fnSRSetRestorePointW(&RestorePtInfo, &SMgrStatus);
;    If(!fRet)
;    {
;       dwErr = SMgrStatus.nStatus;
;       wprintf(L"Failure to end the restore point; error=%u.\n", dwErr);
;       Goto exit;
;    }
	fRet = CallFunctionFast(*pFunc, @RestorePtInfo, @SMgrStatus)
	If Not fRet
		dwErr = SMgrStatus\nStatus
		ProcedureReturn "Failure to end the restore point; error=" + Str(dwErr)
	EndIf	
	
	If hSrClient <> #Null
		FreeLibrary_(hSrClient)
	EndIf
	CoUninitialize_()
	
	ProcedureReturn "alles gut :)"
 EndProcedure 
 
 
 MessageBox_(#Null, PBSetRestorePoint(), "", #MB_OK)
 
 End