8b2e792003-01-14Per Hedbor /* || This file is part of Pike. For copyright information see COPYRIGHT. || Pike is distributed under GPL, LGPL and MPL. See the file COPYING || for more information.
5606222004-08-26David Gourdelier || $Id: e_source_block_pikestream.c,v 1.4 2004/08/25 23:04:24 vida Exp $
8b2e792003-01-14Per Hedbor */ #include "global.h" #include "bignum.h" #include "object.h" #include "interpret.h" #include "fdlib.h" #include "fd_control.h" #include <sys/types.h> #include <sys/stat.h> #include "shuffler.h" /* Source: blocking pike-stream * Argument: Stdio.File lookalike without read * callback support (read only needed) */ struct pf_source { struct source s; struct object *obj;
5606222004-08-26David Gourdelier  INT64 len, skip;
8b2e792003-01-14Per Hedbor }; static struct data get_data( struct source *_s, int len ) { struct pf_source *s = (struct pf_source *)_s; struct data res; res.len = 0; res.off = 0; res.do_free = 0;
5606222004-08-26David Gourdelier  if( s->len>0 && len > s->len ) { len = s->len; s->s.eof = 1; } do {
8b2e792003-01-14Per Hedbor  struct pike_string *st; push_int( len ); apply( s->obj, "read", 1 );
5606222004-08-26David Gourdelier  if(Pike_sp[-1].type != PIKE_T_STRING || !(st = Pike_sp[-1].u.string)->len) { pop_stack(); break;
8b2e792003-01-14Per Hedbor  } if( st->len < s->skip ) s->skip -= st->len;
5606222004-08-26David Gourdelier  else { res.data = malloc(st->len -= s->skip); memcpy(res.data, st->str+s->skip, res.len = st->len);
8b2e792003-01-14Per Hedbor  res.do_free = 1; s->skip = 0; }
5606222004-08-26David Gourdelier  pop_stack();
8b2e792003-01-14Per Hedbor  }
5606222004-08-26David Gourdelier  while(s->skip || !res.len); if(res.len < len)
8b2e792003-01-14Per Hedbor  s->s.eof = 1;
5606222004-08-26David Gourdelier  if(s->len > 0) s->len -= res.len;
8b2e792003-01-14Per Hedbor  return res; } static void free_source( struct source *_s ) { free_object(((struct pf_source *)_s)->obj); } struct source *source_block_pikestream_make( struct svalue *s, INT64 start, INT64 len ) { struct pf_source *res; if( (s->type != PIKE_T_OBJECT) || (find_identifier("read",s->u.object->prog)==-1) ) return 0; res = malloc( sizeof( struct pf_source ) ); MEMSET( res, 0, sizeof( struct pf_source ) ); res->len = len; res->skip = start; res->s.get_data = get_data; res->s.free_source = free_source; res->obj = s->u.object;
50ea682003-03-14Henrik Grubbström (Grubba)  add_ref(res->obj);
8b2e792003-01-14Per Hedbor  return (struct source *)res; } void source_block_pikestream_exit( ) { } void source_block_pikestream_init( ) { }