diff options
-rw-r--r-- | src/cachestorenids.h | 7 | ||||
-rw-r--r-- | src/file.cpp | 10 | ||||
-rw-r--r-- | src/nids.cpp | 12 | ||||
-rw-r--r-- | src/nidsstream.cpp | 14 |
4 files changed, 25 insertions, 18 deletions
diff --git a/src/cachestorenids.h b/src/cachestorenids.h index bd0fcc7..9288008 100644 --- a/src/cachestorenids.h +++ b/src/cachestorenids.h | |||
@@ -48,12 +48,6 @@ namespace Bu | |||
48 | NidsStream ns = nStore.openStream( 0 ); | 48 | NidsStream ns = nStore.openStream( 0 ); |
49 | Bu::Archive ar( ns, Bu::Archive::save ); | 49 | Bu::Archive ar( ns, Bu::Archive::save ); |
50 | ar << hId; | 50 | ar << hId; |
51 | |||
52 | Bu::FString sName; | ||
53 | sName.format("hash-%d.%02d", time( NULL ), iCnt++ ); | ||
54 | Bu::File sTmp( sName, Bu::File::Write|Bu::File::Create ); | ||
55 | Bu::Archive artmp( sTmp, Bu::Archive::save ); | ||
56 | artmp << hId; | ||
57 | } | 51 | } |
58 | 52 | ||
59 | virtual obtype *load( const keytype &key ) | 53 | virtual obtype *load( const keytype &key ) |
@@ -80,7 +74,6 @@ namespace Bu | |||
80 | keytype key = __cacheGetKey<obtype,keytype>( pSrc ); | 74 | keytype key = __cacheGetKey<obtype,keytype>( pSrc ); |
81 | int iStream = nStore.createStream(); | 75 | int iStream = nStore.createStream(); |
82 | hId.insert( key, iStream ); | 76 | hId.insert( key, iStream ); |
83 | printf("Creating stream: %d\n", iStream ); | ||
84 | NidsStream ns = nStore.openStream( iStream ); | 77 | NidsStream ns = nStore.openStream( iStream ); |
85 | Bu::Archive ar( ns, Bu::Archive::save ); | 78 | Bu::Archive ar( ns, Bu::Archive::save ); |
86 | obtype *pOb = new obtype(); | 79 | obtype *pOb = new obtype(); |
diff --git a/src/file.cpp b/src/file.cpp index 20ff5c9..aa0ea32 100644 --- a/src/file.cpp +++ b/src/file.cpp | |||
@@ -113,7 +113,10 @@ bool Bu::File::canRead() | |||
113 | #ifdef WIN32 | 113 | #ifdef WIN32 |
114 | return true; | 114 | return true; |
115 | #else | 115 | #else |
116 | return (fcntl( fd, F_GETFL, 0 )&O_RDONLY) == O_RDONLY; | 116 | int iMode = fcntl( fd, F_GETFL, 0 )&O_ACCMODE; |
117 | if( iMode == O_RDONLY || iMode == O_RDWR ) | ||
118 | return true; | ||
119 | return false; | ||
117 | #endif | 120 | #endif |
118 | } | 121 | } |
119 | 122 | ||
@@ -122,7 +125,10 @@ bool Bu::File::canWrite() | |||
122 | #ifdef WIN32 | 125 | #ifdef WIN32 |
123 | return true; | 126 | return true; |
124 | #else | 127 | #else |
125 | return (fcntl( fd, F_GETFL, 0 )&O_WRONLY) == O_WRONLY; | 128 | int iMode = fcntl( fd, F_GETFL, 0 )&O_ACCMODE; |
129 | if( iMode == O_WRONLY || iMode == O_RDWR ) | ||
130 | return true; | ||
131 | return false; | ||
126 | #endif | 132 | #endif |
127 | } | 133 | } |
128 | 134 | ||
diff --git a/src/nids.cpp b/src/nids.cpp index 00d5f2b..23c11b0 100644 --- a/src/nids.cpp +++ b/src/nids.cpp | |||
@@ -114,8 +114,8 @@ void Bu::Nids::initialize( int iBlockSize, int iPreAllocate ) | |||
114 | 114 | ||
115 | void Bu::Nids::updateHeader() | 115 | void Bu::Nids::updateHeader() |
116 | { | 116 | { |
117 | // if( !sStore.canWrite() ) | 117 | if( !sStore.canWrite() ) |
118 | // return; | 118 | return; |
119 | sStore.setPos( 10 ); // Skip the magic number, version, bpi, block size | 119 | sStore.setPos( 10 ); // Skip the magic number, version, bpi, block size |
120 | sStore.write( &iBlocks, 4 ); | 120 | sStore.write( &iBlocks, 4 ); |
121 | sStore.write( &iUsed, 4 ); | 121 | sStore.write( &iUsed, 4 ); |
@@ -222,8 +222,8 @@ void Bu::Nids::setBlock( uint32_t uIndex, Bu::Nids::Block *pBlock ) | |||
222 | 222 | ||
223 | void Bu::Nids::updateStreamSize( uint32_t uIndex, uint32_t uSize ) | 223 | void Bu::Nids::updateStreamSize( uint32_t uIndex, uint32_t uSize ) |
224 | { | 224 | { |
225 | // if( !sStore.canWrite() ) | 225 | if( !sStore.canWrite() ) |
226 | // return; | 226 | return; |
227 | sStore.setPos( iBlockStart + (iBlockSize*uIndex)+4*3 ); | 227 | sStore.setPos( iBlockStart + (iBlockSize*uIndex)+4*3 ); |
228 | sStore.write( &uSize, 4 ); | 228 | sStore.write( &uSize, 4 ); |
229 | } | 229 | } |
@@ -240,8 +240,8 @@ uint32_t Bu::Nids::getNextBlock( uint32_t uIndex, | |||
240 | sStore.setPos( iBlockStart + (iBlockSize*uIndex)+1*4 ); | 240 | sStore.setPos( iBlockStart + (iBlockSize*uIndex)+1*4 ); |
241 | sStore.write( &uNew, 4 ); | 241 | sStore.write( &uNew, 4 ); |
242 | getBlock( uNew, pBlock ); | 242 | getBlock( uNew, pBlock ); |
243 | printf("Allocated new block (%u) for stream %u.\n", | 243 | //printf("Allocated new block (%u) for stream %u.\n", |
244 | uNew, pBlock->uFirstBlock ); | 244 | // uNew, pBlock->uFirstBlock ); |
245 | } | 245 | } |
246 | else | 246 | else |
247 | { | 247 | { |
diff --git a/src/nidsstream.cpp b/src/nidsstream.cpp index fff73f6..31e1293 100644 --- a/src/nidsstream.cpp +++ b/src/nidsstream.cpp | |||
@@ -55,7 +55,7 @@ size_t Bu::NidsStream::read( void *pBuf, size_t nBytes ) | |||
55 | if( iRead > pCurBlock->uBytesUsed-(uPos%uBlockSize) ) | 55 | if( iRead > pCurBlock->uBytesUsed-(uPos%uBlockSize) ) |
56 | iRead = pCurBlock->uBytesUsed-(uPos%uBlockSize); | 56 | iRead = pCurBlock->uBytesUsed-(uPos%uBlockSize); |
57 | memcpy( pBuf, pCurBlock->pData+(uPos%uBlockSize), iRead ); | 57 | memcpy( pBuf, pCurBlock->pData+(uPos%uBlockSize), iRead ); |
58 | //printf("buffill: %ub, %u-%u/%u -> %d-%d/%d (a:%u)", | 58 | //printf("read buffill: %ub, %u-%u/%u -> %d-%d/%d (a:%u)", |
59 | // iRead, uPos, uPos+iRead-1, uSize, 0, iRead-1, nBytes, uCurBlock ); | 59 | // iRead, uPos, uPos+iRead-1, uSize, 0, iRead-1, nBytes, uCurBlock ); |
60 | uPos += iRead; | 60 | uPos += iRead; |
61 | //printf(" -- %u\n", uPos%uBlockSize ); | 61 | //printf(" -- %u\n", uPos%uBlockSize ); |
@@ -80,7 +80,7 @@ size_t Bu::NidsStream::read( void *pBuf, size_t nBytes ) | |||
80 | iRead = pCurBlock->uBytesUsed-(uPos%uBlockSize); | 80 | iRead = pCurBlock->uBytesUsed-(uPos%uBlockSize); |
81 | memcpy( ((char *)pBuf)+nTotal, | 81 | memcpy( ((char *)pBuf)+nTotal, |
82 | pCurBlock->pData+(uPos%uBlockSize), iRead ); | 82 | pCurBlock->pData+(uPos%uBlockSize), iRead ); |
83 | //printf("buffill: %ub, %u-%u/%u -> %d-%d/%d (b:%u)\n", | 83 | //printf(" read buffill: %ub, %u-%u/%u -> %d-%d/%d (b:%u)\n", |
84 | // iRead, uPos, uPos+iRead-1, uSize, | 84 | // iRead, uPos, uPos+iRead-1, uSize, |
85 | // nTotal, nTotal+nBytes-1, nBytes, uCurBlock ); | 85 | // nTotal, nTotal+nBytes-1, nBytes, uCurBlock ); |
86 | uPos += iRead; | 86 | uPos += iRead; |
@@ -111,6 +111,10 @@ size_t Bu::NidsStream::write( const void *pBuf, size_t nBytes ) | |||
111 | { | 111 | { |
112 | //printf("wa: %u:%u:%u:%u -> ", uPos, uPos%uBlockSize, uSize, pCurBlock->uBytesUsed ); | 112 | //printf("wa: %u:%u:%u:%u -> ", uPos, uPos%uBlockSize, uSize, pCurBlock->uBytesUsed ); |
113 | memcpy( pCurBlock->pData+(uPos%uBlockSize), pBuf, nBytes ); | 113 | memcpy( pCurBlock->pData+(uPos%uBlockSize), pBuf, nBytes ); |
114 | //printf("write buffill: %ub, %u-%u/%u -> %d-%d/%d (a:%u:%u)\n", | ||
115 | // nBytes, 0, nBytes-1, nBytes, | ||
116 | // uPos, uPos+nBytes-1, uSize, uCurBlock, | ||
117 | // pCurBlock->uBytesUsed ); | ||
114 | if( (uPos%uBlockSize)+nBytes > pCurBlock->uBytesUsed ) | 118 | if( (uPos%uBlockSize)+nBytes > pCurBlock->uBytesUsed ) |
115 | pCurBlock->uBytesUsed = (uPos%uBlockSize)+nBytes; | 119 | pCurBlock->uBytesUsed = (uPos%uBlockSize)+nBytes; |
116 | rNids.setBlock( uCurBlock, pCurBlock ); | 120 | rNids.setBlock( uCurBlock, pCurBlock ); |
@@ -132,6 +136,9 @@ size_t Bu::NidsStream::write( const void *pBuf, size_t nBytes ) | |||
132 | uNow = nBytes; | 136 | uNow = nBytes; |
133 | memcpy( pCurBlock->pData+(uPos%uBlockSize), | 137 | memcpy( pCurBlock->pData+(uPos%uBlockSize), |
134 | &((char *)pBuf)[nTotal], uNow ); | 138 | &((char *)pBuf)[nTotal], uNow ); |
139 | //printf("write buffill: %ub, %u-%u/%u -> %d-%d/%d (b:%u:%u)\n", | ||
140 | // uNow, nTotal, nTotal+uNow-1, nBytes, | ||
141 | // uPos, uPos+uNow-1, uSize, uCurBlock, pCurBlock->uBytesUsed ); | ||
135 | if( (uPos%uBlockSize)+uNow > pCurBlock->uBytesUsed ) | 142 | if( (uPos%uBlockSize)+uNow > pCurBlock->uBytesUsed ) |
136 | pCurBlock->uBytesUsed = (uPos%uBlockSize)+uNow; | 143 | pCurBlock->uBytesUsed = (uPos%uBlockSize)+uNow; |
137 | rNids.setBlock( uCurBlock, pCurBlock ); | 144 | rNids.setBlock( uCurBlock, pCurBlock ); |
@@ -142,7 +149,8 @@ size_t Bu::NidsStream::write( const void *pBuf, size_t nBytes ) | |||
142 | nBytes -= uNow; | 149 | nBytes -= uNow; |
143 | //printf("wb: block %u = %ub (%ub total)\n", | 150 | //printf("wb: block %u = %ub (%ub total)\n", |
144 | // uCurBlock, pCurBlock->uBytesUsed, uSize ); | 151 | // uCurBlock, pCurBlock->uBytesUsed, uSize ); |
145 | if( pCurBlock->uBytesUsed == uBlockSize ) | 152 | //if( pCurBlock->uBytesUsed == uBlockSize ) |
153 | if( uPos%uBlockSize == 0 ) | ||
146 | uCurBlock = rNids.getNextBlock( uCurBlock, pCurBlock ); | 154 | uCurBlock = rNids.getNextBlock( uCurBlock, pCurBlock ); |
147 | if( nBytes == 0 ) | 155 | if( nBytes == 0 ) |
148 | return nTotal; | 156 | return nTotal; |