| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| * |
| */ |
| |
| package org.apache.directory.studio.valueeditors.certificate; |
| |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.File; |
| import java.io.IOException; |
| import java.security.cert.Certificate; |
| import java.security.cert.CertificateException; |
| import java.security.cert.CertificateFactory; |
| import java.security.cert.X509Certificate; |
| |
| import org.apache.directory.api.util.FileUtils; |
| import org.apache.directory.studio.common.ui.widgets.BaseWidgetUtils; |
| import org.apache.directory.studio.connection.ui.ConnectionUIPlugin; |
| import org.apache.directory.studio.connection.ui.widgets.CertificateInfoComposite; |
| import org.apache.directory.studio.valueeditors.ValueEditorsActivator; |
| import org.apache.directory.studio.valueeditors.ValueEditorsConstants; |
| import org.eclipse.core.runtime.IStatus; |
| import org.eclipse.core.runtime.Status; |
| import org.eclipse.jface.dialogs.Dialog; |
| import org.eclipse.jface.dialogs.IDialogConstants; |
| import org.eclipse.osgi.util.NLS; |
| import org.eclipse.swt.SWT; |
| import org.eclipse.swt.events.ModifyEvent; |
| import org.eclipse.swt.events.ModifyListener; |
| import org.eclipse.swt.layout.GridData; |
| import org.eclipse.swt.layout.GridLayout; |
| import org.eclipse.swt.widgets.Composite; |
| import org.eclipse.swt.widgets.Control; |
| import org.eclipse.swt.widgets.FileDialog; |
| import org.eclipse.swt.widgets.Shell; |
| import org.eclipse.swt.widgets.Text; |
| |
| |
| /** |
| * Dialog to display a X.509 certificate. It could be |
| * used to load and save the certificate from and to disk. |
| * |
| * @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a> |
| */ |
| public class CertificateDialog extends Dialog |
| { |
| |
| public static final String LOAD_FILE_NAME_TOOLTIP = "LoadFileName"; |
| |
| /** The default title. */ |
| private static final String DIALOG_TITLE = Messages.getString( "CertificateDialog.CertificateDialog" ); //$NON-NLS-1$ |
| |
| /** The button ID for the load button. */ |
| private static final int LOAD_BUTTON_ID = 9998; |
| |
| /** The button ID for the save button. */ |
| private static final int SAVE_BUTTON_ID = 9999; |
| |
| /** Hidden text to set the filename, used for UI tests. */ |
| private Text loadFilenameText; |
| |
| /** The current certificate binary data. */ |
| private byte[] currentData; |
| |
| /** The current certificate. */ |
| private X509Certificate currentCertificate; |
| |
| /** The return data, only set if OK button is pressed, null otherwise. */ |
| private byte[] returnData; |
| |
| /** The certificate info composite. */ |
| private CertificateInfoComposite certificateInfoComposite; |
| |
| |
| /** |
| * Creates a new instance of CertificateDialog. |
| * |
| * @param parentShell the parent shell |
| * @param initialData the initial data |
| */ |
| public CertificateDialog( Shell parentShell, byte[] initialData ) |
| { |
| super( parentShell ); |
| super.setShellStyle( super.getShellStyle() | SWT.RESIZE ); |
| this.currentData = initialData; |
| } |
| |
| |
| /** |
| * @see org.eclipse.jface.dialogs.Dialog#buttonPressed(int) |
| */ |
| protected void buttonPressed( int buttonId ) |
| { |
| if ( buttonId == IDialogConstants.OK_ID ) |
| { |
| returnData = currentData; |
| } |
| else if ( buttonId == SAVE_BUTTON_ID ) |
| { |
| FileDialog fileDialog = new FileDialog( getShell(), SWT.SAVE ); |
| fileDialog.setText( Messages.getString( "CertificateDialog.SaveCertificate" ) ); //$NON-NLS-1$ |
| // fileDialog.setFilterExtensions(new String[]{"*.pem"}); |
| String returnedFileName = fileDialog.open(); |
| if ( returnedFileName != null ) |
| { |
| try |
| { |
| File file = new File( returnedFileName ); |
| FileUtils.writeByteArrayToFile( file, currentData ); |
| } |
| catch ( IOException e ) |
| { |
| ConnectionUIPlugin.getDefault().getExceptionHandler().handleException( |
| new Status( IStatus.ERROR, ValueEditorsConstants.PLUGIN_ID, IStatus.ERROR, Messages |
| .getString( "CertificateDialog.CantWriteToFile" ), e ) ); //$NON-NLS-1$ |
| } |
| } |
| } |
| else if ( buttonId == LOAD_BUTTON_ID ) |
| { |
| FileDialog fileDialog = new FileDialog( getShell(), SWT.OPEN ); |
| fileDialog.setText( Messages.getString( "CertificateDialog.LoadCertificate" ) ); //$NON-NLS-1$ |
| String returnedFileName = fileDialog.open(); |
| if ( returnedFileName != null ) |
| { |
| loadFile( returnedFileName ); |
| } |
| } |
| else |
| { |
| returnData = null; |
| } |
| |
| super.buttonPressed( buttonId ); |
| } |
| |
| |
| private void loadFile( String fileName ) |
| { |
| try |
| { |
| File file = new File( fileName ); |
| currentData = FileUtils.readFileToByteArray( file ); |
| updateInput(); |
| } |
| catch ( IOException e ) |
| { |
| ConnectionUIPlugin.getDefault().getExceptionHandler().handleException( |
| new Status( IStatus.ERROR, ValueEditorsConstants.PLUGIN_ID, IStatus.ERROR, Messages |
| .getString( "CertificateDialog.CantReadFile" ), e ) ); //$NON-NLS-1$ |
| } |
| } |
| |
| |
| /** |
| * @see org.eclipse.jface.window.Window#configureShell(org.eclipse.swt.widgets.Shell) |
| */ |
| protected void configureShell( Shell shell ) |
| { |
| super.configureShell( shell ); |
| shell.setText( DIALOG_TITLE ); |
| shell.setImage( ValueEditorsActivator.getDefault().getImage( ValueEditorsConstants.IMG_CERTIFICATEEDITOR ) ); |
| } |
| |
| |
| /** |
| * @see org.eclipse.jface.dialogs.Dialog#createButtonsForButtonBar(org.eclipse.swt.widgets.Composite) |
| */ |
| protected void createButtonsForButtonBar( Composite parent ) |
| { |
| ((GridLayout) parent.getLayout()).numColumns++; |
| loadFilenameText = new Text( parent, SWT.NONE ); |
| loadFilenameText.setToolTipText( LOAD_FILE_NAME_TOOLTIP ); |
| loadFilenameText.setBackground( parent.getBackground() ); |
| loadFilenameText.addModifyListener( new ModifyListener() |
| { |
| public void modifyText( ModifyEvent e ) |
| { |
| loadFile( loadFilenameText.getText() ); |
| } |
| } ); |
| |
| createButton( parent, LOAD_BUTTON_ID, Messages.getString( "CertificateDialog.LoadCertificateButton" ), false ); //$NON-NLS-1$ |
| createButton( parent, SAVE_BUTTON_ID, Messages.getString( "CertificateDialog.SaveCertificateButton" ), false ); //$NON-NLS-1$ |
| createButton( parent, IDialogConstants.OK_ID, IDialogConstants.OK_LABEL, false ); |
| createButton( parent, IDialogConstants.CANCEL_ID, IDialogConstants.CANCEL_LABEL, false ); |
| } |
| |
| |
| /** |
| * @see org.eclipse.jface.dialogs.Dialog#createDialogArea(org.eclipse.swt.widgets.Composite) |
| */ |
| protected Control createDialogArea( Composite parent ) |
| { |
| // create composite |
| Composite composite = ( Composite ) super.createDialogArea( parent ); |
| |
| Composite certificateInfoContainer = BaseWidgetUtils.createColumnContainer( composite, 1, 1 ); |
| GridData gd = new GridData( GridData.FILL_BOTH ); |
| gd.widthHint = convertHorizontalDLUsToPixels( IDialogConstants.MINIMUM_MESSAGE_AREA_WIDTH * 3 / 2 ); |
| gd.heightHint = convertVerticalDLUsToPixels( IDialogConstants.MINIMUM_MESSAGE_AREA_WIDTH ); |
| certificateInfoContainer.setLayoutData( gd ); |
| |
| certificateInfoComposite = new CertificateInfoComposite( certificateInfoContainer, SWT.NONE ); |
| if ( currentData != null && currentData.length > 0 ) |
| { |
| updateInput(); |
| } |
| |
| applyDialogFont( composite ); |
| return composite; |
| } |
| |
| |
| /** |
| * Parses the certificate binary data and updates the input. |
| */ |
| private void updateInput() |
| { |
| try |
| { |
| // parse the certificate |
| currentCertificate = generateCertificate( currentData ); |
| |
| // update the byte[], this must be done for the case that |
| // the certificate loaded from file is in PEM format |
| currentData = currentCertificate.getEncoded(); |
| |
| // set the input and update button |
| certificateInfoComposite.setInput( new X509Certificate[] |
| { currentCertificate } ); |
| if ( getButton( IDialogConstants.OK_ID ) != null ) |
| { |
| getButton( IDialogConstants.OK_ID ).setEnabled( true ); |
| } |
| } |
| catch ( Exception e ) |
| { |
| ConnectionUIPlugin.getDefault().getExceptionHandler().handleException( |
| new Status( IStatus.ERROR, ValueEditorsConstants.PLUGIN_ID, IStatus.ERROR, Messages |
| .getString( "CertificateDialog.CantParseCertificate" ), //$NON-NLS-1$ |
| e ) ); |
| if ( getButton( IDialogConstants.OK_ID ) != null ) |
| { |
| getButton( IDialogConstants.OK_ID ).setEnabled( false ); |
| } |
| } |
| } |
| |
| |
| /** |
| * Gets the data. |
| * |
| * @return the data |
| */ |
| public byte[] getData() |
| { |
| return returnData; |
| } |
| |
| |
| public static String getCertificateInfo( byte[] data ) |
| { |
| try |
| { |
| X509Certificate certificate = generateCertificate( data ); |
| if ( certificate != null ) |
| { |
| String name = certificate.getSubjectX500Principal().getName(); |
| int version = certificate.getVersion(); |
| String type = certificate.getType(); |
| return type + "v" + version + ": " + name; //$NON-NLS-1$ //$NON-NLS-2$ |
| } |
| else |
| { |
| return NLS.bind( Messages.getString( "CertificateDialog.InvalidCertificate" ), data.length ); //$NON-NLS-1$ |
| } |
| } |
| catch ( Exception e ) |
| { |
| return NLS.bind( Messages.getString( "CertificateDialog.InvalidCertificate" ), data.length ); //$NON-NLS-1$ |
| } |
| } |
| |
| |
| private static X509Certificate generateCertificate( byte[] data ) throws CertificateException |
| { |
| CertificateFactory cf = CertificateFactory.getInstance( "X.509" ); //$NON-NLS-1$ |
| Certificate certificate = cf.generateCertificate( new ByteArrayInputStream( data ) ); |
| if ( certificate instanceof X509Certificate ) |
| { |
| return ( X509Certificate ) certificate; |
| } |
| |
| return null; |
| } |
| |
| } |