调整picid和infoid的校验

chenshijun

2020-03-31 6d9477b720f2ba12b85b3007e2a07777eca9e01b

 shmwrap.go

@@ -11,13 +11,13 @@
func NewBlock(size int, key int) ([]byte, int, error) {
    id, err := shm.Get(key, size, shm.IPC_CREAT|0666)
    if err != nil || id == -1 {
        fmt.Println("NewBlock Get:", key, err)
        fmt.Printf("NewBlock Get:%x, %v\n", key, err)
        return nil, -1, err
    }

    data, err2 := shm.At(id, 0, 0)
    if err2 != nil {
        fmt.Println("NewBlock At:", key, err2)
        fmt.Printf("NewBlock At:%x, %v\n", key, err2)
        return nil, -1, err2
    }

@@ -28,13 +28,13 @@
func AttachBlock(key int) ([]byte, int, error) {
    id, err := shm.Get(key, 0, 0)
    if err != nil || id == -1 { //no exist
        fmt.Println("AttachBlock Get:", key, err)
        fmt.Printf("AttachBlock Get:%x, %v\n", key, err)
        return nil, -1, err
    }

    data, err2 := shm.At(id, 0, 0)
    if err2 != nil {
        fmt.Println("AttachBlock At:", key, err2)
        fmt.Printf("AttachBlock Get:%x, %v\n", key, err2)
        return nil, -1, err2
    }

@@ -167,3 +167,27 @@
        return shm.Rm(id)
    }
}

func ShmAttachNum(shmId int) (int,error) {
    var idDs shm.IdDs

    _, err := shm.Ctl(shmId, shm.IPC_STAT, &idDs)
    if err != nil {
        return 0, err
    }

    return int(idDs.Nattch), nil
}

func ForceRemoveShm(data []byte, shmId int) error {
    count, err := ShmAttachNum(shmId)
    if err != nil{
        return err
    }else{
        for i:=0; i < count; i++{
            Detach(data)
        }
    }
    RemoveShmID(shmId)
    return nil
}