48 lines
1.6 KiB
Nim
48 lines
1.6 KiB
Nim
|
# Copyright (c) 2020-2022 Status Research & Development GmbH
|
||
|
|
# Licensed and distributed under either of
|
||
|
|
# * MIT license: http://opensource.org/licenses/MIT
|
||
|
|
# * Apache License, Version 2.0: http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
||
|
|
|
||
|
|
import std/[os, strutils]
|
||
|
|
import ../stew/io2, ../stew/results
|
||
|
|
|
||
|
|
proc lockFileFlags(path: string, flags: set[OpenFlags],
|
||
|
|
lockType: LockType): IoResult[void] =
|
||
|
|
let handle = ? openFile(path, flags)
|
||
|
|
let info = ? lockFile(handle, lockType)
|
||
|
|
? unlockFile(info)
|
||
|
|
? closeFile(handle)
|
||
|
|
ok()
|
||
|
|
|
||
|
|
when isMainModule:
|
||
|
|
if paramCount() != 1:
|
||
|
|
echo "Not enough parameters"
|
||
|
|
else:
|
||
|
|
const TestFlags = [
|
||
|
|
({OpenFlags.Read}, LockType.Shared),
|
||
|
|
({OpenFlags.Write}, LockType.Exclusive),
|
||
|
|
|
||
|
|
({OpenFlags.Read, OpenFlags.Write}, LockType.Shared),
|
||
|
|
({OpenFlags.Read, OpenFlags.Write}, LockType.Exclusive),
|
||
|
|
|
||
|
|
({OpenFlags.Read, OpenFlags.ShareRead}, LockType.Shared),
|
||
|
|
({OpenFlags.Write, OpenFlags.ShareWrite}, LockType.Exclusive),
|
||
|
|
|
||
|
|
({OpenFlags.Read, OpenFlags.Write,
|
||
|
|
OpenFlags.ShareRead, OpenFlags.ShareWrite}, LockType.Shared),
|
||
|
|
({OpenFlags.Read, OpenFlags.Write,
|
||
|
|
OpenFlags.ShareRead, OpenFlags.ShareWrite}, LockType.Exclusive),
|
||
|
|
]
|
||
|
|
let pathName = paramStr(1)
|
||
|
|
let response =
|
||
|
|
block:
|
||
|
|
var res: seq[string]
|
||
|
|
for test in TestFlags:
|
||
|
|
let
|
||
|
|
lres = lockFileFlags(pathName, test[0], test[1])
|
||
|
|
data = if lres.isOk(): "OK" else: "E" & $int(lres.error())
|
||
|
|
res.add(data)
|
||
|
|
res.join(":")
|
||
|
|
echo response
|