mirror of
				https://github.com/MariaDB/server.git
				synced 2025-10-30 02:16:32 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			146 lines
		
	
	
	
		
			4.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			146 lines
		
	
	
	
		
			4.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /* Copyright (c) 2011, 2023, Oracle and/or its affiliates.
 | |
| 
 | |
|   This program is free software; you can redistribute it and/or modify
 | |
|   it under the terms of the GNU General Public License, version 2.0,
 | |
|   as published by the Free Software Foundation.
 | |
| 
 | |
|   This program is also distributed with certain software (including
 | |
|   but not limited to OpenSSL) that is licensed under separate terms,
 | |
|   as designated in a particular file or component or in included license
 | |
|   documentation.  The authors of MySQL hereby grant you an additional
 | |
|   permission to link the program and your derivative works with the
 | |
|   separately licensed software that they have included with MySQL.
 | |
| 
 | |
|   This program is distributed in the hope that it will be useful,
 | |
|   but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
|   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | |
|   GNU General Public License, version 2.0, for more details.
 | |
| 
 | |
|   You should have received a copy of the GNU General Public License
 | |
|   along with this program; if not, write to the Free Software Foundation,
 | |
|   51 Franklin Street, Fifth Floor, Boston, MA 02110-1335 USA */
 | |
| 
 | |
| #include <my_global.h>
 | |
| #include <my_thread.h>
 | |
| #include <pfs_instr.h>
 | |
| #include <pfs_stat.h>
 | |
| #include <pfs_global.h>
 | |
| #include <pfs_user.h>
 | |
| #include <pfs_buffer_container.h>
 | |
| #include <tap.h>
 | |
| 
 | |
| #include "stub_pfs_global.h"
 | |
| #include "stub_global_status_var.h"
 | |
| 
 | |
| #include <string.h> /* memset */
 | |
| 
 | |
| void test_oom()
 | |
| {
 | |
|   PSI *psi;
 | |
|   PFS_global_param param;
 | |
|   PSI_bootstrap *boot;
 | |
| 
 | |
|   memset(& param, 0xFF, sizeof(param));
 | |
|   param.m_enabled= true;
 | |
|   param.m_mutex_class_sizing= 0;
 | |
|   param.m_rwlock_class_sizing= 0;
 | |
|   param.m_cond_class_sizing= 0;
 | |
|   param.m_thread_class_sizing= 10;
 | |
|   param.m_table_share_sizing= 0;
 | |
|   param.m_file_class_sizing= 0;
 | |
|   param.m_socket_class_sizing= 0;
 | |
|   param.m_mutex_sizing= 0;
 | |
|   param.m_rwlock_sizing= 0;
 | |
|   param.m_cond_sizing= 0;
 | |
|   param.m_thread_sizing= 1000;
 | |
|   param.m_table_sizing= 0;
 | |
|   param.m_file_sizing= 0;
 | |
|   param.m_file_handle_sizing= 0;
 | |
|   param.m_socket_sizing= 0;
 | |
|   param.m_events_waits_history_sizing= 10;
 | |
|   param.m_events_waits_history_long_sizing= 0;
 | |
|   param.m_setup_actor_sizing= 0;
 | |
|   param.m_setup_object_sizing= 0;
 | |
|   param.m_host_sizing= 0;
 | |
|   param.m_user_sizing= 1000;
 | |
|   param.m_account_sizing= 0;
 | |
|   param.m_stage_class_sizing= 50;
 | |
|   param.m_events_stages_history_sizing= 0;
 | |
|   param.m_events_stages_history_long_sizing= 0;
 | |
|   param.m_statement_class_sizing= 50;
 | |
|   param.m_events_statements_history_sizing= 0;
 | |
|   param.m_events_statements_history_long_sizing= 0;
 | |
|   param.m_events_transactions_history_sizing= 0;
 | |
|   param.m_events_transactions_history_long_sizing= 0;
 | |
|   param.m_digest_sizing= 0;
 | |
|   param.m_session_connect_attrs_sizing= 0;
 | |
|   param.m_program_sizing= 0;
 | |
|   param.m_statement_stack_sizing= 0;
 | |
|   param.m_memory_class_sizing= 10;
 | |
|   param.m_metadata_lock_sizing= 0;
 | |
|   param.m_max_digest_length= 0;
 | |
|   param.m_max_sql_text_length= 0;
 | |
| 
 | |
|   /* Setup */
 | |
| 
 | |
|   stub_alloc_always_fails= false;
 | |
|   stub_alloc_fails_after_count= 1000;
 | |
| 
 | |
|   pre_initialize_performance_schema();
 | |
|   boot= initialize_performance_schema(¶m);
 | |
|   psi= (PSI *)boot->get_interface(PSI_VERSION_1);
 | |
| 
 | |
|   PSI_thread_key thread_key_1;
 | |
|   PSI_thread_info all_thread[]=
 | |
|   {
 | |
|     {&thread_key_1, "T-1", 0}
 | |
|   };
 | |
|   psi->register_thread("test", all_thread, 1);
 | |
| 
 | |
|   PSI_thread *thread_1= psi->new_thread(thread_key_1, NULL, 0);
 | |
|   psi->set_thread(thread_1);
 | |
| 
 | |
|   /* Tests */
 | |
| 
 | |
|   int first_fail= 1;
 | |
|   stub_alloc_fails_after_count= first_fail;
 | |
|   psi->set_thread_account("user1", 5, "", 0);
 | |
|   ok(global_user_container.m_lost == 1, "oom (user)");
 | |
| 
 | |
|   stub_alloc_fails_after_count= first_fail + 1;
 | |
|   psi->set_thread_account("user2", 5, "", 0);
 | |
|   ok(global_user_container.m_lost == 2, "oom (user waits)");
 | |
| 
 | |
|   stub_alloc_fails_after_count= first_fail + 2;
 | |
|   psi->set_thread_account("user3", 5, "", 0);
 | |
|   ok(global_user_container.m_lost == 3, "oom (user stages)");
 | |
| 
 | |
|   stub_alloc_fails_after_count= first_fail + 3;
 | |
|   psi->set_thread_account("user4", 5, "", 0);
 | |
|   ok(global_user_container.m_lost == 4, "oom (user statements)");
 | |
| 
 | |
|   stub_alloc_fails_after_count= first_fail + 4;
 | |
|   psi->set_thread_account("user5", 5, "", 0);
 | |
|   ok(global_user_container.m_lost == 5, "oom (user transactions)");
 | |
| 
 | |
|   stub_alloc_fails_after_count= first_fail + 5;
 | |
|   psi->set_thread_account("user6", 5, "", 0);
 | |
|   ok(global_user_container.m_lost == 6, "oom (user memory)");
 | |
| 
 | |
|   shutdown_performance_schema();
 | |
| }
 | |
| 
 | |
| void do_all_tests()
 | |
| {
 | |
|   test_oom();
 | |
| }
 | |
| 
 | |
| int main(int, char **)
 | |
| {
 | |
|   plan(6);
 | |
|   MY_INIT("pfs_user-oom-t");
 | |
|   do_all_tests();
 | |
|   my_end(0);
 | |
|   return (exit_status());
 | |
| }
 | |
| 
 | 
