github.com/pawelgaczynski/giouring@v0.0.0-20230826085535-69588b89acb9/probe.go (about)

     1  // MIT License
     2  //
     3  // Copyright (c) 2023 Paweł Gaczyński
     4  //
     5  // Permission is hereby granted, free of charge, to any person obtaining a
     6  // copy of this software and associated documentation files (the
     7  // "Software"), to deal in the Software without restriction, including
     8  // without limitation the rights to use, copy, modify, merge, publish,
     9  // distribute, sublicense, and/or sell copies of the Software, and to
    10  // permit persons to whom the Software is furnished to do so, subject to
    11  // the following conditions:
    12  //
    13  // The above copyright notice and this permission notice shall be included
    14  // in all copies or substantial portions of the Software.
    15  //
    16  // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
    17  // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
    18  // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
    19  // IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
    20  // CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
    21  // TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
    22  // SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
    23  
    24  package giouring
    25  
    26  const (
    27  	probeOpsSize = 256
    28  	// int(OpLast + 1)
    29  )
    30  
    31  // liburing: io_uring_get_probe_ring
    32  func (ring *Ring) GetProbeRing() (*Probe, error) {
    33  	probe := &Probe{}
    34  	_, err := ring.RegisterProbe(probe, probeOpsSize)
    35  	if err != nil {
    36  		return nil, err
    37  	}
    38  
    39  	return probe, nil
    40  }
    41  
    42  const probeEntries = 2
    43  
    44  // liburing: io_uring_get_probe - https://manpages.debian.org/unstable/liburing-dev/io_uring_get_probe.3.en.html
    45  func GetProbe() (*Probe, error) {
    46  	ring, err := CreateRing(probeEntries)
    47  	if err != nil {
    48  		return nil, err
    49  	}
    50  
    51  	probe, err := ring.GetProbeRing()
    52  	if err != nil {
    53  		return nil, err
    54  	}
    55  	ring.QueueExit()
    56  
    57  	return probe, nil
    58  }
    59  
    60  // liburing: io_uring_opcode_supported - https://manpages.debian.org/unstable/liburing-dev/io_uring_opcode_supported.3.en.html
    61  func (p Probe) IsSupported(op uint8) bool {
    62  	for i := uint8(0); i < p.OpsLen; i++ {
    63  		if p.Ops[i].Op != op {
    64  			continue
    65  		}
    66  
    67  		return p.Ops[i].Flags&opSupported != 0
    68  	}
    69  
    70  	return false
    71  }