Return to pa_common.C CVS log

Up to [parser3project] / parser3 / src / main

code documentation ++

/** @file
	Parser: commonly functions.

	Copyright(c) 2001 ArtLebedev Group(http://www.artlebedev.com)

	Author: Alexander Petrosyan <paf@design.ru>(http://design.ru/paf)

	$Id: pa_common.C,v 1.52 2001/04/26 14:55:28 paf Exp $
*/

#include "pa_config_includes.h"
#include "pa_common.h"
#include "pa_types.h"
#include "pa_exception.h"
#include "pa_pool.h"
#include "pa_globals.h"
#include "pa_value.h"
#include "pa_hash.h"
#include "pa_string.h"

#ifndef WIN32
#	ifndef _O_TEXT
#		define _O_TEXT 0
#	endif
#	ifndef _O_BINARY
#		define _O_BINARY 0
#	endif
#endif

#if _MSC_VER

int __vsnprintf(char *b, size_t s, const char *f, va_list l) {
	int r=_vsnprintf(b, --s, f, l);
	b[s]=0;
	return r;
}
int __snprintf(char *b, size_t s, const char *f, ...) {
	va_list l;
    va_start(l, f);
    int r=__vsnprintf(b, s, f, l);
    va_end(l);
	return r;
}

#endif


char *file_read_text(Pool& pool, const String& file_spec, bool fail_on_read_problem) {
	void *result;
	size_t size;
	return file_read(pool, file_spec, result, size, true, 
		fail_on_read_problem)?(char *)result:0;
}
bool file_read(Pool& pool, const String& file_spec, 
			   void*& data, size_t& read_size, bool as_text,
			   bool fail_on_read_problem,
			   size_t offset, size_t limit) {
	const char *fname=file_spec.cstr(String::UL_FILE_NAME);
	int f;
    struct stat finfo;

	// first open, next stat:
	// directory update of NTFS hard links performed on open.
	// ex: 
	//   a.html:^test[] and b.html hardlink to a.html
	//   user inserts ! before ^test in a.html
	//   directory entry of b.html in NTFS not updated at once,
	//   they delay update till open, so we would receive "!^test[" string
	//   if would do stat, next open.
    if(
		(f=open(fname, O_RDONLY|(as_text?_O_TEXT:_O_BINARY)))>=0 && 
		stat(fname, &finfo)==0) {
		/*if(exclusive)
			flock(f, LOCK_EX);*/
		size_t max_size=limit?min(offset+limit, finfo.st_size)-offset:finfo.st_size;
		if(!max_size) { // eof
			data=0;
			read_size=0;
			return true;
		}
		data=pool.malloc(max_size+(as_text?1:0));
		if(offset)
			lseek(f, offset, SEEK_SET);
		read_size=read(f, data, max_size);
		/*if(exclusive)
			flock(f, LOCK_UN);*/
		close(f);

		if(read_size>=0 && read_size<=max_size) {
			if(as_text)
				((char*&)data)[read_size]=0;
		} else
			PTHROW(0, 0, 
				&file_spec, 
				"read failed: actually read %d bytes count not in [0..%lu] valid range", 
					read_size, (unsigned long)max_size); //never
		
		return true;//prepare_config(result, remove_empty_lines);
    }
	if(fail_on_read_problem)
		PTHROW(0, 0, 
			&file_spec, 
			"read failed: %s (%d)", strerror(errno), errno);
    return false;
}

