Method getMmapPtr

phxsqlclient/utils/leakybucket.cpp:38–92

}

void * MmapLeakyBucket::getMmapPtr(const char * filePath, uint32_t len, int * isNewFile) {
    *isNewFile = 0;

    int fd = open(filePath, O_RDWR);
    if (fd < 0) {
        if (ENOENT == errno) {
            fd = open(filePath, O_RDWR | O_CREAT, 0660);
            if (fd >= 0) {
                *isNewFile = 1;

                char buffer[1024] = { 0 };
                memset(buffer, 0, sizeof(buffer));
                for (int i = 0; i < (int) len; i += sizeof(buffer)) {
                    write(fd, buffer, sizeof(buffer));
                }

                ftruncate(fd, len);

                lseek(fd, 0, SEEK_SET);
            } else {
                //Log( COMM_LOG_ERR, "ERR: Create %s fail, errno %d, %s\n",
                //filePath, errno, strerror( errno ) );
            }
        } else {
            //Log( COMM_LOG_ERR, "ERR: Open %s fail, errno %d, %s\n",
            //filePath, errno, strerror( errno ) );
        }
    }

    void * ret = NULL;

    if (fd >= 0) {
        struct stat fileStat;
        if (0 == fstat(fd, &fileStat)) {
            if (fileStat.st_size == (int) len) {
                ret = mmap(0, len, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
                if (MAP_FAILED == ret) {
                    //Log( COMM_LOG_ERR, "ERR: mmap %s fail, errno %d, %s\n",
                    //		filePath, errno, strerror( errno ) );
                    ret = NULL;
                }
            } else {
                //Log( COMM_LOG_ERR, "ERR: invalid file size, real %lld, except %i",
                //		(long long)fileStat.st_size, len );
            }
        } else {
            //Log( COMM_LOG_ERR, "ERR: Stat %s fail, errno %d, %s\n",
            //		filePath, errno, strerror( errno ) );
        }

        close(fd);
    }

    return ret;
}

void MmapLeakyBucket::freeMmapPtr(void * ptr, uint32_t len) {
    if (0 != munmap(ptr, len)) {