PureBasic Forums - English

Code: Select all

; creating a directory junction - direct quick quasi-psuedo code conversion to PB from http://www.flexhex.com/docs/articles/hard-links.phtml and is not intended to be correct or used as is 
#SE_BACKUP_NAME$ = "SeBackupPrivilege"
#SE_RESTORE_NAME$ = "SeRestorePrivilege"
#IO_REPARSE_TAG_MOUNT_POINT = $A0000003L
#REPARSE_MOUNTPOINT_HEADER_SIZE = 8
#FILE_FLAG_OPEN_REPARSE_POINT = $00200000

Structure REPARSE_MOUNTPOINT_DATA_BUFFER
  ReparseTag.i
  ReparseDataLength.i
  Reserved.w
  ReparseTargetLength.w
  ReparseTargetMaximumLength.w
  Reserved1.w
  ReparseTarget.w[1]
EndStructure


Procedure.b IsPrivilegeEnabled(Privilege.s)
  Protected hToken.i, PrvEnb.b, TokenPriv.PRIVILEGE_SET
  Protected FuncRet.i, PChkRes.i
  hToken = 0
  PChkRes = #False
  
  FuncRet = OpenThreadToken_(GetCurrentThread(), #TOKEN_QUERY, #False, @Token)
  If FuncRet = 0 And GetLastError_() = #ERROR_NO_TOKEN
    FuncRet = OpenProcessToken_(GetCurrentProcess_(), #TOKEN_QUERY, @hToken)
    If FuncRet <> 0
      TokenPriv\PrivilegeCount = 1
      TokenPriv\Control = 0
      FuncRet = LookupPrivilegeValue_(#Null, @Privilege, @TokenPriv\Privilege[0]\Luid)
      If FuncRet <> 0
        PChkRes = PrivilegeCheck_(hToken, @TokenPriv, @PrvEnb) And PrvEnb
      Else
        PrvEnb = #False
      EndIf
    Else
      PrvEnb = #False
    EndIf
  EndIf
  
  CloseHandle_(hToken)
  ProcedureReturn PrvEnb
  
EndProcedure

Procedure.b Enable_Privilege(priv.s)
  Protected TokenPriv.TOKEN_PRIVILEGES, token.i
  Protected hToken.i, ProcRet.b, FuncRet.i
  ProcRet = #False
  
  FuncRet = OpenProcessToken_(GetCurrentProcess_(), #TOKEN_ADJUST_PRIVILEGES, @hToken)
  If FuncRet <> #False
    TokenPriv\PrivilegeCount = 1
    FuncRet = LookupPrivilegeValue_(#Null, @priv, @TokenPriv\Privileges[0]\Luid)
    TokenPriv\Privileges[0]\Attributes = #SE_PRIVILEGE_ENABLED
    If FuncRet <> #False
      FuncRet = AdjustTokenPrivileges_(hToken, #False, @TokenPriv, SizeOf(TokenPriv), #Null, #Null)
      token = CloseHandle_(hToken)
      If FuncRet <> #False
        ProcRet = #True
      Else
        ProcRet = #False
      EndIf
    Else
      ProcRet = #False
    EndIf
  Else
    ProcRet = #False
  EndIf
  
  If token <> 0
    CloseHandle_(hToken)
  EndIf
  
  ProcedureReturn ProcRet 
EndProcedure

;pszPath = our created directory path
Procedure OpenDirectory(pszPath.s, bReadWrite.b)
  Protected  hDir.i
;;// Obtain backup/Restore privilege in case we don't have it
Enable_Privilege(#SE_RESTORE_NAME$)
Enable_Privilege(#SE_BACKUP_NAME$)
If IsPrivilegeEnabled(#SE_RESTORE_NAME$) = #True And IsPrivilegeEnabled(#SE_BACKUP_NAME$) = #True
  dwAccess.i = #GENERIC_READ | #GENERIC_WRITE
  hDir = CreateFile_(pszPath, dwAccess, 0, #Null, #OPEN_EXISTING, #FILE_FLAG_OPEN_REPARSE_POINT | #FILE_FLAG_BACKUP_SEMANTICS, #Null)
Else
  hDir = #False
EndIf
ProcedureReturn hDir
EndProcedure

; step 1: creating a directory if it does not exists. If the directory exists, it must be empty - you cannot associate a reparse point with a non-empty directory.
; Note that szTarget string must contain the path prefixed with the "non-parsed" prefix "\??\", and terminated with the backslash character, for example "\??\C:\Some Dir\". 
; szJunction = The path of the directory to be created, e.g. szJunction.s = ""\??\C:\Some Dir\"
CreateDirectory_(szJunction, #Null)

; step 2 = get a handle the the created directory 
hDir = OpenDirectory(szJunction, #True)
If hDir > 0
  *ReparseBuffer.REPARSE_MOUNTPOINT_DATA_BUFFER
  *ReparseBuffer = AllocateMemory(SizeOf(REPARSE_MOUNTPOINT_DATA_BUFFER) + #MAX_PATH * SizeOf(WCHAR))

  ; step 3 = where the magic happens
  ;; Prepare reparse point Data
  ;memset(buf, 0, SizeOf(buf));
  *ReparseBuffer\ReparseTag = #IO_REPARSE_TAG_MOUNT_POINT
  len.i = MultiByteToWideChar_(#CP_ACP, 0, szJunction, -1, *ReparseBuffer\ReparseTarget, #MAX_PATH)
  *ReparseBuffer\ReparseTargetMaximumLength = len * SizeOf(WCHAR)
  *ReparseBuffer\ReparseTargetLength = len * SizeOf(WCHAR)
  *ReparseBuffer\ReparseDataLength = *ReparseBuffer\ReparseTarget + 12
  ;;// Attach reparse point
  dwRet.i
  DeviceIoControl_(hDir, #FSCTL_SET_REPARSE_POINT, *ReparseBuffer, *ReparseBuffer\ReparseDataLength + #REPARSE_MOUNTPOINT_HEADER_SIZE, #Null, 0, @dwRet, #Null) ; this function call and parameters causes the reparse point to be set
Else
  Debug "can't get handle to directory"
EndIf

Code: Select all

#define FSCTL_SET_REPARSE_POINT         CTL_CODE(FILE_DEVICE_FILE_SYSTEM, 41, METHOD_BUFFERED, FILE_SPECIAL_ACCESS) // REPARSE_DATA_BUFFER