void file_write(Pool& pool, 
				const String& file_spec, 
				const void *data, size_t size, 
				bool as_text/*, 
				bool exclusive*/) {
	const char *fname=file_spec.cstr(String::UL_FILE_NAME);
	int f;
	if(access(fname, W_OK)!=0) {/*no*/
		size_t pos_after=1;
		int pos_before;
		while((pos_before=file_spec.pos("/", 1, pos_after))>=0) {
			mkdir(file_spec.mid(0, pos_before).cstr(String::UL_FILE_NAME), 0775);
			pos_after=pos_before+1;
		}

		if((f=open(fname, O_WRONLY|O_CREAT|_O_BINARY, 0666))>0)
			close(f);
	}
	if(access(fname, R_OK|W_OK)==0) {
		int mode=O_RDWR|(as_text?_O_TEXT:_O_BINARY)
#ifdef WIN32
			|O_TRUNC
#endif
		;
		if((f=open(fname, mode, 0666))>=0) {
			/*if(exclusive)
				flock(f, LOCK_EX);*/
			
			if(size) write(f, data, size);
#ifndef WIN32
			ftruncate(f, size);
#endif
			/*if(exclusive)
				flock(f, LOCK_UN);*/
			close(f);
			return;
		}
	}
	PTHROW(0, 0, 
		&file_spec, 
		"write failed: %s (%d)", strerror(errno), errno);
}

static void rmdir(const String& file_spec, size_t pos_after) {
	int pos_before;
	if((pos_before=file_spec.pos("/", 1, pos_after))>=0)
		rmdir(file_spec, pos_before+1);
	
	rmdir(file_spec.mid(0, pos_after-1/* / */).cstr(String::UL_FILE_NAME));
}
void file_delete(Pool& pool, const String& file_spec) {
	if(unlink(file_spec.cstr(String::UL_FILE_NAME))!=0)
		PTHROW(0, 0, 
			&file_spec, 
			"unlink failed: %s (%d)", strerror(errno), errno);

	rmdir(file_spec, 1);
}


static bool entry_readable(const String& file_spec, bool need_dir) {
    const char *fname=file_spec.cstr(String::UL_FILE_NAME);
	struct stat finfo;
	if(access(fname, R_OK)==0 && stat(fname, &finfo)==0) {
		bool is_dir=(bool)(finfo.st_mode&S_IFDIR);
		return is_dir==need_dir;
	}
	return false;
}
bool file_readable(const String& file_spec) {
	return entry_readable(file_spec, false);
}
bool dir_readable(const String& file_spec) {
	return entry_readable(file_spec, true);
}
bool file_executable(const String& file_spec) {
    return access(file_spec.cstr(String::UL_FILE_NAME), X_OK)==0;
}

size_t file_size(const String& file_spec) {
	Pool& pool=file_spec.pool();
	const char *fname=file_spec.cstr(String::UL_FILE_NAME);
    struct stat finfo;
	if(stat(fname, &finfo)!=0)
		PTHROW(0, 0, 
			&file_spec, 
			"write failed: %s (%d)", strerror(errno), errno);
	return finfo.st_size;
}

char *getrow(char **row_ref, char delim) {
    char *result=*row_ref;
    if(result) {
		*row_ref=strchr(result, delim);
		if(*row_ref) 
			*((*row_ref)++)=0; 
		else if(!*result) 
			return 0;
    }
    return result;
}

char *lsplit(char *string, char delim) {
    if(string) {
		char *v=strchr(string, delim);
		if(v) {
			*v=0;
			return v+1;
		}
    }
    return 0;
}

char *lsplit(char **string_ref, char delim) {
    char *result=*string_ref;
	char *next=lsplit(*string_ref, delim);
    *string_ref=next;
    return result;
}

char *rsplit(char *string, char delim) {
    if(string) {
		char *v=strrchr(string, delim);
		if(v) {
			*v=0;
			return v+1;
		}
    }
    return NULL;	
}

/// @todo less stupid type detection
char *format(Pool& pool, double value, char *fmt) {
	char *result=(char *)pool.malloc(MAX_NUMBER);
	if(fmt)
		if(strpbrk(fmt, "diouxX"))
			if(strpbrk(fmt, "ouxX"))
				snprintf(result, MAX_NUMBER, fmt, (uint)value );
			else
				snprintf(result, MAX_NUMBER, fmt, (int)value );
		else
			snprintf(result, MAX_NUMBER, fmt, value);
	else
		snprintf(result, MAX_NUMBER, "%d", (int)value);
	
	return result;
}

