ebiten/internal/glfw/load_windows.go

// Copyright 2018 The Ebiten Authors
//
// 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 glfw

import (
	"bytes"
	"compress/gzip"
	"fmt"
	"io"
	"io/ioutil"
	"os"
	"unsafe"

	"golang.org/x/sys/windows"
)

type dll struct {
	d     *windows.LazyDLL
	path  string
	procs map[string]*windows.LazyProc
}

func (d *dll) call(name string, args ...uintptr) uintptr {
	if d.procs == nil {
		d.procs = map[string]*windows.LazyProc{}
	}
	if _, ok := d.procs[name]; !ok {
		d.procs[name] = d.d.NewProc(name)
	}
	r, _, err := d.procs[name].Call(args...)
	if err != nil && err.(windows.Errno) != 0 {
		panic(err)
	}
	return r
}

func createTempDLL(content io.Reader) (string, error) {
	f, err := ioutil.TempFile("", "glfw.*.dll")
	if err != nil {
		return "", err
	}
	defer f.Close()

	fn := f.Name()

	if _, err := io.Copy(f, content); err != nil {
		return "", err
	}

	return fn, nil
}

func loadDLL() (*dll, error) {
	f, err := gzip.NewReader(bytes.NewReader(glfwDLLCompressed))
	if err != nil {
		return nil, err
	}
	defer f.Close()

	fn, err := createTempDLL(f)
	if err != nil {
		return nil, err
	}

	return &dll{
		d:    windows.NewLazyDLL(fn),
		path: fn,
	}, nil
}

func (d *dll) unload() error {
	if err := windows.FreeLibrary(windows.Handle(d.d.Handle())); err != nil {
		return err
	}
	if err := os.Remove(d.path); err != nil {
		return err
	}
	return nil
}

func uintptrToString(ptr uintptr) string {
	var bs []byte
	for {
		b := *(*byte)(unsafe.Pointer(ptr))
		if b == 0 {
			break
		}
		bs = append(bs, b)
		ptr++
	}
	return string(bs)
}

type glfwError struct {
	code ErrorCode
	desc string
}

func (e *glfwError) Error() string {
	return fmt.Sprintf("glfw: %s: %s", e.code.String(), e.desc)
}

var lastErr = make(chan *glfwError, 1)

func fetchError() error {
	select {
	case err := <-lastErr:
		return err
	default:
		return nil
	}
}

func panicError() {
	if err := acceptError(); err != nil {
		panic(err)
	}
}

func flushErrors() {
	if err := fetchError(); err != nil {
		panic(fmt.Sprintf("glfw: uncaught error: %s", err))
	}
}

func acceptError(codes ...ErrorCode) error {
	err := fetchError()
	if err == nil {
		return nil
	}
	for _, c := range codes {
		if err.(*glfwError).code == c {
			return nil
		}
	}
	if err.(*glfwError).code == PlatformError {
		// PlatformError is not handled here (See github.com/go-gl/glfw's implementation).
		// TODO: Should we log this error?
		return nil
	}
	return err
}

func goGLFWErrorCallback(code uintptr, desc uintptr) uintptr {
	flushErrors()
	err := &glfwError{
		code: ErrorCode(code),
		desc: uintptrToString(desc),
	}
	select {
	case lastErr <- err:
	default:
		panic(fmt.Sprintf("glfw: uncaught error: %s", err))
	}
	return 0
}

var glfwDLL *dll

func init() {
	dll, err := loadDLL()
	if err != nil {
		panic(err)
	}
	glfwDLL = dll

	glfwDLL.call("glfwSetErrorCallback", windows.NewCallbackCDecl(goGLFWErrorCallback))
}