61 lines
1.7 KiB
Go
61 lines
1.7 KiB
Go
/**
|
|
* Copyright 2023 ByteDance Inc.
|
|
*
|
|
* 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 loader
|
|
|
|
import (
|
|
"sync/atomic"
|
|
"unsafe"
|
|
_ `unsafe`
|
|
)
|
|
|
|
//go:linkname lastmoduledatap runtime.lastmoduledatap
|
|
//goland:noinspection GoUnusedGlobalVariable
|
|
var lastmoduledatap *moduledata
|
|
|
|
func registerModule(mod *moduledata) {
|
|
registerModuleLockFree(&lastmoduledatap, mod)
|
|
}
|
|
|
|
//go:linkname moduledataverify1 runtime.moduledataverify1
|
|
func moduledataverify1(_ *moduledata)
|
|
|
|
func registerModuleLockFree(tail **moduledata, mod *moduledata) {
|
|
for {
|
|
oldTail := loadModule(tail)
|
|
if casModule(tail, oldTail, mod) {
|
|
storeModule(&oldTail.next, mod)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
func loadModule(p **moduledata) *moduledata {
|
|
return (*moduledata)(atomic.LoadPointer((*unsafe.Pointer)(unsafe.Pointer(p))))
|
|
}
|
|
|
|
func storeModule(p **moduledata, value *moduledata) {
|
|
atomic.StorePointer((*unsafe.Pointer)(unsafe.Pointer(p)), unsafe.Pointer(value))
|
|
}
|
|
|
|
func casModule(p **moduledata, oldValue *moduledata, newValue *moduledata) bool {
|
|
return atomic.CompareAndSwapPointer(
|
|
(*unsafe.Pointer)(unsafe.Pointer(p)),
|
|
unsafe.Pointer(oldValue),
|
|
unsafe.Pointer(newValue),
|
|
)
|
|
}
|