This repository has been archived by the owner on Dec 16, 2020. It is now read-only.
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
9 changed files
with
311 additions
and
51 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,115 @@ | ||
| #include <stdio.h> | ||
| #include <stdlib.h> | ||
| #include <ctype.h> | ||
| #include <inttypes.h> | ||
| #include <string.h> | ||
| #include <assert.h> | ||
| #ifdef WIN32 | ||
| # include <WinSock2.h> | ||
| # include <Windows.h> | ||
| #else | ||
| # include <unistd.h> | ||
| #endif | ||
|
|
||
| #include "cache.h" | ||
| #include "pcs/pcs_mem.h" | ||
|
|
||
| #ifdef WIN32 | ||
| #ifndef __MINGW32__ | ||
| # define lseek _lseek | ||
| # define fileno _fileno | ||
| # define fseeko _fseeki64 | ||
| # define ftello _ftelli64 | ||
| #endif | ||
| #endif | ||
|
|
||
| int cache_init(cathe_t *cache) | ||
| { | ||
| memset(cache, 0, sizeof(cathe_t)); | ||
| cache->blocks.prev = &cache->blocks; | ||
| cache->blocks.next = &cache->blocks; | ||
| return 0; | ||
| } | ||
|
|
||
| void cache_uninit(cathe_t *cache) | ||
| { | ||
| if (cache) { | ||
| block_t *block = cache->blocks.next, *tmp; | ||
| while (block != &cache->blocks) { | ||
| tmp = block; | ||
| block = block->next; | ||
| pcs_free(tmp->data); | ||
| pcs_free(tmp); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| int cache_reset(cathe_t *cache) | ||
| { | ||
| cache_uninit(cache); | ||
| cache->blocks.prev = &cache->blocks; | ||
| cache->blocks.next = &cache->blocks; | ||
| cache->total_size = 0; | ||
| return 0; | ||
| } | ||
|
|
||
| block_t *cache_newblock(curl_off_t start, char *data, size_t size) | ||
| { | ||
| block_t *block; | ||
| block = pcs_malloc(sizeof(block_t)); | ||
| if (!block) | ||
| return NULL; | ||
| memset(block, 0, sizeof(block_t)); | ||
| block->start = start; | ||
| block->size = size; | ||
| block->data = pcs_malloc(size); | ||
| if (!block->data) { | ||
| free(block); | ||
| return NULL; | ||
| } | ||
| memcpy(block->data, data, size); | ||
| return block; | ||
| } | ||
|
|
||
| int cache_addblock(cathe_t *cache, block_t *block) | ||
| { | ||
| block_t *pos = cache->blocks.next; | ||
| cache->total_size += block->size; | ||
| /* find position */ | ||
| while (pos != &cache->blocks) { | ||
| if (pos->start > block->start) | ||
| break; | ||
| pos = pos->next; | ||
| } | ||
| block->next = pos; | ||
| block->prev = pos->prev; | ||
| pos->prev->next = block; | ||
| pos->prev = block; | ||
| return 0; | ||
| } | ||
|
|
||
| int cache_add(cathe_t *cache, curl_off_t start, char *data, size_t size) | ||
| { | ||
| block_t *block, *pos = cache->blocks.next; | ||
| block = cache_newblock(start, data, size); | ||
| if (!block) | ||
| return -1; | ||
| return cache_addblock(cache, block); | ||
| } | ||
|
|
||
| int cache_flush(cathe_t *cache) | ||
| { | ||
| block_t *block = cache->blocks.next; | ||
| int r; | ||
| while (block != &cache->blocks) { | ||
| r = fseeko((cache->fp), block->start, SEEK_SET); | ||
| if (r) | ||
| return -1; | ||
| r = fwrite(block->data, 1, block->size, cache->fp); | ||
| if (r != block->size) | ||
| return -1; | ||
| block = block->next; | ||
| } | ||
| return 0; | ||
| } | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,40 @@ | ||
| #ifndef _PCS_SHELL_CAHCE_H | ||
| #define _PCS_SHELL_CAHCE_H | ||
|
|
||
| #include "pcs/pcs.h" | ||
|
|
||
| #define MAX_CACHE_SIZE (1 * 1024) /* 1MiB */ | ||
|
|
||
| #ifdef __cplusplus | ||
| extern "C" { | ||
| #endif | ||
|
|
||
| typedef struct block { | ||
| curl_off_t start; | ||
| size_t size; | ||
| char *data; | ||
| struct block *prev; | ||
| struct block *next; | ||
| } block_t; | ||
|
|
||
| typedef struct cache { | ||
| block_t blocks; | ||
| FILE *fp; | ||
| size_t total_size; | ||
| } cathe_t; | ||
|
|
||
| int cache_init(cathe_t *cache); | ||
|
|
||
| void cache_uninit(cathe_t *cache); | ||
|
|
||
| int cache_add(cathe_t *cache, curl_off_t start, char *data, size_t size); | ||
|
|
||
| int cache_flush(cathe_t *cache); | ||
|
|
||
| int cache_reset(cathe_t *cache); | ||
|
|
||
| #ifdef __cplusplus | ||
| } | ||
| #endif | ||
|
|
||
| #endif |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.