123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960 |
- # Copyright (C) 2012-2023 Zammad Foundation, https://zammad-foundation.org/
- module ApplicationModel::ChecksAttributeValuesAndLength
- extend ActiveSupport::Concern
- included do
- before_create :check_attribute_values_and_length
- before_update :check_attribute_values_and_length
- end
- =begin
- 1) check string/varchar size and cut them if needed
- 2) check string for null byte \u0000 and remove it
- =end
- def check_attribute_values_and_length
- columns = self.class.columns_hash
- attributes.each do |name, value|
- next if !value.instance_of?(String)
- column = columns[name]
- next if !column
- if column.type == :binary
- self[name].force_encoding('BINARY')
- end
- next if value.blank?
- # strip null byte chars (postgresql will complain about it)
- if column.type == :text && Rails.application.config.db_null_byte == false
- self[name].delete!("\u0000")
- end
- # for varchar check length and replace null bytes
- limit = column.limit
- if limit
- current_length = value.length
- if limit < current_length
- logger.warn "WARNING: cut string because of database length #{self.class}.#{name}(#{limit} but is #{current_length}:#{value})"
- self[name] = value[0, limit]
- end
- # strip null byte chars (postgresql will complain about it)
- if Rails.application.config.db_null_byte == false
- self[name].delete!("\u0000")
- end
- end
- # strip 4 bytes utf8 chars if needed (mysql/mariadb will complain it)
- next if self[name].blank?
- self[name] = self[name].utf8_to_3bytesutf8
- end
- true
- end
- end
|