fusego/fuseops/common_op.go

// Copyright 2015 Google Inc. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package fuseops

import (
	"fmt"
	"log"
	"reflect"
	"strings"
	"unsafe"

	"github.com/jacobsa/fuse/internal/fusekernel"
	"github.com/jacobsa/fuse/internal/fuseshim"
	"github.com/jacobsa/reqtrace"
	"golang.org/x/net/context"
)

// An interface that all ops inside which commonOp is embedded must
// implement.
type internalOp interface {
	Op

	// Create a response message for the kernel, with leading pading for a
	// fusekernel.OutHeader struct.
	kernelResponse() []byte
}

// A function that sends a reply message back to the kernel for the request
// with the given fuse unique ID. The error argument is for informational
// purposes only; the error to hand to the kernel is encoded in the message.
type replyFunc func(uint64, []byte, error) error

// A helper for embedding common behavior.
type commonOp struct {
	// The context exposed to the user.
	ctx context.Context

	// The op in which this struct is embedded.
	op internalOp

	// The fuse unique ID of this request, as assigned by the kernel.
	fuseID uint64

	// A function that can be used to send a reply to the kernel.
	sendReply replyFunc

	// A function that can be used to log debug information about the op. The
	// first argument is a call depth.
	//
	// May be nil.
	debugLog func(int, string, ...interface{})

	// A logger to be used for logging exceptional errors.
	//
	// May be nil.
	errorLogger *log.Logger
}

func (o *commonOp) ShortDesc() (desc string) {
	v := reflect.ValueOf(o.op)
	opName := v.Type().String()

	// Attempt to better handle the usual case: a string that looks like
	// "*fuseops.GetInodeAttributesOp".
	const prefix = "*fuseops."
	const suffix = "Op"
	if strings.HasPrefix(opName, prefix) && strings.HasSuffix(opName, suffix) {
		opName = opName[len(prefix) : len(opName)-len(suffix)]
	}

	desc = opName

	// Include the inode number to which the op applies, if possible.
	if f := v.Elem().FieldByName("Inode"); f.IsValid() {
		desc = fmt.Sprintf("%s(inode=%v)", desc, f.Interface())
	}

	return
}

func (o *commonOp) init(
	ctx context.Context,
	op internalOp,
	fuseID uint64,
	sendReply replyFunc,
	debugLog func(int, string, ...interface{}),
	errorLogger *log.Logger) {
	// Initialize basic fields.
	o.ctx = ctx
	o.op = op
	o.fuseID = fuseID
	o.sendReply = sendReply
	o.debugLog = debugLog
	o.errorLogger = errorLogger

	// Set up a trace span for this op.
	var reportForTrace reqtrace.ReportFunc
	o.ctx, reportForTrace = reqtrace.StartSpan(o.ctx, o.op.ShortDesc())

	// When the op is finished, report to both reqtrace and the connection.
	prevSendReply := o.sendReply
	o.sendReply = func(fuseID uint64, msg []byte, opErr error) (err error) {
		reportForTrace(opErr)
		err = prevSendReply(fuseID, msg, opErr)
		return
	}
}

func (o *commonOp) Context() context.Context {
	return o.ctx
}

func (o *commonOp) Logf(format string, v ...interface{}) {
	if o.debugLog == nil {
		return
	}

	const calldepth = 2
	o.debugLog(calldepth, format, v...)
}

func (o *commonOp) Respond(err error) {
	// If successful, we ask the op for an appopriate response to the kernel, and
	// it is responsible for leaving room for the fusekernel.OutHeader struct.
	// Otherwise, create our own.
	var msg []byte
	if err == nil {
		msg = o.op.kernelResponse()
	} else {
		msg = fuseshim.NewBuffer(0)
	}

	// Fill in the header.
	h := (*fusekernel.OutHeader)(unsafe.Pointer(&msg[0]))
	h.Unique = o.fuseID
	h.Len = uint32(len(msg))
	if err != nil {
		errno := fuseshim.EIO
		if ferr, ok := err.(fuseshim.ErrorNumber); ok {
			errno = ferr.Errno()
		}

		h.Error = -int32(errno)
	}

	// Log the error, if any.
	if err != nil {
		if o.debugLog != nil {
			o.Logf(
				"-> (%s) error: %v",
				o.op.ShortDesc(),
				err)
		}

		if o.errorLogger != nil {
			o.errorLogger.Printf(
				"(%s) error: %v",
				o.op.ShortDesc(),
				err)
		}
	}

	// Reply.
	replyErr := o.sendReply(o.fuseID, msg, err)
	if replyErr != nil && o.errorLogger != nil {
		o.errorLogger.Printf("Error from sendReply: %v", replyErr)
	}
}