size_t stdout_write(const void *buf, size_t size) {
#ifdef WIN32
	do{
		int chunk_written=fwrite(buf, 1, min(8*0x400, size), stdout);
		if(chunk_written<=0)
			break;
		size-=chunk_written;
		buf=((const char*)buf)+chunk_written;
	} while(size>0);

	return size;
#else
	return fwrite(buf, 1, size, stdout);
#endif
}

const char *unescape_chars(Pool& pool, const char *cp, int len) {
	char *s=(char *)pool.malloc(len + 1);
	enum EscapeState {
		EscapeRest, 
		EscapeFirst, 
		EscapeSecond
	} escapeState=EscapeRest;
	int escapedValue=0;
	int srcPos=0;
	int dstPos=0;
	while(srcPos < len) {
		int ch=cp[srcPos];
		switch(escapeState) {
			case EscapeRest:
			if(ch=='%') {
				escapeState=EscapeFirst;
			} else if(ch=='+') {
				s[dstPos++]=' ';
			} else {
				s[dstPos++]=ch;	
			}
			break;
			case EscapeFirst:
			escapedValue=hex_value[ch] << 4;	
			escapeState=EscapeSecond;
			break;
			case EscapeSecond:
			escapedValue +=hex_value[ch];
			s[dstPos++]=escapedValue;
			escapeState=EscapeRest;
			break;
		}
		srcPos++;
	}
	s[dstPos]=0;
	return s;
}

/// used by attributed_meaning_to_string / append_attribute_subattribute
struct Attributed_meaning_info {
	String *header; // header line being constructed
	String::Untaint_lang lang; // language in which to append to that line
};
static void append_attribute_subattribute(const Hash::Key& akey, Hash::Val *avalue, 
										  void *info) {
	if(akey==VALUE_NAME)
		return;

	Attributed_meaning_info& ami=*static_cast<Attributed_meaning_info *>(info);

	// ...; charset=windows1251
	*ami.header << "; ";
	ami.header->append(akey, ami.lang, true);
	*ami.header << "=";
	ami.header->append(static_cast<Value *>(avalue)->as_string(), ami.lang, true);
}
const String& attributed_meaning_to_string(Value& meaning, 
										   String::Untaint_lang lang) {
	String &result=*new(meaning.pool()) String(meaning.pool());
	if(Hash *hash=meaning.get_hash()) {
		// $value(value) $subattribute(subattribute value)
		if(Value *value=static_cast<Value *>(hash->get(*value_name)))
			result.append(value->as_string(), lang, true);

		Attributed_meaning_info attributed_meaning_info={
			&result,
			lang
		};
		hash->for_each(append_attribute_subattribute, &attributed_meaning_info);
	} else // result value
		result.append(meaning.as_string(), lang, true);

	return result;
}

#ifdef WIN32
void back_slashes_to_slashes(char *s) {
	if(s)
		for(; *s; s++)
			if(*s=='\\')
				*s='/';
}
/*
void slashes_to_back_slashes(char *s) {
	if(s)
		for(; *s; s++)
			if(*s=='/')
				*s='\\';
}
*/
#endif

bool StrEqNc(const char *s1, const char *s2, bool strict) {
	while(true) {
		if(!(*s1)) {
			if(!(*s2))
				return true;
			else
				return !strict;
		} else if(!(*s2))
			return !strict;
		if(isalpha(*s1)) {
			if(tolower(*s1) !=tolower(*s2))
				return false;
		} else if((*s1) !=(*s2))
			return false;
		s1++;
		s2++;
	}
}

char *unquote(char*& current, char stop_at) {
	char *result=current;
	char *dest=current;
	// skip leading WS
	while(*current==' ' || *current=='\t')
		current++;
	if(!*current)
		return current=0;

	bool quoted=*current=='"';
	if(quoted)
		current++;

	for(; *current; ) {
		if(quoted)
			switch(*current) {
			case '\\': // "...\c
				if(current[1])
					current++;
				break;
			case '"':
				current++;
				goto break2;
			}
		else
			if(*current==stop_at)
				break;

		*dest++=*current++;
	}
break2:
	if(*current)
		current++; // skip 'stop_at'
	*dest=0;
	return result;
}