2 * Copyright (c) 1990, 1993, 1994
3 * The Regents of the University of California. All rights reserved.
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of the University nor the names of its contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * @(#)rec_get.c 8.9 (Berkeley) 8/18/94
30 * $FreeBSD: src/lib/libc/db/recno/rec_get.c,v 1.2.6.2 2001/01/02 09:11:10 obrien Exp $
31 * $DragonFly: src/lib/libc/db/recno/rec_get.c,v 1.4 2005/11/12 23:01:55 swildner Exp $
34 #include <sys/types.h>
47 * __REC_GET -- Get a record from the btree.
50 * dbp: pointer to access method
52 * data: data to return
53 * flag: currently unused
56 * RET_ERROR, RET_SUCCESS and RET_SPECIAL if the key not found.
59 __rec_get(const DB
*dbp
, const DBT
*key
, DBT
*data
, u_int flags
)
68 /* Toss any page pinned across calls. */
69 if (t
->bt_pinned
!= NULL
) {
70 mpool_put(t
->bt_mp
, t
->bt_pinned
, 0);
74 /* Get currently doesn't take any flags, and keys of 0 are illegal. */
75 if (flags
|| (nrec
= *(recno_t
*)key
->data
) == 0) {
81 * If we haven't seen this record yet, try to find it in the
84 if (nrec
> t
->bt_nrecs
) {
85 if (F_ISSET(t
, R_EOF
| R_INMEM
))
87 if ((status
= t
->bt_irec(t
, nrec
)) != RET_SUCCESS
)
92 if ((e
= __rec_search(t
, nrec
, SEARCH
)) == NULL
)
95 status
= __rec_ret(t
, e
, 0, NULL
, data
);
96 if (F_ISSET(t
, B_DB_LOCK
))
97 mpool_put(t
->bt_mp
, e
->page
, 0);
99 t
->bt_pinned
= e
->page
;
104 * __REC_FPIPE -- Get fixed length records from a pipe.
108 * cnt: records to read
111 * RET_ERROR, RET_SUCCESS
114 __rec_fpipe(BTREE
*t
, recno_t top
)
122 if (t
->bt_rdata
.size
< t
->bt_reclen
) {
123 t
->bt_rdata
.data
= t
->bt_rdata
.data
== NULL
?
124 malloc(t
->bt_reclen
) :
125 reallocf(t
->bt_rdata
.data
, t
->bt_reclen
);
126 if (t
->bt_rdata
.data
== NULL
)
128 t
->bt_rdata
.size
= t
->bt_reclen
;
130 data
.data
= t
->bt_rdata
.data
;
131 data
.size
= t
->bt_reclen
;
133 for (nrec
= t
->bt_nrecs
; nrec
< top
;) {
135 for (p
= t
->bt_rdata
.data
;; *p
++ = ch
)
136 if ((ch
= getc(t
->bt_rfp
)) == EOF
|| !--len
) {
140 memset(p
, t
->bt_bval
, len
);
142 nrec
, &data
, 0) != RET_SUCCESS
)
152 return (RET_SPECIAL
);
154 return (RET_SUCCESS
);
158 * __REC_VPIPE -- Get variable length records from a pipe.
162 * cnt: records to read
165 * RET_ERROR, RET_SUCCESS
168 __rec_vpipe(BTREE
*t
, recno_t top
)
178 for (nrec
= t
->bt_nrecs
; nrec
< top
; ++nrec
) {
179 for (p
= t
->bt_rdata
.data
,
180 sz
= t
->bt_rdata
.size
;; *p
++ = ch
, --sz
) {
181 if ((ch
= getc(t
->bt_rfp
)) == EOF
|| ch
== bval
) {
182 data
.data
= t
->bt_rdata
.data
;
183 data
.size
= p
- (u_char
*)t
->bt_rdata
.data
;
184 if (ch
== EOF
&& data
.size
== 0)
186 if (__rec_iput(t
, nrec
, &data
, 0)
192 len
= p
- (u_char
*)t
->bt_rdata
.data
;
193 t
->bt_rdata
.size
+= (sz
= 256);
194 t
->bt_rdata
.data
= t
->bt_rdata
.data
== NULL
?
195 malloc(t
->bt_rdata
.size
) :
196 reallocf(t
->bt_rdata
.data
, t
->bt_rdata
.size
);
197 if (t
->bt_rdata
.data
== NULL
)
199 p
= (u_char
*)t
->bt_rdata
.data
+ len
;
207 return (RET_SPECIAL
);
209 return (RET_SUCCESS
);
213 * __REC_FMAP -- Get fixed length records from a file.
217 * cnt: records to read
220 * RET_ERROR, RET_SUCCESS
223 __rec_fmap(BTREE
*t
, recno_t top
)
230 if (t
->bt_rdata
.size
< t
->bt_reclen
) {
231 t
->bt_rdata
.data
= t
->bt_rdata
.data
== NULL
?
232 malloc(t
->bt_reclen
) :
233 reallocf(t
->bt_rdata
.data
, t
->bt_reclen
);
234 if (t
->bt_rdata
.data
== NULL
)
236 t
->bt_rdata
.size
= t
->bt_reclen
;
238 data
.data
= t
->bt_rdata
.data
;
239 data
.size
= t
->bt_reclen
;
241 sp
= (u_char
*)t
->bt_cmap
;
242 ep
= (u_char
*)t
->bt_emap
;
243 for (nrec
= t
->bt_nrecs
; nrec
< top
; ++nrec
) {
246 return (RET_SPECIAL
);
249 for (p
= t
->bt_rdata
.data
;
250 sp
< ep
&& len
> 0; *p
++ = *sp
++, --len
);
252 memset(p
, t
->bt_bval
, len
);
253 if (__rec_iput(t
, nrec
, &data
, 0) != RET_SUCCESS
)
256 t
->bt_cmap
= (caddr_t
)sp
;
257 return (RET_SUCCESS
);
261 * __REC_VMAP -- Get variable length records from a file.
265 * cnt: records to read
268 * RET_ERROR, RET_SUCCESS
271 __rec_vmap(BTREE
*t
, recno_t top
)
278 sp
= (u_char
*)t
->bt_cmap
;
279 ep
= (u_char
*)t
->bt_emap
;
282 for (nrec
= t
->bt_nrecs
; nrec
< top
; ++nrec
) {
285 return (RET_SPECIAL
);
287 for (data
.data
= sp
; sp
< ep
&& *sp
!= bval
; ++sp
);
288 data
.size
= sp
- (u_char
*)data
.data
;
289 if (__rec_iput(t
, nrec
, &data
, 0) != RET_SUCCESS
)
293 t
->bt_cmap
= (caddr_t
)sp
;
294 return (RET_SUCCESS
);