← Back to branch summary

~ubuntu-branches/ubuntu/maverick/eucalyptus/maverick

« back to all changes in this revision

Viewing changes to clc/modules/core/src/main/java/edu/ucsb/eucalyptus/StartupChecks.java

  • Committer: Bazaar Package Importer
  • Author(s): Dave Walker (Daviey)
  • Date: 2010-07-21 17:27:10 UTC
  • mfrom: (1.1.38 upstream)
  • Revision ID: james.westby@ubuntu.com-20100721172710-7xv07dmdqgivc3t9
Tags: 2.0~bzr1211-0ubuntu1
https://launchpad.net/bugs/597330
https://launchpad.net/bugs/595421
* New major upstream version merge, 2.0 (r1211).
* debian/patches/:
  - 01-wsdl-stubs.patch, debian/wsdl.md5sums: wsdl stubs updated.
  - 02-Makefile.patch: Updated to reflect new code layout.
  - 07-local_support_euca_conf-in.patch: Updated to reflect new code layout.
  - 08-ubuntu-default-networking.patch: Refreshed.
  - 09-small-128-192MB.patch: Updated to point to new location.
  - 10-disable-iscsi.patch: Refreshed.
  - 11-state-cleanup-memleakfix.patch: Removed, fixed upstream.
  - 15-fix-default-ramdisk.patch: Updated to point to new location.
  - 16-kvm_libvirt_xml_default_use_kvm.patch: Updated to reflect changes.
  - 17-fix_walrus_OOM_errors.patch: Removed, fixed upstream.
  - 18-priv_security.patch: Updated to reflect upstream changes.
  - 20-brute-force-webui.patch: Updated to reflect upstream changes. 
  - 21-eucalyptus-1.7-with-gwt-1.6.4.patch: New patch, allows 
    eucalyptus-1.7 to be built against gwt 1.6.4. Based on patch courtesy 
    of Dmitrii Zagorodnov, upstream. (LP: #597330)
* debian/eucalyptus-java-common.links: 
  - Changed symlink for groovy, point to groovy.all.jar, making compatiable 
    with groovy versions >1.7. (LP: #595421)
  - Added ant.jar & jetty-rewrite-handler.jar as they are now required.
* debian/control
  - & debian/build-jars: Added libjavassist-java and libjetty-extra-java as 
    build dependencies.
  - Added libjetty-extra-java as a dependency of eucalyptus-java-common
* The binary resulting jar's have been renamed from eucalyptus-*-1.6.2.jar
  to eucalyptus-*-main.jar:    
  - debian/eucalyptus-cc.upstart
  - debian/eucalyptus-cloud.install
  - debian/eucalyptus-common.eucalyptus.upstart
  - debian/eucalyptus-java-common.install
  - debian/eucalyptus-network.upstart
  - debian/eucalyptus-sc.install
  - debian/eucalyptus-walrus.install
* debian/eucalyptus-java-common.install: New upstream jars that have been
  installed:
  - eucalyptus-db-hsqldb-ext-main.jar
  - eucalyptus-component-main.jar
* debian/control:
  - Updated Standards Version to 3.8.4 (no change)
  - Updated the upstream Homepage to: http://open.eucalyptus.com/
  - Changed Vcs-Bzr to reflect new location of Ubuntu hosted development branch.
  - Made the Build Dependency of groovy and the binary eucalyptus-java-common
    package depend on version >=1.7.

Show diffs side-by-side

added added

removed removed

Lines of Context:
clc/modules/core/src/main/java/edu/ucsb/eucalyptus/StartupChecks.java
1
 
/*******************************************************************************
2
 
*Copyright (c) 2009  Eucalyptus Systems, Inc.
3
 
* 
4
 
*  This program is free software: you can redistribute it and/or modify
5
 
*  it under the terms of the GNU General Public License as published by
6
 
*  the Free Software Foundation, only version 3 of the License.
7
 
* 
8
 
* 
9
 
*  This file is distributed in the hope that it will be useful, but WITHOUT
10
 
*  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11
 
*  FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12
 
*  for more details.
13
 
* 
14
 
*  You should have received a copy of the GNU General Public License along
15
 
*  with this program.  If not, see <http://www.gnu.org/licenses/>.
16
 
* 
17
 
*  Please contact Eucalyptus Systems, Inc., 130 Castilian
18
 
*  Dr., Goleta, CA 93101 USA or visit <http://www.eucalyptus.com/licenses/>
19
 
*  if you need additional information or have any questions.
20
 
* 
21
 
*  This file may incorporate work covered under the following copyright and
22
 
*  permission notice:
23
 
* 
24
 
*    Software License Agreement (BSD License)
25
 
* 
26
 
*    Copyright (c) 2008, Regents of the University of California
27
 
*    All rights reserved.
28
 
* 
29
 
*    Redistribution and use of this software in source and binary forms, with
30
 
*    or without modification, are permitted provided that the following
31
 
*    conditions are met:
32
 
* 
33
 
*      Redistributions of source code must retain the above copyright notice,
34
 
*      this list of conditions and the following disclaimer.
35
 
* 
36
 
*      Redistributions in binary form must reproduce the above copyright
37
 
*      notice, this list of conditions and the following disclaimer in the
38
 
*      documentation and/or other materials provided with the distribution.
39
 
* 
40
 
*    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
41
 
*    IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
42
 
*    TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
43
 
*    PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
44
 
*    OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
45
 
*    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
46
 
*    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
47
 
*    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
48
 
*    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
49
 
*    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
50
 
*    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. USERS OF
51
 
*    THIS SOFTWARE ACKNOWLEDGE THE POSSIBLE PRESENCE OF OTHER OPEN SOURCE
52
 
*    LICENSED MATERIAL, COPYRIGHTED MATERIAL OR PATENTED MATERIAL IN THIS
53
 
*    SOFTWARE, AND IF ANY SUCH MATERIAL IS DISCOVERED THE PARTY DISCOVERING
54
 
*    IT MAY INFORM DR. RICH WOLSKI AT THE UNIVERSITY OF CALIFORNIA, SANTA
55
 
*    BARBARA WHO WILL THEN ASCERTAIN THE MOST APPROPRIATE REMEDY, WHICH IN
56
 
*    THE REGENTS’ DISCRETION MAY INCLUDE, WITHOUT LIMITATION, REPLACEMENT
57
 
*    OF THE CODE SO IDENTIFIED, LICENSING OF THE CODE SO IDENTIFIED, OR
58
 
*    WITHDRAWAL OF THE CODE CAPABILITY TO THE EXTENT NEEDED TO COMPLY WITH
59
 
*    ANY SUCH LICENSES OR RIGHTS.
60
 
*******************************************************************************/
61
 
/*
62
 
 * Author: chris grzegorczyk <grze@eucalyptus.com>
63
 
 */
64
 
package edu.ucsb.eucalyptus;
65
 
 
66
 
import edu.ucsb.eucalyptus.cloud.entities.CertificateInfo;
67
 
import edu.ucsb.eucalyptus.cloud.entities.Counters;
68
 
import edu.ucsb.eucalyptus.cloud.entities.ImageInfo;
69
 
import edu.ucsb.eucalyptus.cloud.entities.UserGroupInfo;
70
 
import edu.ucsb.eucalyptus.cloud.entities.UserInfo;
71
 
import edu.ucsb.eucalyptus.cloud.entities.ObjectInfo;
72
 
import edu.ucsb.eucalyptus.cloud.entities.VmType;
73
 
import edu.ucsb.eucalyptus.cloud.entities.SystemConfiguration;
74
 
import edu.ucsb.eucalyptus.msgs.Volume;
75
 
import edu.ucsb.eucalyptus.util.SystemUtil;
76
 
import edu.ucsb.eucalyptus.util.UserManagement;
77
 
import com.eucalyptus.util.StorageProperties;
78
 
import org.apache.log4j.Logger;
79
 
import org.bouncycastle.jce.provider.BouncyCastleProvider;
80
 
import org.bouncycastle.util.encoders.UrlBase64;
81
 
import org.hibernate.HibernateException;
82
 
import org.mortbay.jetty.security.Credential;
83
 
 
84
 
import com.eucalyptus.auth.Credentials;
85
 
import com.eucalyptus.auth.util.AbstractKeyStore;
86
 
import com.eucalyptus.auth.util.EucaKeyStore;
87
 
import com.eucalyptus.bootstrap.Component;
88
 
import com.eucalyptus.util.BaseDirectory;
89
 
import com.eucalyptus.util.DatabaseUtil;
90
 
import com.eucalyptus.util.EntityWrapper;
91
 
import com.eucalyptus.util.EucalyptusCloudException;
92
 
import com.eucalyptus.util.EucalyptusProperties;
93
 
import com.eucalyptus.util.SubDirectory;
94
 
 
95
 
import java.io.File;
96
 
import java.io.IOException;
97
 
import java.lang.reflect.InvocationTargetException;
98
 
import java.security.GeneralSecurityException;
99
 
import java.security.PrivateKey;
100
 
import java.security.Security;
101
 
import java.security.cert.X509Certificate;
102
 
import java.util.List;
103
 
import java.util.concurrent.ConcurrentNavigableMap;
104
 
import java.util.concurrent.ConcurrentSkipListMap;
105
 
 
106
 
public class StartupChecks {
107
 
 
108
 
  private static String _KS_FORMAT     = "bks";
109
 
  private static String _KS_PASS       = Component.eucalyptus.name( );
110
 
  private static String _USER_KS       = BaseDirectory.CONF.toString( ) + File.separator + "users." + _KS_FORMAT.toLowerCase( );
111
 
  private static Logger LOG            = Logger.getLogger( StartupChecks.class );
112
 
  private static String HEADER_FSTRING = "=================================:: %-20s ::=================================";
113
 
 
114
 
 
115
 
  private static void fail( String... error ) {
116
 
    for ( String s : error )
117
 
      LOG.fatal( s );
118
 
    if ( DatabaseUtil.getEntityManagerFactory( Component.eucalyptus.name( ) ).isOpen( ) ) {
119
 
      DatabaseUtil.getEntityManagerFactory( Component.eucalyptus.name( ) ).close( );
120
 
    }
121
 
    SystemUtil.shutdownWithError( String.format( HEADER_FSTRING, "STARTUP FAILURE" ) );
122
 
  }
123
 
 
124
 
  public static boolean doChecks( ) {
125
 
    StartupChecks.checkDirectories( );
126
 
 
127
 
 
128
 
    boolean hasDb = StartupChecks.checkDatabase( );
129
 
 
130
 
    if ( !hasDb ) {
131
 
      StartupChecks.createDb( );
132
 
      hasDb = StartupChecks.checkDatabase( );
133
 
    }
134
 
 
135
 
    if ( !hasDb ) {
136
 
      LOG.fatal( String.format( HEADER_FSTRING, "STARTUP FAILURE" ) );
137
 
      if ( !hasDb ) LOG.fatal( "Failed to initialize the database in: " + SubDirectory.DB );
138
 
      StartupChecks.fail( "See errors messages above." );
139
 
    }
140
 
 
141
 
 
142
 
    EntityWrapper<ImageInfo> db2 = new EntityWrapper<ImageInfo>( );
143
 
    try {
144
 
      List<ImageInfo> imageList = db2.query( new ImageInfo( ) );
145
 
      for ( ImageInfo image : imageList ) {
146
 
        if ( image.getImageLocation( ).split( "/" ).length != 2 ) {
147
 
          LOG.info( "Image with invalid location, needs to be reregistered: " + image.getImageId( ) );
148
 
          LOG.info( "Removing entry for: " + image.getImageId( ) );
149
 
          db2.delete( image );
150
 
        }
151
 
      }
152
 
      db2.commit( );
153
 
    } catch ( Exception e1 ) {
154
 
      db2.rollback( );
155
 
    }
156
 
 
157
 
    checkWalrus( );
158
 
    checkStorage( );
159
 
 
160
 
    return true;
161
 
  }
162
 
 
163
 
  private static void checkWalrus( ) {
164
 
    EntityWrapper<ObjectInfo> db = new EntityWrapper<ObjectInfo>( );
165
 
    ObjectInfo searchObjectInfo = new ObjectInfo( );
166
 
    List<ObjectInfo> objectInfos = db.query( searchObjectInfo );
167
 
    for ( ObjectInfo objectInfo : objectInfos ) {
168
 
      if ( objectInfo.getObjectKey( ) == null ) objectInfo.setObjectKey( objectInfo.getObjectName( ) );
169
 
    }
170
 
    db.commit( );
171
 
  }
172
 
 
173
 
  private static void checkStorage( ) {
174
 
    EntityWrapper<SystemConfiguration> db = new EntityWrapper<SystemConfiguration>( );
175
 
    try {
176
 
      SystemConfiguration systemConfig = db.getUnique( new SystemConfiguration( ) );
177
 
      //if ( systemConfig.getStorageVolumesDir( ) == null ) systemConfig.setStorageVolumesDir( StorageProperties.storageRootDirectory );
178
 
    } catch ( EucalyptusCloudException ex ) {
179
 
    }
180
 
    db.commit( );
181
 
  }
182
 
 
183
 
  public static boolean checkDatabase( ) {
184
 
    /** initialize the counters **/
185
 
 
186
 
    EntityWrapper<UserInfo> db = new EntityWrapper<UserInfo>( );
187
 
    EntityWrapper<Volume> db2 = new EntityWrapper<Volume>( "eucalyptus_volumes" );
188
 
    try {
189
 
      UserInfo adminUser = db.getUnique( new UserInfo( "admin" ) );
190
 
      db.rollback( );
191
 
      return true;
192
 
    } catch ( EucalyptusCloudException e ) {
193
 
      db.rollback( );
194
 
      return false;
195
 
    } 
196
 
  }
197
 
 
198
 
  public static boolean createDb( ) {
199
 
    EntityWrapper<VmType> db2 = new EntityWrapper<VmType>( );
200
 
    try {
201
 
      if( db2.query( new VmType() ).size( ) == 0 ) { 
202
 
        db2.add( new VmType( "m1.small", 1, 2, 128 ) );
203
 
        db2.add( new VmType( "c1.medium", 1, 5, 256 ) );
204
 
        db2.add( new VmType( "m1.large", 2, 10, 512 ) );
205
 
        db2.add( new VmType( "m1.xlarge", 2, 20, 1024 ) );
206
 
        db2.add( new VmType( "c1.xlarge", 4, 20, 2048 ) );
207
 
      }
208
 
      db2.commit( );
209
 
    } catch ( Exception e ) {
210
 
      db2.rollback( );
211
 
      return false;
212
 
    }
213
 
    EntityWrapper<UserGroupInfo> db3 = new EntityWrapper<UserGroupInfo>( );
214
 
    try {
215
 
      db3.getUnique( new UserGroupInfo( "all" ) );
216
 
      db3.commit( );
217
 
    } catch ( EucalyptusCloudException e ) {
218
 
      db3.add( new UserGroupInfo( "all" ) );
219
 
      db3.commit( );
220
 
    }
221
 
    try {
222
 
      UserGroupInfo allGroup = UserGroupInfo.named( "all" );
223
 
    } catch ( EucalyptusCloudException e1 ) {
224
 
    }
225
 
    EntityWrapper<UserInfo> db = new EntityWrapper<UserInfo>( );
226
 
    try {
227
 
      db.getUnique( new UserInfo("admin") );
228
 
      db.commit( );
229
 
      return true;
230
 
    } catch ( Exception e ) {
231
 
      try {
232
 
        db.getSession( ).persist( new Counters( ) );
233
 
        UserInfo u = UserManagement.generateAdmin( );
234
 
        db.add( u );
235
 
        db.commit( );
236
 
      } catch ( HibernateException e1 ) {
237
 
        db.rollback( );
238
 
        return false;
239
 
      }
240
 
      return true;
241
 
    }
242
 
  }
243
 
 
244
 
  private static void checkDirectories( ) {
245
 
    for ( BaseDirectory dir : BaseDirectory.values( ) )
246
 
      dir.check( );
247
 
    SubDirectory.KEYS.create( );
248
 
  }
249
 
 
250
 
}