off_t File_class::size(FILE* f) { MY_STAT s; // Note that my_fstat behaves *differently* than my_stat. ARGGGHH! if (my_fstat(fileno(f), &s, MYF(0))) return 0; return s.st_size; }
int mi_status(MI_INFO *info, register MI_ISAMINFO *x, uint flag) { MY_STAT state; MYISAM_SHARE *share=info->s; DBUG_ENTER("mi_status"); x->recpos = info->lastpos; if (flag == HA_STATUS_POS) DBUG_RETURN(0); /* Compatible with ISAM */ if (!(flag & HA_STATUS_NO_LOCK)) { pthread_mutex_lock(&share->intern_lock); VOID(_mi_readinfo(info,F_RDLCK,0)); fast_mi_writeinfo(info); pthread_mutex_unlock(&share->intern_lock); } if (flag & HA_STATUS_VARIABLE) { x->records = info->state->records; x->deleted = info->state->del; x->delete_length = info->state->empty; x->data_file_length =info->state->data_file_length; x->index_file_length=info->state->key_file_length; x->keys = share->state.header.keys; x->check_time = share->state.check_time; x->mean_reclength= x->records ? (ulong) ((x->data_file_length - x->delete_length) / x->records) : (ulong) share->min_pack_length; } if (flag & HA_STATUS_ERRKEY) { x->errkey = info->errkey; x->dupp_key_pos= info->dupp_key_pos; } if (flag & HA_STATUS_CONST) { x->reclength = share->base.reclength; x->max_data_file_length=share->base.max_data_file_length; x->max_index_file_length=info->s->base.max_key_file_length; x->filenr = info->dfile; x->options = share->options; x->create_time=share->state.create_time; x->reflength= mi_get_pointer_length(share->base.max_data_file_length, myisam_data_pointer_size); x->record_offset= ((share->options & (HA_OPTION_PACK_RECORD | HA_OPTION_COMPRESS_RECORD)) ? 0L : share->base.pack_reclength); x->sortkey= -1; /* No clustering */ x->rec_per_key = share->state.rec_per_key_part; x->key_map = share->state.key_map; x->data_file_name = share->data_file_name; x->index_file_name = share->index_file_name; /* The following should be included even if we are not compiling with USE_RAID as the client must be able to request it! */ x->raid_type= share->base.raid_type; x->raid_chunks= share->base.raid_chunks; x->raid_chunksize= share->base.raid_chunksize; } if ((flag & HA_STATUS_TIME) && !my_fstat(info->dfile,&state,MYF(0))) x->update_time=state.st_mtime; else x->update_time=0; if (flag & HA_STATUS_AUTO) { x->auto_increment= share->state.auto_increment+1; if (!x->auto_increment) /* This shouldn't happen */ x->auto_increment= ~(ulonglong) 0; } DBUG_RETURN(0); }
int modify_defaults_file(const char *file_location, const char *option, const char *option_value, const char *section_name, int remove_option) { FILE *cnf_file; MY_STAT file_stat; char linebuff[BUFF_SIZE], *src_ptr, *dst_ptr, *file_buffer; size_t opt_len= 0, optval_len= 0, sect_len; uint nr_newlines= 0, buffer_size; my_bool in_section= FALSE, opt_applied= 0; uint reserve_extended; uint new_opt_len; int reserve_occupied= 0; DBUG_ENTER("modify_defaults_file"); if (!(cnf_file= my_fopen(file_location, O_RDWR | O_BINARY, MYF(0)))) DBUG_RETURN(2); /* my_fstat doesn't use the flag parameter */ if (my_fstat(fileno(cnf_file), &file_stat, MYF(0))) goto malloc_err; if (option && option_value) { opt_len= strlen(option); optval_len= strlen(option_value); } new_opt_len= opt_len + 1 + optval_len + NEWLINE_LEN; /* calculate the size of the buffer we need */ reserve_extended= (opt_len + 1 + /* For '=' char */ optval_len + /* Option value len */ NEWLINE_LEN + /* Space for newline */ RESERVE); /* Some additional space */ buffer_size= (file_stat.st_size + 1); /* The ending zero */ /* Reserve space to read the contents of the file and some more for the option we want to add. */ if (!(file_buffer= (char*) my_malloc(buffer_size + reserve_extended, MYF(MY_WME)))) goto malloc_err; sect_len= strlen(section_name); for (dst_ptr= file_buffer; fgets(linebuff, BUFF_SIZE, cnf_file); ) { /* Skip over whitespaces */ for (src_ptr= linebuff; my_isspace(&my_charset_latin1, *src_ptr); src_ptr++) {} if (!*src_ptr) /* Empty line */ { nr_newlines++; continue; } /* correct the option (if requested) */ if (option && in_section && !strncmp(src_ptr, option, opt_len) && (*(src_ptr + opt_len) == '=' || my_isspace(&my_charset_latin1, *(src_ptr + opt_len)) || *(src_ptr + opt_len) == '\0')) { char *old_src_ptr= src_ptr; src_ptr= strend(src_ptr+ opt_len); /* Find the end of the line */ /* could be negative */ reserve_occupied+= (int) new_opt_len - (int) (src_ptr - old_src_ptr); if (reserve_occupied >= (int) reserve_extended) { reserve_extended= (uint) reserve_occupied + RESERVE; if (!(file_buffer= (char*) my_realloc(file_buffer, buffer_size + reserve_extended, MYF(MY_WME|MY_FREE_ON_ERROR)))) goto malloc_err; } opt_applied= 1; dst_ptr= add_option(dst_ptr, option_value, option, remove_option); } else { /* If we are going to the new group and have an option to apply, do it now. If we are removing a single option or the whole section this will only trigger opt_applied flag. */ if (in_section && !opt_applied && *src_ptr == '[') { dst_ptr= add_option(dst_ptr, option_value, option, remove_option); opt_applied= 1; /* set the flag to do write() later */ reserve_occupied= new_opt_len+ opt_len + 1 + NEWLINE_LEN; } for (; nr_newlines; nr_newlines--) dst_ptr= strmov(dst_ptr, NEWLINE); /* Skip the section if MY_REMOVE_SECTION was given */ if (!in_section || remove_option != MY_REMOVE_SECTION) dst_ptr= strmov(dst_ptr, linebuff); } /* Look for a section */ if (*src_ptr == '[') { /* Copy the line to the buffer */ if (!strncmp(++src_ptr, section_name, sect_len)) { src_ptr+= sect_len; /* Skip over whitespaces. They are allowed after section name */ for (; my_isspace(&my_charset_latin1, *src_ptr); src_ptr++) {} if (*src_ptr != ']') { in_section= FALSE; continue; /* Missing closing parenthesis. Assume this was no group */ } if (remove_option == MY_REMOVE_SECTION) dst_ptr= dst_ptr - strlen(linebuff); in_section= TRUE; } else in_section= FALSE; /* mark that this section is of no interest to us */ } } /* File ended. Apply an option or set opt_applied flag (in case of MY_REMOVE_SECTION) so that the changes are saved. Do not do anything if we are removing non-existent option. */ if (!opt_applied && in_section && (remove_option != MY_REMOVE_OPTION)) { /* New option still remains to apply at the end */ if (!remove_option && *(dst_ptr - 1) != '\n') dst_ptr= strmov(dst_ptr, NEWLINE); dst_ptr= add_option(dst_ptr, option_value, option, remove_option); opt_applied= 1; } for (; nr_newlines; nr_newlines--) dst_ptr= strmov(dst_ptr, NEWLINE); if (opt_applied) { /* Don't write the file if there are no changes to be made */ if (my_chsize(fileno(cnf_file), (my_off_t) (dst_ptr - file_buffer), 0, MYF(MY_WME)) || my_fseek(cnf_file, 0, MY_SEEK_SET, MYF(0)) || my_fwrite(cnf_file, (uchar*) file_buffer, (size_t) (dst_ptr - file_buffer), MYF(MY_NABP))) goto err; } if (my_fclose(cnf_file, MYF(MY_WME))) DBUG_RETURN(1); my_free(file_buffer, MYF(0)); DBUG_RETURN(0); err: my_free(file_buffer, MYF(0)); malloc_err: my_fclose(cnf_file, MYF(0)); DBUG_RETURN(1); /* out of resources */ }