src/lockdir.go

   1 // NNCP -- Node to Node copy, utilities for store-and-forward data exchange
   2 // Copyright (C) 2016-2024 Sergey Matveev <stargrave@stargrave.org>
   3 //
   4 // This program is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU General Public License as published by
   6 // the Free Software Foundation, version 3 of the License.
   7 //
   8 // This program is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 // GNU General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU General Public License
  14 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15
  16 package nncp
  17
  18 import (
  19         "os"
  20         "path/filepath"
  21
  22         "golang.org/x/sys/unix"
  23 )
  24
  25 func (ctx *Ctx) LockDir(nodeId *NodeId, lockCtx string) (*os.File, error) {
  26         if err := ctx.ensureRxDir(nodeId); err != nil {
  27                 return nil, err
  28         }
  29         lockPath := filepath.Join(ctx.Spool, nodeId.String(), lockCtx) + ".lock"
  30         dirLock, err := os.OpenFile(
  31                 lockPath,
  32                 os.O_CREATE|os.O_WRONLY,
  33                 os.FileMode(0666),
  34         )
  35         if err != nil {
  36                 ctx.LogE("lockdir-open", LEs{{"Path", lockPath}}, err, func(les LEs) string {
  37                         return "Locking directory: opening " + lockPath
  38                 })
  39                 return nil, err
  40         }
  41         err = unix.Flock(int(dirLock.Fd()), unix.LOCK_EX|unix.LOCK_NB)
  42         if err != nil {
  43                 ctx.LogE("lockdir-flock", LEs{{"Path", lockPath}}, err, func(les LEs) string {
  44                         return "Locking directory: locking " + lockPath
  45                 })
  46                 dirLock.Close()
  47                 return nil, err
  48         }
  49         return dirLock, nil
  50 }
  51
  52 func (ctx *Ctx) UnlockDir(fd *os.File) {
  53         if fd != nil {
  54                 unix.Flock(int(fd.Fd()), unix.LOCK_UN)
  55                 fd.Close()
  56         }
  57 }